ArmNN
 20.02
NeonInterceptorScheduler Class Reference

#include <NeonInterceptorScheduler.hpp>

Inheritance diagram for NeonInterceptorScheduler:

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 ()
 

Detailed Description

Definition at line 17 of file NeonInterceptorScheduler.hpp.

Constructor & Destructor Documentation

◆ NeonInterceptorScheduler()

NeonInterceptorScheduler ( arm_compute::IScheduler &  realScheduler)

Definition at line 12 of file NeonInterceptorScheduler.cpp.

13  : m_Kernels(nullptr), m_RealScheduler(realScheduler)
14 {
15 }

◆ ~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 22 of file NeonInterceptorScheduler.cpp.

23 {
24  return m_RealScheduler.num_threads();
25 }

◆ run_tagged_workloads()

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

Definition at line 47 of file NeonInterceptorScheduler.cpp.

48 {
49  WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
50  m_RealScheduler.run_tagged_workloads(workloads, tag);
51  WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
52 
53  const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
54  m_Kernels->emplace_back(std::string(tag != nullptr ? tag : "Unknown"), delta.count(), Measurement::Unit::TIME_US);
55 }

◆ run_workloads()

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

Definition at line 37 of file NeonInterceptorScheduler.cpp.

38 {
39  WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
40  m_RealScheduler.run_tagged_workloads(workloads, nullptr);
41  WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
42 
43  const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
44  m_Kernels->emplace_back(std::string("Workload"), delta.count(), Measurement::Unit::TIME_US);
45 }

◆ schedule()

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

Definition at line 27 of file NeonInterceptorScheduler.cpp.

28 {
29  WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
30  m_RealScheduler.schedule(kernel, hints.split_dimension());
31  WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
32 
33  const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime);
34  m_Kernels->emplace_back(kernel->name(), delta.count(), Measurement::Unit::TIME_US);
35 }

◆ set_num_threads()

void set_num_threads ( unsigned int  numThreads)
override

Definition at line 17 of file NeonInterceptorScheduler.cpp.

18 {
19  m_RealScheduler.set_num_threads(numThreads);
20 }

◆ 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: