From c577f2c6a3b4ddb6ba87a882723c53a248afbeba Mon Sep 17 00:00:00 2001
From: telsoa01 <telmo.soares@arm.com>
Date: Fri, 31 Aug 2018 09:22:23 +0100
Subject: Release 18.08

---
 src/armnn/NeonTimer.cpp | 56 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/armnn/NeonTimer.cpp

(limited to 'src/armnn/NeonTimer.cpp')

diff --git a/src/armnn/NeonTimer.cpp b/src/armnn/NeonTimer.cpp
new file mode 100644
index 0000000000..0c1e2e6a34
--- /dev/null
+++ b/src/armnn/NeonTimer.cpp
@@ -0,0 +1,56 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// See LICENSE file in the project root for full license information.
+//
+
+#include "NeonTimer.hpp"
+#include "NeonInterceptorScheduler.hpp"
+
+#include <memory>
+
+#include <boost/assert.hpp>
+#include <boost/format.hpp>
+
+namespace armnn
+{
+
+void NeonTimer::Start()
+{
+    m_Kernels.clear();
+    m_RealSchedulerType = arm_compute::Scheduler::get_type();
+    //Note: We can't currently replace a custom scheduler
+    if(m_RealSchedulerType != arm_compute::Scheduler::Type::CUSTOM)
+    {
+        // Keep the real schedule and add NeonInterceptorScheduler as an interceptor
+        m_RealScheduler  = &arm_compute::Scheduler::get();
+        auto interceptor = std::make_shared<NeonInterceptorScheduler>(m_Kernels, *m_RealScheduler);
+        arm_compute::Scheduler::set(std::static_pointer_cast<arm_compute::IScheduler>(interceptor));
+    }
+}
+
+void NeonTimer::Stop()
+{
+    // Restore real scheduler
+    arm_compute::Scheduler::set(m_RealSchedulerType);
+    m_RealScheduler = nullptr;
+}
+
+std::vector<Measurement> NeonTimer::GetMeasurements() const
+{
+    std::vector<Measurement> measurements = m_Kernels;
+    unsigned int kernel_number = 0;
+    for (auto & kernel : measurements)
+    {
+        std::string kernelName = std::string(this->GetName()) + "/" + std::to_string(kernel_number++) + ": " + kernel
+                .m_Name;
+        kernel.m_Name = kernelName;
+    }
+    return measurements;
+}
+
+const char* NeonTimer::GetName() const
+{
+    return "NeonKernelTimer";
+}
+
+}
-- 
cgit v1.2.1