diff options
Diffstat (limited to 'tests/framework/instruments/SchedulerTimer.cpp')
-rw-r--r-- | tests/framework/instruments/SchedulerTimer.cpp | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/framework/instruments/SchedulerTimer.cpp b/tests/framework/instruments/SchedulerTimer.cpp index 50d77dd5b9..76f1a58e9c 100644 --- a/tests/framework/instruments/SchedulerTimer.cpp +++ b/tests/framework/instruments/SchedulerTimer.cpp @@ -76,19 +76,26 @@ public: _kernels.push_back(std::move(info)); } - void run_workloads(std::vector<Workload> &workloads) override + void run_tagged_workloads(std::vector<Workload> &workloads, const char *tag) override { _timer.start(); - _real_scheduler.run_workloads(workloads); + _real_scheduler.run_tagged_workloads(workloads, tag); _timer.stop(); SchedulerTimer::kernel_info info; - info.name = "Unknown"; + info.name = tag != nullptr ? tag : "Unknown"; info.prefix = _prefix; info.measurements = _timer.measurements(); _kernels.push_back(std::move(info)); } +protected: + void run_workloads(std::vector<Workload> &workloads) override + { + ARM_COMPUTE_UNUSED(workloads); + ARM_COMPUTE_ERROR("Can't be reached"); + } + private: std::list<SchedulerTimer::kernel_info> &_kernels; IScheduler &_real_scheduler; |