diff options
author | David Monahan <David.Monahan@arm.com> | 2021-11-04 16:31:46 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-11-05 17:23:24 +0000 |
commit | 6642b8a1f23c5f7333abf1133983eb3134af7c00 (patch) | |
tree | 01a37677720ee4e959367f53b7873322721adbba /src/backends/cl/ClBackend.cpp | |
parent | 0f13284d702b3a37f56ffce1412867b5ad9eda89 (diff) | |
download | armnn-6642b8a1f23c5f7333abf1133983eb3134af7c00.tar.gz |
IVGCVSW-6412 Implement CLBackend ICustomAllocator
* Added implementation of ClBackendDefaultAllocator
* Added back in some pure virtual functions that were mistakenly removed from the CustomMemoryAllocatorSample
* Added a new Gralloc MemorySource for memory which originates from the gpu
* Added unittests
Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: Id2abb33d82697df36426f5709756c616af3e8ed7
Diffstat (limited to 'src/backends/cl/ClBackend.cpp')
-rw-r--r-- | src/backends/cl/ClBackend.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp index dd58e002be..5c582695e5 100644 --- a/src/backends/cl/ClBackend.cpp +++ b/src/backends/cl/ClBackend.cpp @@ -5,6 +5,7 @@ #include "ClBackend.hpp" #include "ClBackendContext.hpp" +#include "ClBackendDefaultAllocator.hpp" #include "ClBackendId.hpp" #include "ClBackendModelContext.hpp" #include "ClImportTensorHandleFactory.hpp" @@ -216,6 +217,11 @@ IBackendInternal::ILayerSupportSharedPtr ClBackend::GetLayerSupport(const ModelO return layerSupport; } +std::unique_ptr<ICustomAllocator> ClBackend::GetDefaultAllocator() const +{ + return std::make_unique<ClBackendDefaultAllocator>(); +} + OptimizationViews ClBackend::OptimizeSubgraphView(const SubgraphView& subgraph, const ModelOptions& modelOptions) const { |