diff --git a/src/framework/cl/cl_image.h b/src/framework/cl/cl_image.h index f7d86ec8532b8557bb51d088bdae10c92dfe77fe..4c5ce128dd13fda4005135ae95048d028988c8bc 100644 --- a/src/framework/cl/cl_image.h +++ b/src/framework/cl/cl_image.h @@ -212,10 +212,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++;