blob: fe80c62c5fbf80055c13962538996a1778106e32 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include "NeonTimer.hpp"
#include "WallClockTimer.hpp"
#include <arm_compute/runtime/IScheduler.h>
#include <arm_compute/runtime/Scheduler.h>
#include <arm_compute/core/CPP/ICPPKernel.h>
#include <arm_compute/core/experimental/Types.h>
namespace armnn
{
class NeonInterceptorScheduler : public arm_compute::IScheduler
{
public:
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) { m_Kernels = kernels; }
NeonTimer::KernelMeasurements* GetKernels() { return m_Kernels; }
void schedule_op(arm_compute::ICPPKernel *kernel,
const Hints &hints,
const arm_compute::InputTensorMap &inputs,
const arm_compute::OutputTensorMap &outputs ) override;
private:
NeonTimer::KernelMeasurements* m_Kernels;
arm_compute::IScheduler& m_RealScheduler;
};
} // namespace armnn
|
