diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-07-11 15:00:49 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-07-12 09:29:37 +0000 |
commit | e1651a55c637708347c9adda4a77f1deffacf7d2 (patch) | |
tree | ad58c09a1a9b141b17ad108799b70727df3c1667 /src/core/CL/kernels/CLSobel5x5Kernel.cpp | |
parent | bffb41e06c1276af00e1605ef934d05fa61f7127 (diff) | |
download | ComputeLibrary-e1651a55c637708347c9adda4a77f1deffacf7d2.tar.gz |
COMPMID-2414 Add add_2D_tensor_argument_if for OpenCL
Change-Id: I8675b4b4face0be9416ae6c7a7023234d50fc0cb
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1524
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLSobel5x5Kernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLSobel5x5Kernel.cpp | 28 |
1 files changed, 6 insertions, 22 deletions
diff --git a/src/core/CL/kernels/CLSobel5x5Kernel.cpp b/src/core/CL/kernels/CLSobel5x5Kernel.cpp index f07c1c9098..09ccf140c6 100644 --- a/src/core/CL/kernels/CLSobel5x5Kernel.cpp +++ b/src/core/CL/kernels/CLSobel5x5Kernel.cpp @@ -127,16 +127,8 @@ void CLSobel5x5HorKernel::run(const Window &window, cl::CommandQueue &queue) { unsigned int idx = 0; add_2D_tensor_argument(idx, _input, slice); - - if(_run_sobel_x) - { - add_2D_tensor_argument(idx, _output_x, slice); - } - - if(_run_sobel_y) - { - add_2D_tensor_argument(idx, _output_y, slice); - } + add_2D_tensor_argument_if((_run_sobel_x), idx, _output_x, slice); + add_2D_tensor_argument_if((_run_sobel_y), idx, _output_y, slice); enqueue(queue, *this, slice, lws_hint()); } @@ -237,18 +229,10 @@ void CLSobel5x5VertKernel::run(const Window &window, cl::CommandQueue &queue) do { unsigned int idx = 0; - - if(_run_sobel_x) - { - add_2D_tensor_argument(idx, _input_x, slice); - add_2D_tensor_argument(idx, _output_x, slice); - } - - if(_run_sobel_y) - { - add_2D_tensor_argument(idx, _input_y, slice); - add_2D_tensor_argument(idx, _output_y, slice); - } + add_2D_tensor_argument_if((_run_sobel_x), idx, _input_x, slice); + add_2D_tensor_argument_if((_run_sobel_x), idx, _output_x, slice); + add_2D_tensor_argument_if((_run_sobel_y), idx, _input_y, slice); + add_2D_tensor_argument_if((_run_sobel_y), idx, _output_y, slice); _kernel.setArg(idx++, 0 /*dummy*/); |