Hugging Face's logo

Spaces:
natasa365
/
whisper.cpp
Sleeping

App Files Community
Sigbjørn Skjæret commited on
Commit
2fdd2df
·
1 Parent(s): d579f20

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

Browse files
Files changed (2) hide show
  1. ggml/src/ggml-opencl/kernels/im2col_f16.cl +1 -1
  2. ggml/src/ggml-opencl/kernels/im2col_f32.cl +1 -1
ggml/src/ggml-opencl/kernels/im2col_f16.cl CHANGED
@@ -31,7 +31,7 @@ kernel void kernel_im2col_f16(
31
  src1 = (global float*)((global char*)src1 + offset1);
32
  dst = (global half*)((global char*)dst + offsetd);
33
 
34
- long ksize = OW * (KH > 1 ? KW : 1);
35
  long kx = i / ksize;
36
  long kd = kx * ksize;
37
  long ky = (i - kd) / OW;
 
31
  src1 = (global float*)((global char*)src1 + offset1);
32
  dst = (global half*)((global char*)dst + offsetd);
33
 
34
+ long ksize = OW * KH;
35
  long kx = i / ksize;
36
  long kd = kx * ksize;
37
  long ky = (i - kd) / OW;
ggml/src/ggml-opencl/kernels/im2col_f32.cl CHANGED
@@ -31,7 +31,7 @@ kernel void kernel_im2col_f32(
31
  src1 = (global float*)((global char*)src1 + offset1);
32
  dst = (global float*)((global char*)dst + offsetd);
33
 
34
- long ksize = OW * (KH > 1 ? KW : 1);
35
  long kx = i / ksize;
36
  long kd = kx * ksize;
37
  long ky = (i - kd) / OW;
 
31
  src1 = (global float*)((global char*)src1 + offset1);
32
  dst = (global float*)((global char*)dst + offsetd);
33
 
34
+ long ksize = OW * KH;
35
  long kx = i / ksize;
36
  long kd = kx * ksize;
37
  long ky = (i - kd) / OW;