IVGCVSW-2554 Update NEON scheduler to output workload names rather than unknown

Change-Id: I0d14baf9002c894790da171a6d60c0e4ac929078

diff --git a/src/backends/neon/NeonInterceptorScheduler.cpp b/src/backends/neon/NeonInterceptorScheduler.cpp
index 03b4670..d8dd01b 100644 (file)
--- a/src/backends/neon/NeonInterceptorScheduler.cpp
+++ b/src/backends/neon/NeonInterceptorScheduler.cpp
@@ -44,4 +44,14 @@ void NeonInterceptorScheduler::run_workloads(std::vector <Workload>& workloads)
     m_Kernels->emplace_back(std::string("Workload"), delta.count(), Measurement::Unit::TIME_US);
 }
 
+void NeonInterceptorScheduler::run_tagged_workloads(std::vector<Workload> &workloads, const char *tag)
+{
+    WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now();
+    m_RealScheduler.run_tagged_workloads(workloads, tag);
+    WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now();
+
+    const auto delta       = std::chrono::duration<double, std::micro>(stopTime - startTime);
+    m_Kernels->emplace_back(std::string(tag != nullptr ? tag : "Unknown"), delta.count(), Measurement::Unit::TIME_US);
+}
+
 } // namespace armnn
\ No newline at end of file

diff --git a/src/backends/neon/NeonInterceptorScheduler.hpp b/src/backends/neon/NeonInterceptorScheduler.hpp
index f33b79a..3a6f2aa 100644 (file)
--- a/src/backends/neon/NeonInterceptorScheduler.hpp
+++ b/src/backends/neon/NeonInterceptorScheduler.hpp
@@ -28,6 +28,8 @@ public:
 
     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; }
 private: