opencl: fix `im2col` when `KW!=KH` (llama/14803)

diff --git a/src/ggml-opencl/kernels/im2col_f16.cl b/src/ggml-opencl/kernels/im2col_f16.cl
index b84c8984653c2bd9d7f6ebeae04b548fed0624be..cf6cdaa4ce58cf78aebd86246092e0d58e6eb9a7 100644 (file)
--- a/src/ggml-opencl/kernels/im2col_f16.cl
+++ b/src/ggml-opencl/kernels/im2col_f16.cl
@@ -31,7 +31,7 @@ kernel void kernel_im2col_f16(
     src1 = (global float*)((global char*)src1 + offset1);
     dst = (global half*)((global char*)dst + offsetd);
 
-    long  ksize = OW * (KH > 1 ? KW : 1);
+    long  ksize = OW * KH;
     long  kx = i / ksize;
     long  kd = kx * ksize;
     long  ky = (i - kd) / OW;

diff --git a/src/ggml-opencl/kernels/im2col_f32.cl b/src/ggml-opencl/kernels/im2col_f32.cl
index 4bf65e4eaafba82f50988e736cfeb495d411845b..1ecdb2344ad9de9e9c2e59d615d774c8a3ec15d9 100644 (file)
--- a/src/ggml-opencl/kernels/im2col_f32.cl
+++ b/src/ggml-opencl/kernels/im2col_f32.cl
@@ -31,7 +31,7 @@ kernel void kernel_im2col_f32(
     src1 = (global float*)((global char*)src1 + offset1);
     dst = (global float*)((global char*)dst + offsetd);
 
-    long  ksize = OW * (KH > 1 ? KW : 1);
+    long  ksize = OW * KH;
     long  kx = i / ksize;
     long  kd = kx * ksize;
     long  ky = (i - kd) / OW;