diff options
Diffstat (limited to 'src/backends/reference/RefBackend.cpp')
-rw-r--r-- | src/backends/reference/RefBackend.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp index 0a296be016..3680831fb3 100644 --- a/src/backends/reference/RefBackend.cpp +++ b/src/backends/reference/RefBackend.cpp @@ -15,6 +15,7 @@ #include <Optimizer.hpp> #include <boost/cast.hpp> +#include <boost/polymorphic_pointer_cast.hpp> namespace armnn { @@ -43,7 +44,7 @@ const BackendId& RefBackend::GetIdStatic() IBackendInternal::IWorkloadFactoryPtr RefBackend::CreateWorkloadFactory( const IBackendInternal::IMemoryManagerSharedPtr& memoryManager) const { - return std::make_unique<RefWorkloadFactory>(); + return std::make_unique<RefWorkloadFactory>(boost::polymorphic_pointer_downcast<RefMemoryManager>(memoryManager)); } IBackendInternal::IBackendContextPtr RefBackend::CreateBackendContext(const IRuntime::CreationOptions&) const @@ -53,7 +54,7 @@ IBackendInternal::IBackendContextPtr RefBackend::CreateBackendContext(const IRun IBackendInternal::IMemoryManagerUniquePtr RefBackend::CreateMemoryManager() const { - return IMemoryManagerUniquePtr{}; + return std::make_unique<RefMemoryManager>(); } IBackendInternal::Optimizations RefBackend::GetOptimizations() const |