diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-11 11:24:25 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-11 15:49:07 +0000 |
commit | 0739fee85f102961869b80756d1243d10a989e19 (patch) | |
tree | 0d1b51e196c8eec36ad867585a92bb1e5474621c /src/dynamic/sample/SampleDynamicBackend.cpp | |
parent | 137813f2d107046e0836c8557cee0c57acd46c4b (diff) | |
download | armnn-0739fee85f102961869b80756d1243d10a989e19.tar.gz |
IVGCVSW-5203 Fix Dynamic Sample Backend build
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie2a8b0e4f439500a1978f66dd332281220c75a82
Diffstat (limited to 'src/dynamic/sample/SampleDynamicBackend.cpp')
-rw-r--r-- | src/dynamic/sample/SampleDynamicBackend.cpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/src/dynamic/sample/SampleDynamicBackend.cpp b/src/dynamic/sample/SampleDynamicBackend.cpp index 2ef8faa627..7a3475b8d4 100644 --- a/src/dynamic/sample/SampleDynamicBackend.cpp +++ b/src/dynamic/sample/SampleDynamicBackend.cpp @@ -44,9 +44,15 @@ public: } IBackendInternal::IWorkloadFactoryPtr CreateWorkloadFactory( - class TensorHandleFactoryRegistry& /*tensorHandleFactoryRegistry*/) const override + class TensorHandleFactoryRegistry& tensorHandleFactoryRegistry) const override { - return IWorkloadFactoryPtr{}; + auto memoryManager = std::make_shared<SampleMemoryManager>(); + + tensorHandleFactoryRegistry.RegisterMemoryManager(memoryManager); + tensorHandleFactoryRegistry.RegisterFactory(std::make_unique<SampleDynamicTensorHandleFactory>(memoryManager)); + + return std::make_unique<SampleDynamicWorkloadFactory>( + PolymorphicPointerDowncast<SampleMemoryManager>(memoryManager)); } IBackendInternal::IBackendProfilingContextPtr CreateBackendProfilingContext( @@ -80,7 +86,7 @@ public: return optimizationViews; } - void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) const override + void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) override { auto memoryManager = std::make_shared<SampleMemoryManager>(); |