diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/neon/NeonTimer.cpp | 3 | ||||
-rw-r--r-- | src/backends/reference/test/RefWorkloadFactoryHelper.hpp | 4 |
2 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/neon/NeonTimer.cpp b/src/backends/neon/NeonTimer.cpp index 5cce0511ab..a7d3032b13 100644 --- a/src/backends/neon/NeonTimer.cpp +++ b/src/backends/neon/NeonTimer.cpp @@ -7,6 +7,7 @@ #include "NeonInterceptorScheduler.hpp" #include <armnn/utility/Assert.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> #include <memory> @@ -29,7 +30,7 @@ void NeonTimer::Start() { // Keep the real schedule and add NeonInterceptorScheduler as an interceptor m_RealScheduler = &arm_compute::Scheduler::get(); - arm_compute::Scheduler::set(std::static_pointer_cast<arm_compute::IScheduler>(g_Interceptor)); + arm_compute::Scheduler::set(armnn::PolymorphicPointerDowncast<arm_compute::IScheduler>(g_Interceptor)); } } 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)); } }; |