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;