tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-08-03 15:57:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

opencl: fix im2col when KW!=KH (#14803)

Browse Source
This commit is contained in:
Sigbjørn Skjæret
2025-07-21 22:55:10 +02:00
committed by Aaron Teo
parent 120add9ef4
commit e77f241b84
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ggml/src/ggml-opencl/kernels/im2col_f16.cl
View File

@@ -31,7 +31,7 @@ kernel void kernel_im2col_f16(
src1 = (global float*)((global char*)src1 + offset1); src1 = (global float*)((global char*)src1 + offset1);
dst = (global half*)((global char*)dst + offsetd); dst = (global half*)((global char*)dst + offsetd);
long ksize = OW * (KH > 1 ? KW : 1); long ksize = OW * KH;
long kx = i / ksize; long kx = i / ksize;
long kd = kx * ksize; long kd = kx * ksize;
long ky = (i - kd) / OW; long ky = (i - kd) / OW;

2
ggml/src/ggml-opencl/kernels/im2col_f32.cl
View File

@@ -31,7 +31,7 @@ kernel void kernel_im2col_f32(
src1 = (global float*)((global char*)src1 + offset1); src1 = (global float*)((global char*)src1 + offset1);
dst = (global float*)((global char*)dst + offsetd); dst = (global float*)((global char*)dst + offsetd);
long ksize = OW * (KH > 1 ? KW : 1); long ksize = OW * KH;
long kx = i / ksize; long kx = i / ksize;
long kd = kx * ksize; long kd = kx * ksize;
long ky = (i - kd) / OW; long ky = (i - kd) / OW;