diff options
Diffstat (limited to 'src/core/CL/kernels/CLMagnitudePhaseKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLMagnitudePhaseKernel.cpp | 12 |
1 files changed, 2 insertions, 10 deletions
diff --git a/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp b/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp index 354113bba5..0af738603e 100644 --- a/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp +++ b/src/core/CL/kernels/CLMagnitudePhaseKernel.cpp @@ -161,16 +161,8 @@ void CLMagnitudePhaseKernel::run(const Window &window, cl::CommandQueue &queue) 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()); } |