diff options
author | Jan Eilers <jan.eilers@arm.com> | 2021-07-22 13:17:04 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2021-08-10 16:35:33 +0100 |
commit | c1c872f12797ef6fe52c4589113e7efc353e56eb (patch) | |
tree | 911320c5306f9d2273ee76201806bfb12cbe4cd3 /src/backends/cl/ClRegistryInitializer.cpp | |
parent | f487486c843a38fced90229923433d09f99fc2e5 (diff) | |
download | armnn-c1c872f12797ef6fe52c4589113e7efc353e56eb.tar.gz |
Adds CustomAllocator interface and Sample App
* Updates the runtime options with a CustomAllocatorMap which allows to define a CustomAllocator for specific backends
* Change IBackendInternal interface to use a shared pointer to a custom allocator
* Update ClBackend.hpp/cpp to use the CustomAllocator
* Adds an example application and unit test which uses a CustomAllocator for GpuAcc
* Refactor of the interface to use MemorySource instead of the user Mapping cl_mem directly
* Modify the BackendRegistry to also hold a registry of CustomAllocators
* BackendRegistry Deregister will also deregister any allocators associated with that backend id
* set_global_allocator within the BaseMemoryManager so that it always matches the currently used allocator
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I156d819686021865f4375e6cb7a5c5dec8fee9e8
Signed-off-by: David Monahan <david.monahan@arm.com>
Diffstat (limited to 'src/backends/cl/ClRegistryInitializer.cpp')
-rw-r--r-- | src/backends/cl/ClRegistryInitializer.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/cl/ClRegistryInitializer.cpp b/src/backends/cl/ClRegistryInitializer.cpp index 8decd6f689..aadc14bd68 100644 --- a/src/backends/cl/ClRegistryInitializer.cpp +++ b/src/backends/cl/ClRegistryInitializer.cpp @@ -18,6 +18,14 @@ static BackendRegistry::StaticRegistryInitializer g_RegisterHelper ClBackend::GetIdStatic(), []() { + // Check if we have a CustomMemoryAllocator associated with the backend + // and if so register it with the backend. + auto customAllocators = BackendRegistryInstance().GetAllocators(); + auto allocatorIterator = customAllocators.find(ClBackend::GetIdStatic()); + if (allocatorIterator != customAllocators.end()) + { + return IBackendInternalUniquePtr(new ClBackend(allocatorIterator->second)); + } return IBackendInternalUniquePtr(new ClBackend); } }; |