diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-01-22 10:02:38 +0000 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-01-25 16:05:12 +0000 |
commit | e5617954db782628ca49919a627d01ee0088fb67 (patch) | |
tree | b72cbd4397438887404af486180b3035c01baba1 /src/backends/neon | |
parent | 5dc64fe99227efece77eb7cd14abc47474a48fb8 (diff) | |
download | armnn-e5617954db782628ca49919a627d01ee0088fb67.tar.gz |
Update ACL pin to 6a4ebe1305b544aec1ba0bfc67ed65d94fcc8c2e
* Pass in new window parameter to the schedule_op function
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: I8579d2d6e55ab9888471bd780628df0f73438498
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonInterceptorScheduler.cpp | 9 | ||||
-rw-r--r-- | src/backends/neon/NeonInterceptorScheduler.hpp | 7 |
2 files changed, 9 insertions, 7 deletions
diff --git a/src/backends/neon/NeonInterceptorScheduler.cpp b/src/backends/neon/NeonInterceptorScheduler.cpp index c20be2b359..932fdb5433 100644 --- a/src/backends/neon/NeonInterceptorScheduler.cpp +++ b/src/backends/neon/NeonInterceptorScheduler.cpp @@ -52,13 +52,14 @@ void NeonInterceptorScheduler::run_tagged_workloads(std::vector<Workload> &workl m_Kernels->emplace_back(std::string(tag != nullptr ? tag : "Unknown"), delta.count(), Measurement::Unit::TIME_US); } -void NeonInterceptorScheduler::schedule_op(arm_compute::ICPPKernel *kernel, - const Hints &hints, - arm_compute::ITensorPack &tensors ) +void NeonInterceptorScheduler::schedule_op(arm_compute::ICPPKernel* kernel, + const Hints& hints, + const arm_compute::Window& window, + arm_compute::ITensorPack& tensors ) { WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now(); - m_RealScheduler.schedule_op(kernel, hints, tensors); + m_RealScheduler.schedule_op(kernel, hints, window, tensors); WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now(); const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime); diff --git a/src/backends/neon/NeonInterceptorScheduler.hpp b/src/backends/neon/NeonInterceptorScheduler.hpp index f1d1578c8e..aeccca26e2 100644 --- a/src/backends/neon/NeonInterceptorScheduler.hpp +++ b/src/backends/neon/NeonInterceptorScheduler.hpp @@ -33,9 +33,10 @@ public: void SetKernels(NeonTimer::KernelMeasurements* kernels) { m_Kernels = kernels; } NeonTimer::KernelMeasurements* GetKernels() { return m_Kernels; } - void schedule_op(arm_compute::ICPPKernel *kernel, - const Hints &hints, - arm_compute::ITensorPack &tensors ) override; + void schedule_op(arm_compute::ICPPKernel* kernel, + const Hints& hints, + const arm_compute::Window& window, + arm_compute::ITensorPack& tensors ) override; private: NeonTimer::KernelMeasurements* m_Kernels; arm_compute::IScheduler& m_RealScheduler; |