diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-09-14 13:50:31 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | f8349fed4e64d081fd45e4a50c86c9d0f2ddac21 (patch) | |
| tree | 0b5f8ea0f7634c5a761d0e1af8d05bbf6000b614 /src/core/CL/kernels/CLReductionOperationKernel.cpp | |
| parent | 56dd726ee074cb145612d03240b710f8adb82ddd (diff) | |
| download | ComputeLibrary-f8349fed4e64d081fd45e4a50c86c9d0f2ddac21.tar.gz | |
COMPMID-417 Fix reduction kernel's __local buffer size
Change-Id: If97a79d86b174b1d9b41360303d624e3b2d22001
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87703
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLReductionOperationKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLReductionOperationKernel.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLReductionOperationKernel.cpp b/src/core/CL/kernels/CLReductionOperationKernel.cpp index 7595d8e79b..18a8e353d7 100644 --- a/src/core/CL/kernels/CLReductionOperationKernel.cpp +++ b/src/core/CL/kernels/CLReductionOperationKernel.cpp @@ -126,7 +126,8 @@ void CLReductionOperationKernel::run(const Window &window, cl::CommandQueue &que in_slice.set(Window::DimX, Window::Dimension(in_slice.x().start(), in_slice.x().end() + border_width, in_slice.x().step())); // Set local sums buffer - _kernel.setArg(num_arguments_per_1D_tensor() * 2, _lws_hint[0], nullptr); + unsigned int local_sum_size = _lws_hint[0] * _input->info()->element_size(); + _kernel.setArg(num_arguments_per_1D_tensor() * 2, local_sum_size, nullptr); do { |