diff options
| author | Derek Lamberti <derek.lamberti@arm.com> | 2018-10-01 09:28:57 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
| commit | e4ba53a85c559d4fe574305276ac815cf7995762 (patch) | |
| tree | e3f8a3e1b54b2a9cf1e21909f5e4ad9e285be49d /src/armnn/NeonInterceptorScheduler.hpp | |
| parent | fcb382af87e79ae4bdc6a604241efbf2533e1737 (diff) | |
| download | armnn-e4ba53a85c559d4fe574305276ac815cf7995762.tar.gz | |
IVGCVSW-1824 Fix slow profiling of neon. (~50% reduced end-to-end time)
Change-Id: I58295c298934317a2b365887bd9f9f6705cd0a21
Diffstat (limited to 'src/armnn/NeonInterceptorScheduler.hpp')
| -rw-r--r-- | src/armnn/NeonInterceptorScheduler.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/armnn/NeonInterceptorScheduler.hpp b/src/armnn/NeonInterceptorScheduler.hpp index 37966b8178..f33b79a2da 100644 --- a/src/armnn/NeonInterceptorScheduler.hpp +++ b/src/armnn/NeonInterceptorScheduler.hpp @@ -17,7 +17,7 @@ namespace armnn class NeonInterceptorScheduler : public arm_compute::IScheduler { public: - NeonInterceptorScheduler(NeonTimer::KernelMeasurements &kernels, arm_compute::IScheduler &realScheduler); + NeonInterceptorScheduler(arm_compute::IScheduler &realScheduler); ~NeonInterceptorScheduler() = default; void set_num_threads(unsigned int numThreads) override; @@ -28,10 +28,11 @@ public: void run_workloads(std::vector<Workload> &workloads) override; + void SetKernels(NeonTimer::KernelMeasurements* kernels) { m_Kernels = kernels; } + NeonTimer::KernelMeasurements* GetKernels() { return m_Kernels; } private: - NeonTimer::KernelMeasurements& m_Kernels; + NeonTimer::KernelMeasurements* m_Kernels; arm_compute::IScheduler& m_RealScheduler; - WallClockTimer m_Timer; }; } // namespace armnn |