Romain Biessy commited on
Commit
931edc1
·
1 Parent(s): 43ba97c

sycl: Fix im2col (llama/14797)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-sycl/im2col.cpp +1 -1
ggml/src/ggml-sycl/im2col.cpp CHANGED
@@ -26,7 +26,7 @@ static void im2col_kernel(const float * x, T * dst, int64_t batch_offset, int64_
26
 
27
  // make each work-item deal with more elements since sycl global range can not exceed max int
28
  for (int64_t i = global_id; i < pelements; i += (work_group_size * item_ct1.get_group_range(2))) {
29
- const int64_t ksize = OW * (KH > 1 ? KW : 1);
30
  const int64_t kx = i / ksize;
31
  const int64_t kd = kx * ksize;
32
  const int64_t ky = (i - kd) / OW;
 
26
 
27
  // make each work-item deal with more elements since sycl global range can not exceed max int
28
  for (int64_t i = global_id; i < pelements; i += (work_group_size * item_ct1.get_group_range(2))) {
29
+ const int64_t ksize = OW * KH;
30
  const int64_t kx = i / ksize;
31
  const int64_t kd = kx * ksize;
32
  const int64_t ky = (i - kd) / OW;