ArmNN
 24.02
NeonInterceptorScheduler Class Reference

#include <NeonInterceptorScheduler.hpp>

Inheritance diagram for NeonInterceptorScheduler:
[legend]
Collaboration diagram for NeonInterceptorScheduler:
[legend]

Public Member Functions

 NeonInterceptorScheduler (arm_compute::IScheduler &realScheduler)
 
 ~NeonInterceptorScheduler ()=default
 
void set_num_threads (unsigned int numThreads) override
 
unsigned int num_threads () const override
 
void schedule (arm_compute::ICPPKernel *kernel, const Hints &hints) override
 
void run_workloads (std::vector< Workload > &workloads) override
 
void run_tagged_workloads (std::vector< Workload > &workloads, const char *tag) override
 
void SetKernels (NeonTimer::KernelMeasurements *kernels)
 
NeonTimer::KernelMeasurementsGetKernels ()
 
void schedule_op (arm_compute::ICPPKernel *kernel, const Hints &hints, const arm_compute::Window &window, arm_compute::ITensorPack &tensors) override
 

Detailed Description

Definition at line 17 of file NeonInterceptorScheduler.hpp.

Constructor & Destructor Documentation

◆ NeonInterceptorScheduler()

NeonInterceptorScheduler ( arm_compute::IScheduler &  realScheduler)

Definition at line 10 of file NeonInterceptorScheduler.cpp.

11  : m_Kernels(nullptr), m_RealScheduler(realScheduler)
12 {
13 }

◆ ~NeonInterceptorScheduler()

Member Function Documentation

◆ GetKernels()

NeonTimer::KernelMeasurements* GetKernels ( )
inline

Definition at line 34 of file NeonInterceptorScheduler.hpp.

34 { return m_Kernels; }

◆ num_threads()

unsigned int num_threads ( ) const
override

Definition at line 20 of file NeonInterceptorScheduler.cpp.

21 {
22  return m_RealScheduler.num_threads();
23 }

◆ run_tagged_workloads()

void run_tagged_workloads ( std::vector< Workload > &  workloads,
const char *  tag 
)
override

Definition at line 45 of file NeonInterceptorScheduler.cpp.

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 }

◆ run_workloads()

void run_workloads ( std::vector< Workload > &  workloads)
override

Definition at line 35 of file NeonInterceptorScheduler.cpp.

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 }

◆ schedule()

void schedule ( arm_compute::ICPPKernel *  kernel,
const Hints &  hints 
)
override

Definition at line 25 of file NeonInterceptorScheduler.cpp.

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 }

◆ schedule_op()

void schedule_op ( arm_compute::ICPPKernel *  kernel,
const Hints &  hints,
const arm_compute::Window &  window,
arm_compute::ITensorPack &  tensors 
)
override

Definition at line 55 of file NeonInterceptorScheduler.cpp.

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 }

◆ set_num_threads()

void set_num_threads ( unsigned int  numThreads)
override

Definition at line 15 of file NeonInterceptorScheduler.cpp.

16 {
17  m_RealScheduler.set_num_threads(numThreads);
18 }

◆ SetKernels()

void SetKernels ( NeonTimer::KernelMeasurements kernels)
inline

Definition at line 33 of file NeonInterceptorScheduler.hpp.

33 { m_Kernels = kernels; }

The documentation for this class was generated from the following files: