p13709645
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fmt 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-07-20 16:02:12 +08:00
parent a3d8ce36a0
commit 93778f212e
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/kernels/bang_kernel/include/bang_unary.h
View File

@ -4,13 +4,14 @@
namespace infini {
typedef enum {
Abs = 1,
Relu = 2,
Sigmoid = 3,
Abs = 1,
Relu = 2,
Sigmoid = 3,
} UnaryOpType;
void unary_kernel(cnnlHandle_t handle, const float *input,
float *output, const uint32_t num, const uint32_t op_num, UnaryOpType list[]);
void unary_kernel(cnnlHandle_t handle, const float *input, float *output,
const uint32_t num, const uint32_t op_num,
UnaryOpType list[]);
__mlu_global__ void MLUUnaryKernelUnion1(float *output, float *input,
uint32_t num, uint32_t op_list);