diff options
Diffstat (limited to 'src/kernels')
-rw-r--r-- | src/kernels/levelx/col2im.opencl | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/kernels/levelx/col2im.opencl b/src/kernels/levelx/col2im.opencl index 5cadeec6..a37db24f 100644 --- a/src/kernels/levelx/col2im.opencl +++ b/src/kernels/levelx/col2im.opencl @@ -80,9 +80,10 @@ void col2im(const int input_h, const int input_w, const int channels, } } - // Sets the resulting value + // Accumulates the resulting value with the existing im-buffer (+= val) const int input_index = w_index + input_w * (h_index + input_h * c_id); - im_buffer[input_index + im_offset] = val; + real im_buffer_value = im_buffer[input_index + im_offset]; + Add(im_buffer[input_index + im_offset], im_buffer_value, val); } } |