Skip to content

Commit a3c3084

Browse files
authored
opencl: add new ops - argsort, div, sub, addrows, sigmoid, group_norm (#13787)
* opencl: add `argsort` * opencl: add `div` * opencl: add `add_rows` * opencl: add `sub` * opencl: add `sigmoid`, both `f16` and `f32` * opencl: add `group_norm`
1 parent 1701d4c commit a3c3084

File tree

8 files changed

+1013
-1
lines changed

8 files changed

+1013
-1
lines changed

ggml/src/ggml-opencl/CMakeLists.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,14 +55,17 @@ endfunction()
5555

5656
set(GGML_OPENCL_KERNELS
5757
add
58+
argsort
5859
clamp
5960
cpy
6061
cvt
6162
diag_mask_inf
63+
div
6264
gelu
6365
gemv_noshuffle_general
6466
gemv_noshuffle
6567
get_rows
68+
group_norm
6669
im2col_f32
6770
im2col_f16
6871
mul_mat_Ab_Bi_8x4
@@ -83,11 +86,14 @@ set(GGML_OPENCL_KERNELS
8386
rms_norm
8487
rope
8588
scale
89+
sigmoid
8690
silu
8791
softmax_4_f32
8892
softmax_4_f16
8993
softmax_f32
9094
softmax_f16
95+
sub
96+
sum_rows
9197
transpose
9298
)
9399

0 commit comments

Comments
 (0)