diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-07-02 12:43:53 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-07-03 17:15:00 +0000 |
commit | 2aad21a900a21f467b3ec6b37420f892f0d80221 (patch) | |
tree | 7973bbf13d2bc7ea88ab0bf9d7c51e6b2d3e6907 /src/runtime/CL/functions/CLReductionOperation.cpp | |
parent | d13931d05b0d5ccea4265c342c6a3bf40a3b85cc (diff) | |
download | ComputeLibrary-2aad21a900a21f467b3ec6b37420f892f0d80221.tar.gz |
COMPMID-3388: Async support to CLReshapeLayerKernel kernels/functions
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Change-Id: I141a943dfd691069317860e852ecdd0ba7391604
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3501
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLReductionOperation.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLReductionOperation.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/runtime/CL/functions/CLReductionOperation.cpp b/src/runtime/CL/functions/CLReductionOperation.cpp index b659ecfaf6..2d7db0a20a 100644 --- a/src/runtime/CL/functions/CLReductionOperation.cpp +++ b/src/runtime/CL/functions/CLReductionOperation.cpp @@ -39,7 +39,7 @@ namespace arm_compute { CLReductionOperation::CLReductionOperation(std::shared_ptr<IMemoryManager> memory_manager) - : _memory_group(std::move(memory_manager)), _results_vector(), _reduction_kernels_vector(), _border_handlers_vector(), _reshape_kernel(), _op(), _num_of_stages(), _reduction_axis(), _is_serial(), + : _memory_group(std::move(memory_manager)), _results_vector(), _reduction_kernels_vector(), _border_handlers_vector(), _reshape(), _op(), _num_of_stages(), _reduction_axis(), _is_serial(), _is_reshape_required(false) { } @@ -152,7 +152,7 @@ Status CLReductionOperation::validate(const ITensorInfo *input, const ITensorInf if(is_reshape_required) { - ARM_COMPUTE_RETURN_ON_ERROR(CLReshapeLayerKernel::validate(output_internal, output)); + ARM_COMPUTE_RETURN_ON_ERROR(CLReshapeLayer::validate(output_internal, output)); } return Status{}; @@ -351,7 +351,7 @@ void CLReductionOperation::configure(const CLCompileContext &compile_context, IC if(_is_reshape_required) { - _reshape_kernel.configure(compile_context, &_results_vector.back(), output); + _reshape.configure(compile_context, &_results_vector.back(), output); _results_vector.back().allocator()->allocate(); } } @@ -375,7 +375,7 @@ void CLReductionOperation::run() if(_is_reshape_required) { - CLScheduler::get().enqueue(_reshape_kernel, false); + _reshape.run(); } } } // namespace arm_compute |