diff options
Diffstat (limited to 'src/armnn/NeonInterceptorScheduler.cpp')
| -rw-r--r-- | src/armnn/NeonInterceptorScheduler.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/NeonInterceptorScheduler.cpp b/src/armnn/NeonInterceptorScheduler.cpp index 8363def68e..7e2737e89e 100644 --- a/src/armnn/NeonInterceptorScheduler.cpp +++ b/src/armnn/NeonInterceptorScheduler.cpp @@ -31,6 +31,7 @@ void NeonInterceptorScheduler::schedule(arm_compute::ICPPKernel* kernel, const H m_RealScheduler.schedule(kernel, hints.split_dimension()); m_Timer.Stop(); + m_Timer.SetScaleFactor(Measurement::Unit::TIME_US); std::vector<Measurement> measurements = m_Timer.GetMeasurements(); BOOST_ASSERT(!measurements.empty()); @@ -46,6 +47,7 @@ void NeonInterceptorScheduler::run_workloads(std::vector <Workload>& workloads) m_RealScheduler.run_tagged_workloads(workloads, nullptr); m_Timer.Stop(); + m_Timer.SetScaleFactor(Measurement::Unit::TIME_US); std::vector<Measurement> measurements = m_Timer.GetMeasurements(); BOOST_ASSERT_MSG(measurements.size() == 3, "WallClockTimer does not have correct amount of measurements."); |