ArmNN
 20.05
NeonTimer Class Reference

#include <NeonTimer.hpp>

Inheritance diagram for NeonTimer:
Instrument

Public Types

using KernelMeasurements = std::vector< Measurement >
 

Public Member Functions

 NeonTimer ()=default
 
 ~NeonTimer ()=default
 
void Start () override
 
void Stop () override
 
std::vector< MeasurementGetMeasurements () const override
 
const char * GetName () const override
 
- Public Member Functions inherited from Instrument
virtual ~Instrument ()
 

Detailed Description

Definition at line 21 of file NeonTimer.hpp.

Member Typedef Documentation

◆ KernelMeasurements

using KernelMeasurements = std::vector<Measurement>

Definition at line 24 of file NeonTimer.hpp.

Constructor & Destructor Documentation

◆ NeonTimer()

NeonTimer ( )
default

◆ ~NeonTimer()

~NeonTimer ( )
default

Member Function Documentation

◆ GetMeasurements()

std::vector< Measurement > GetMeasurements ( ) const
overridevirtual

Implements Instrument.

Definition at line 46 of file NeonTimer.cpp.

References NeonTimer::GetName().

47 {
48  std::vector<Measurement> measurements = m_Kernels;
49  unsigned int kernel_number = 0;
50  for (auto & kernel : measurements)
51  {
52  std::string kernelName = std::string(this->GetName()) + "/" + std::to_string(kernel_number++) + ": " + kernel
53  .m_Name;
54  kernel.m_Name = kernelName;
55  }
56  return measurements;
57 }
const char * GetName() const override
Definition: NeonTimer.cpp:59

◆ GetName()

const char * GetName ( ) const
overridevirtual

Implements Instrument.

Definition at line 59 of file NeonTimer.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NeonTimer::GetMeasurements().

60 {
61  return "NeonKernelTimer";
62 }

◆ Start()

void Start ( )
overridevirtual

Implements Instrument.

Definition at line 22 of file NeonTimer.cpp.

References ARMNN_ASSERT.

Referenced by BOOST_AUTO_TEST_CASE().

23 {
24  m_Kernels.clear();
25  ARMNN_ASSERT(g_Interceptor->GetKernels() == nullptr);
26  g_Interceptor->SetKernels(&m_Kernels);
27 
28  m_RealSchedulerType = arm_compute::Scheduler::get_type();
29  //Note: We can't currently replace a custom scheduler
30  if(m_RealSchedulerType != arm_compute::Scheduler::Type::CUSTOM)
31  {
32  // Keep the real schedule and add NeonInterceptorScheduler as an interceptor
33  m_RealScheduler = &arm_compute::Scheduler::get();
34  arm_compute::Scheduler::set(std::static_pointer_cast<arm_compute::IScheduler>(g_Interceptor));
35  }
36 }
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14

◆ Stop()

void Stop ( )
overridevirtual

Implements Instrument.

Definition at line 38 of file NeonTimer.cpp.

39 {
40  // Restore real scheduler
41  g_Interceptor->SetKernels(nullptr);
42  arm_compute::Scheduler::set(m_RealSchedulerType);
43  m_RealScheduler = nullptr;
44 }

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