diff options
Diffstat (limited to 'src/runtime/CL/CLHelpers.cpp')
-rw-r--r-- | src/runtime/CL/CLHelpers.cpp | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/src/runtime/CL/CLHelpers.cpp b/src/runtime/CL/CLHelpers.cpp index edfc8ed2aa..c4c7ee2107 100644 --- a/src/runtime/CL/CLHelpers.cpp +++ b/src/runtime/CL/CLHelpers.cpp @@ -26,6 +26,7 @@ #include "arm_compute/core/CL/CLHelpers.h" #include "arm_compute/core/Error.h" +#include "arm_compute/runtime/CL/CLRuntimeContext.h" namespace { @@ -103,4 +104,19 @@ create_opencl_context_and_device() ARM_COMPUTE_ERROR_ON_MSG(err != CL_SUCCESS, "Failed to create OpenCL context"); return std::make_tuple(cl_context, device, err); } + +void schedule_kernel_on_ctx(CLRuntimeContext *ctx, ICLKernel *kernel, bool flush) +{ + ARM_COMPUTE_ERROR_ON_NULLPTR(kernel); + if(ctx) + { + ARM_COMPUTE_ERROR_ON(ctx->gpu_scheduler() == nullptr); + ctx->gpu_scheduler()->enqueue(*kernel, flush); + } + else + { + CLScheduler::get().enqueue(*kernel, flush); + } +} + } // namespace arm_compute |