diff options
Diffstat (limited to 'src/graph/backends/CL/CLDeviceBackend.cpp')
| -rw-r--r-- | src/graph/backends/CL/CLDeviceBackend.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/graph/backends/CL/CLDeviceBackend.cpp b/src/graph/backends/CL/CLDeviceBackend.cpp index ea3b6b801a..58c666c3cc 100644 --- a/src/graph/backends/CL/CLDeviceBackend.cpp +++ b/src/graph/backends/CL/CLDeviceBackend.cpp @@ -34,6 +34,7 @@ #include "arm_compute/graph/backends/CL/CLSubTensorHandle.h" #include "arm_compute/graph/backends/CL/CLTensorHandle.h" +#include "arm_compute/core/CL/CLCoreRuntimeContext.h" #include "arm_compute/core/TensorInfo.h" #include "arm_compute/runtime/BlobLifetimeManager.h" #include "arm_compute/runtime/CL/CLBufferAllocator.h" @@ -64,7 +65,7 @@ bool file_exists(const std::string &filename) static detail::BackendRegistrar<CLDeviceBackend> CLDeviceBackend_registrar(Target::CL); CLDeviceBackend::CLDeviceBackend() - : _context_count(0), _tuner(), _allocator(nullptr), _tuner_file() + : _context_count(0), _tuner(), _allocator(nullptr), _tuner_file(), _legacy_ctx() { } @@ -91,9 +92,9 @@ void CLDeviceBackend::initialize_backend() { // Setup Scheduler CLScheduler::get().default_init(&_tuner); - + _legacy_ctx = support::cpp14::make_unique<CLCoreRuntimeContext>(nullptr, CLScheduler::get().context(), CLScheduler::get().queue()); // Create allocator with new context - _allocator = support::cpp14::make_unique<CLBufferAllocator>(); + _allocator = support::cpp14::make_unique<CLBufferAllocator>(_legacy_ctx.get()); } void CLDeviceBackend::release_backend_context(GraphContext &ctx) |