diff options
| -rw-r--r-- | src/armnn/NeonInterceptorScheduler.cpp | 3 | ||||
| -rw-r--r-- | src/backends/ClWorkloadFactory.cpp | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/NeonInterceptorScheduler.cpp b/src/armnn/NeonInterceptorScheduler.cpp index 6f52729092..8363def68e 100644 --- a/src/armnn/NeonInterceptorScheduler.cpp +++ b/src/armnn/NeonInterceptorScheduler.cpp @@ -42,7 +42,8 @@ void NeonInterceptorScheduler::schedule(arm_compute::ICPPKernel* kernel, const H void NeonInterceptorScheduler::run_workloads(std::vector <Workload>& workloads) { m_Timer.Start(); - m_RealScheduler.run_workloads(workloads); + // NOTE: we should think about utilising the tag to make profiling more understandable + m_RealScheduler.run_tagged_workloads(workloads, nullptr); m_Timer.Stop(); std::vector<Measurement> measurements = m_Timer.GetMeasurements(); diff --git a/src/backends/ClWorkloadFactory.cpp b/src/backends/ClWorkloadFactory.cpp index 5f395a2f6f..328c22174d 100644 --- a/src/backends/ClWorkloadFactory.cpp +++ b/src/backends/ClWorkloadFactory.cpp @@ -170,7 +170,7 @@ std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateAddition(const Additi std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateMultiplication( const MultiplicationQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<ClMultiplicationFloatWorkload, NullWorkload>(descriptor, info); + return MakeWorkload<ClMultiplicationFloatWorkload, ClMultiplicationFloatWorkload>(descriptor, info); } std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateDivision( |