mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-05-22 03:10:03 +00:00
opencl: fix im2col when KW!=KH (#14803)
This commit is contained in:
parent
6c9ee3b17e
commit
38d3af1b73
2 changed files with 2 additions and 2 deletions
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue