Spaces:
Running
Running
Romain Biessy
commited on
Commit
·
931edc1
1
Parent(s):
43ba97c
sycl: Fix im2col (llama/14797)
Browse files
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 *
|
| 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;
|