Skip to content

Commit cd465d8

Browse files
authored
sycl: Fix im2col (#14797)
1 parent 9220426 commit cd465d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-sycl/im2col.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ static void im2col_kernel(const float * x, T * dst, int64_t batch_offset, int64_
2626

2727
// make each work-item deal with more elements since sycl global range can not exceed max int
2828
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);
29+
const int64_t ksize = OW * KH;
3030
const int64_t kx = i / ksize;
3131
const int64_t kd = kx * ksize;
3232
const int64_t ky = (i - kd) / OW;

0 commit comments

Comments
 (0)