From 2b7a1581f18b88b55153d4edc5cee0e602fd1bfc Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Wed, 5 Sep 2018 16:33:58 +0100
Subject: MLCE-52 Where has PROFILING gone?

* Call the Print() function on the Profiler when profiling is enabled.

Change-Id: I4c06ce9bfa961ac8d81d02a516502337d1fbc968
---
 tests/InferenceModel.hpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests/InferenceModel.hpp')

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index c20dd3e4b5..6b81f80cbc 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -299,6 +299,13 @@ public:
         armnn::Status ret = m_Runtime->EnqueueWorkload(m_NetworkIdentifier,
                                                        MakeInputTensors(input),
                                                        MakeOutputTensors(output));
+
+        // if profiling is enabled print out the results
+        if (profiler && profiler->IsProfilingEnabled())
+        {
+            profiler->Print(std::cout);
+        }
+
         if (ret == armnn::Status::Failure)
         {
             throw armnn::Exception("IRuntime::EnqueueWorkload failed");
-- 
cgit v1.2.1