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>
diff --git a/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp b/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp
index 354113b..0af7386 100644
--- a/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp
+++ b/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp
@@ -161,16 +161,8 @@
unsigned int idx = 0;
add_2D_tensor_argument(idx, _gx, slice);
add_2D_tensor_argument(idx, _gy, slice);
-
- if(_run_mag)
- {
- add_2D_tensor_argument(idx, _magnitude, slice);
- }
-
- if(_run_phase)
- {
- add_2D_tensor_argument(idx, _phase, slice);
- }
+ add_2D_tensor_argument_if((_run_mag), idx, _magnitude, slice);
+ add_2D_tensor_argument_if((_run_phase), idx, _phase, slice);
enqueue(queue, *this, slice, lws_hint());
}