From 5ed7db1506f6f63db7f36b8984738b03285adcc2 Mon Sep 17 00:00:00 2001
From: zhangyunze <z13785159769@163.com>
Date: Thu, 14 Dec 2023 11:15:28 +0800
Subject: [PATCH] feat: support powOp int8

---
 src/kernels/cuda/element_wise.cu | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/kernels/cuda/element_wise.cu b/src/kernels/cuda/element_wise.cu
index f125f2b3..98a12571 100644
--- a/src/kernels/cuda/element_wise.cu
+++ b/src/kernels/cuda/element_wise.cu
@@ -204,6 +204,9 @@ void pow_kernel(int dType, void *a, void *b, void *c, int a0, int a1, int a2,
     if (dType == 1) {
         _pow_kernel<float><<<gridsize, blocksize>>>(a, b, c, a0, a1, a2, a3, b0,
                                                     b1, b2, b3, c0, c1, c2, c3);
+    } else if (dType == 3) {
+        _pow_kernel<int8_t><<<gridsize, blocksize>>>(
+            a, b, c, a0, a1, a2, a3, b0, b1, b2, b3, c0, c1, c2, c3);
     } else if (dType == 10) {
         int a_size = a0 * a1 * a2 * a3;
         int b_size = b0 * b1 * b2 * b3;