Skip to content

Commit 7cba58b

Browse files
authored
opencl: add sqr, sqrt, mean and ssm_conv (#17476)
* opencl: add sqr * opencl: add sqrt * opencl: add mean * opencl: add ssm_conv * opencl: add missing cl_khr_fp16 * opencl: do sqrt in f32 then convert to f16 for better precision
1 parent 5449367 commit 7cba58b

File tree

6 files changed

+557
-0
lines changed

6 files changed

+557
-0
lines changed

ggml/src/ggml-opencl/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ set(GGML_OPENCL_KERNELS
7070
group_norm
7171
im2col_f32
7272
im2col_f16
73+
mean
7374
mul_mat_Ab_Bi_8x4
7475
mul_mv_f16_f16
7576
mul_mv_f16_f32_1row
@@ -109,6 +110,9 @@ set(GGML_OPENCL_KERNELS
109110
softmax_4_f16
110111
softmax_f32
111112
softmax_f16
113+
sqr
114+
sqrt
115+
ssm_conv
112116
sub
113117
sum_rows
114118
transpose

0 commit comments

Comments
 (0)