diff options
Diffstat (limited to 'src/runtime/CL/functions/CLComparison.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLComparison.cpp | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/src/runtime/CL/functions/CLComparison.cpp b/src/runtime/CL/functions/CLComparison.cpp index ac56d5782c..8d5ec3571d 100644 --- a/src/runtime/CL/functions/CLComparison.cpp +++ b/src/runtime/CL/functions/CLComparison.cpp @@ -32,8 +32,13 @@ namespace arm_compute { void CLComparison::configure(ICLTensor *input1, ICLTensor *input2, ICLTensor *output, ComparisonOperation operation) { + configure(CLKernelLibrary::get().get_compile_context(), input1, input2, output, operation); +} + +void CLComparison::configure(const CLCompileContext &compile_context, ICLTensor *input1, ICLTensor *input2, ICLTensor *output, ComparisonOperation operation) +{ auto k = arm_compute::support::cpp14::make_unique<CLComparisonKernel>(); - k->configure(input1, input2, output, operation); + k->configure(compile_context, input1, input2, output, operation); _kernel = std::move(k); if(output->info()->dimension(0) > 1) @@ -42,7 +47,7 @@ void CLComparison::configure(ICLTensor *input1, ICLTensor *input2, ICLTensor *ou if(broadcasted_info->info()->dimension(0) == 1) { - _border_handler.configure(broadcasted_info, _kernel->border_size(), BorderMode::REPLICATE); + _border_handler.configure(compile_context, broadcasted_info, _kernel->border_size(), BorderMode::REPLICATE); } } } @@ -55,8 +60,14 @@ Status CLComparison::validate(const ITensorInfo *input1, const ITensorInfo *inpu template <ComparisonOperation COP> void CLComparisonStatic<COP>::configure(ICLTensor *input1, ICLTensor *input2, ICLTensor *output) { + configure(CLKernelLibrary::get().get_compile_context(), input1, input2, output); +} + +template <ComparisonOperation COP> +void CLComparisonStatic<COP>::configure(const CLCompileContext &compile_context, ICLTensor *input1, ICLTensor *input2, ICLTensor *output) +{ auto k = arm_compute::support::cpp14::make_unique<CLComparisonKernel>(); - k->configure(input1, input2, output, COP); + k->configure(compile_context, input1, input2, output, COP); _kernel = std::move(k); if(output->info()->dimension(0) > 1) @@ -65,7 +76,7 @@ void CLComparisonStatic<COP>::configure(ICLTensor *input1, ICLTensor *input2, IC if(broadcasted_info->info()->dimension(0) == 1) { - _border_handler.configure(broadcasted_info, _kernel->border_size(), BorderMode::REPLICATE); + _border_handler.configure(compile_context, broadcasted_info, _kernel->border_size(), BorderMode::REPLICATE); } } } |