From e7de63ec7150c7fc3d3f110f4f177b3f0e738a72 Mon Sep 17 00:00:00 2001 From: Michal Tarnowski Date: Mon, 4 Feb 2019 09:19:58 +0100 Subject: [PATCH] Fixed bug in GPU Eltwise for channel broadcasting --- mace/ops/opencl/cl/eltwise.cl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/ops/opencl/cl/eltwise.cl b/mace/ops/opencl/cl/eltwise.cl index 6f352d4f..ffbb953b 100644 --- a/mace/ops/opencl/cl/eltwise.cl +++ b/mace/ops/opencl/cl/eltwise.cl @@ -36,7 +36,7 @@ __kernel void eltwise(OUT_OF_RANGE_PARAMS #elif INPUT_TYPE == 3 DATA_TYPE4 in1 = READ_IMAGET(input1, SAMPLER, (int2)(chan_idx, 0)); #elif INPUT_TYPE == 4 - DATA_TYPE4 tmp = READ_IMAGET(input1, SAMPLER, (int2)(pos, hb)); + DATA_TYPE4 tmp = READ_IMAGET(input1, SAMPLER, (int2)(width_idx, hb)); DATA_TYPE4 in1 = (DATA_TYPE4)(tmp.x, tmp.x, tmp.x, tmp.x); #else DATA_TYPE4 in1 = READ_IMAGET(input1, SAMPLER, (int2)(pos, hb)); -- GitLab