diff options
Diffstat (limited to 'src/runtime/CL/functions/CLActivationLayer.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLActivationLayer.cpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/runtime/CL/functions/CLActivationLayer.cpp b/src/runtime/CL/functions/CLActivationLayer.cpp index 2b66795cf9..00dbb71f4c 100644 --- a/src/runtime/CL/functions/CLActivationLayer.cpp +++ b/src/runtime/CL/functions/CLActivationLayer.cpp @@ -25,18 +25,21 @@ #include "arm_compute/core/CL/kernels/CLActivationLayerKernel.h" #include "arm_compute/core/Types.h" +#include "arm_compute/runtime/CL/CLRuntimeContext.h" #include "support/ToolchainSupport.h" namespace arm_compute { -CLActivationLayer::CLActivationLayer(void *ctx) +CLActivationLayer::CLActivationLayer(CLRuntimeContext *ctx) + : ICLSimpleFunction(ctx) { - ARM_COMPUTE_UNUSED(ctx); } void CLActivationLayer::configure(ICLTensor *input, ICLTensor *output, ActivationLayerInfo act_info) { - auto k = arm_compute::support::cpp14::make_unique<CLActivationLayerKernel>(); + auto core_ctx = _ctx ? _ctx->core_runtime_context() : /* Legacy */ nullptr; + + auto k = arm_compute::support::cpp14::make_unique<CLActivationLayerKernel>(core_ctx); k->configure(input, output, act_info); _kernel = std::move(k); } |