ArmNN
 25.11
Loading...
Searching...
No Matches
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}

References Measurement::TIME_US.

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

References Measurement::TIME_US.

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

References Measurement::TIME_US.

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

References Measurement::TIME_US.

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