1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "NeonInterceptorScheduler.hpp"
7
8 namespace armnn{
9
NeonInterceptorScheduler(arm_compute::IScheduler & realScheduler)10 NeonInterceptorScheduler::NeonInterceptorScheduler(arm_compute::IScheduler &realScheduler)
11 : m_Kernels(nullptr), m_RealScheduler(realScheduler)
12 {
13 }
14
set_num_threads(unsigned int numThreads)15 void NeonInterceptorScheduler::set_num_threads(unsigned int numThreads)
16 {
17 m_RealScheduler.set_num_threads(numThreads);
18 }
19
num_threads() const20 unsigned int NeonInterceptorScheduler::num_threads() const
21 {
22 return m_RealScheduler.num_threads();
23 }
24
schedule(arm_compute::ICPPKernel * kernel,const Hints & hints)25 void NeonInterceptorScheduler::schedule(arm_compute::ICPPKernel* kernel, const Hints& hints)
26 {
27 WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
28 m_RealScheduler.schedule(kernel, hints.split_dimension());
29 WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
30
31 const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
32 m_Kernels->emplace_back(kernel->name(), delta.count(), Measurement::Unit::TIME_US);
33 }
34
run_workloads(std::vector<Workload> & workloads)35 void NeonInterceptorScheduler::run_workloads(std::vector <Workload>& workloads)
36 {
37 WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
38 m_RealScheduler.run_tagged_workloads(workloads, nullptr);
39 WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
40
41 const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
42 m_Kernels->emplace_back(std::string("Workload"), delta.count(), Measurement::Unit::TIME_US);
43 }
44
run_tagged_workloads(std::vector<Workload> & workloads,const char * tag)45 void NeonInterceptorScheduler::run_tagged_workloads(std::vector<Workload> &workloads, const char *tag)
46 {
47 WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
48 m_RealScheduler.run_tagged_workloads(workloads, tag);
49 WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
50
51 const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
52 m_Kernels->emplace_back(std::string(tag != nullptr ? tag : "Unknown"), delta.count(), Measurement::Unit::TIME_US);
53 }
54
schedule_op(arm_compute::ICPPKernel * kernel,const Hints & hints,const arm_compute::Window & window,arm_compute::ITensorPack & tensors)55 void NeonInterceptorScheduler::schedule_op(arm_compute::ICPPKernel* kernel,
56 const Hints& hints,
57 const arm_compute::Window& window,
58 arm_compute::ITensorPack& tensors )
59 {
60
61 WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
62 m_RealScheduler.schedule_op(kernel, hints, window, tensors);
63 WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
64
65 const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
66 m_Kernels->emplace_back(kernel->name(), delta.count(), Measurement::Unit::TIME_US);
67 }
68
69 } // namespace armnn