diff options
Diffstat (limited to 'src/backends/reference/test/RefWorkloadFactoryHelper.hpp')
-rw-r--r-- | src/backends/reference/test/RefWorkloadFactoryHelper.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/reference/test/RefWorkloadFactoryHelper.hpp b/src/backends/reference/test/RefWorkloadFactoryHelper.hpp index e413d045f9..f0a842de57 100644 --- a/src/backends/reference/test/RefWorkloadFactoryHelper.hpp +++ b/src/backends/reference/test/RefWorkloadFactoryHelper.hpp @@ -7,6 +7,8 @@ #include <backendsCommon/test/WorkloadFactoryHelper.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> + #include <reference/RefBackend.hpp> #include <reference/RefWorkloadFactory.hpp> #include "reference/RefTensorHandleFactory.hpp" @@ -34,7 +36,7 @@ struct WorkloadFactoryHelper<armnn::RefWorkloadFactory> const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr) { - return armnn::RefTensorHandleFactory(std::static_pointer_cast<armnn::RefMemoryManager>(memoryManager)); + return armnn::RefTensorHandleFactory(armnn::PolymorphicPointerDowncast<armnn::RefMemoryManager>(memoryManager)); } }; |