diff options
Diffstat (limited to 'src/graph/backends')
-rw-r--r-- | src/graph/backends/CL/CLDeviceBackend.cpp | 16 | ||||
-rw-r--r-- | src/graph/backends/GLES/GCDeviceBackend.cpp | 8 | ||||
-rw-r--r-- | src/graph/backends/NEON/NEDeviceBackend.cpp | 9 |
3 files changed, 28 insertions, 5 deletions
diff --git a/src/graph/backends/CL/CLDeviceBackend.cpp b/src/graph/backends/CL/CLDeviceBackend.cpp index 6717f9f8e3..f35daf4ae5 100644 --- a/src/graph/backends/CL/CLDeviceBackend.cpp +++ b/src/graph/backends/CL/CLDeviceBackend.cpp @@ -63,7 +63,7 @@ bool file_exists(const std::string &filename) static detail::BackendRegistrar<CLDeviceBackend> CLDeviceBackend_registrar(Target::CL); CLDeviceBackend::CLDeviceBackend() - : _initialized(false), _tuner(), _allocator(nullptr), _tuner_file() + : _context_count(0), _tuner(), _allocator(nullptr), _tuner_file() { } @@ -90,13 +90,23 @@ void CLDeviceBackend::initialize_backend() _allocator = support::cpp14::make_unique<CLBufferAllocator>(); } +void CLDeviceBackend::release_backend_context(GraphContext &ctx) +{ + ARM_COMPUTE_UNUSED(ctx); + _context_count--; + if(_context_count == 0) // No more context using the backend: free resources + { + _allocator = nullptr; + } +} + void CLDeviceBackend::setup_backend_context(GraphContext &ctx) { // Force backend initialization - if(!_initialized) + _context_count++; + if(_context_count == 1) { initialize_backend(); - _initialized = true; } // Setup tuner diff --git a/src/graph/backends/GLES/GCDeviceBackend.cpp b/src/graph/backends/GLES/GCDeviceBackend.cpp index bfac31ac2e..ec3cf4f21e 100644 --- a/src/graph/backends/GLES/GCDeviceBackend.cpp +++ b/src/graph/backends/GLES/GCDeviceBackend.cpp @@ -63,6 +63,12 @@ void GCDeviceBackend::initialize_backend() GCScheduler::get().default_init(); } +void GCDeviceBackend::release_backend_context(GraphContext &ctx) +{ + //Nothing to do + ARM_COMPUTE_UNUSED(ctx); +} + void GCDeviceBackend::setup_backend_context(GraphContext &ctx) { // Force backend initialization @@ -151,4 +157,4 @@ std::shared_ptr<arm_compute::IMemoryManager> GCDeviceBackend::create_memory_mana } } // namespace backends } // namespace graph -} // namespace arm_compute
\ No newline at end of file +} // namespace arm_compute diff --git a/src/graph/backends/NEON/NEDeviceBackend.cpp b/src/graph/backends/NEON/NEDeviceBackend.cpp index 7c2db40260..5fc44d0c68 100644 --- a/src/graph/backends/NEON/NEDeviceBackend.cpp +++ b/src/graph/backends/NEON/NEDeviceBackend.cpp @@ -61,6 +61,13 @@ NEDeviceBackend::NEDeviceBackend() void NEDeviceBackend::initialize_backend() { + //Nothing to do +} + +void NEDeviceBackend::release_backend_context(GraphContext &ctx) +{ + //Nothing to do + ARM_COMPUTE_UNUSED(ctx); } void NEDeviceBackend::setup_backend_context(GraphContext &ctx) @@ -155,4 +162,4 @@ std::shared_ptr<arm_compute::IMemoryManager> NEDeviceBackend::create_memory_mana } } // namespace backends } // namespace graph -} // namespace arm_compute
\ No newline at end of file +} // namespace arm_compute |