aboutsummaryrefslogtreecommitdiff
path: root/src/dynamic/sample/SampleDynamicBackend.cpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2020-08-11 11:24:25 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2020-08-11 15:49:07 +0000
commit0739fee85f102961869b80756d1243d10a989e19 (patch)
tree0d1b51e196c8eec36ad867585a92bb1e5474621c /src/dynamic/sample/SampleDynamicBackend.cpp
parent137813f2d107046e0836c8557cee0c57acd46c4b (diff)
downloadarmnn-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.cpp12
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>();