diff options
Diffstat (limited to 'src/core/CL/cl_kernels/direct_convolution.cl')
-rw-r--r-- | src/core/CL/cl_kernels/direct_convolution.cl | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/core/CL/cl_kernels/direct_convolution.cl b/src/core/CL/cl_kernels/direct_convolution.cl index 220179effb..a9a997f9ac 100644 --- a/src/core/CL/cl_kernels/direct_convolution.cl +++ b/src/core/CL/cl_kernels/direct_convolution.cl @@ -150,7 +150,12 @@ __kernel void direct_convolution_nhwc( } // Initialize the accumulators - TILE(ACC_DATA_TYPE, M0, N0, c) = { { { 0 } } }; + TILE(ACC_DATA_TYPE, M0, N0, c); + + LOOP_UNROLLING(int, i, 0, M0, 1) + { + c[i].v = 0; + } for(int i = 0; i < (_IWEI_WIDTH * _IWEI_HEIGHT); ++i) { |