diff --git a/src/framework/cl/cl_image.h b/src/framework/cl/cl_image.h index cac203ccc722114deb39d4dd09d433f39f8b12b3..4ee64d77c1cb62df50ff8e0dce1bda22490cc567 100644 --- a/src/framework/cl/cl_image.h +++ b/src/framework/cl/cl_image.h @@ -220,10 +220,11 @@ class CLImage { size_t i0 = 0; for (int n = 0; n < N; n++) { for (int c = 0; c < C; c++) { - size_t i1 = i0; + size_t i1 = i0 + (c / 4) * W; for (int h = 0; h < H; h++) { size_t i2 = (i1 << 2) + c % 4; for (int w = 0; w < W; w++) { + // int x = (n * width * H + h * width + (c / 4) * W + w) * 4 + (c % 4); imageData[i2] = Float2Half(*p); i2 += 4; p++;