commit 2b7a1581f18b88b55153d4edc5cee0e602fd1bfc
author Sadik Armagan <sadik.armagan@arm.com> Wed Sep 05 16:33:58 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Tue Sep 25 14:54:07 2018 +0100
tree cb5e1c662cbea2d3e6446d783fe181a60ab29c9c
parent ecb56cd7ce8b45966b3fc6db89441c2c00d74975

MLCE-52 Where has PROFILING gone?

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

Change-Id: I4c06ce9bfa961ac8d81d02a516502337d1fbc968

tests/InferenceModel.hpp

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index c20dd3e..6b81f80 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -299,6 +299,13 @@
         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");

tests/InferenceTest.cpp

diff --git a/tests/InferenceTest.cpp b/tests/InferenceTest.cpp
index f9a3c68..8733bc5 100644
--- a/tests/InferenceTest.cpp
+++ b/tests/InferenceTest.cpp
@@ -231,6 +231,12 @@
     BOOST_LOG_TRIVIAL(info) << std::fixed << std::setprecision(3) <<
         "Average time per test case: " << averageTimePerTestCaseMs << " ms";
 
+    // if profiling is enabled print out the results
+    if (profiler && profiler->IsProfilingEnabled())
+    {
+        profiler->Print(std::cout);
+    }
+
     if (!success)
     {
         BOOST_LOG_TRIVIAL(error) << "One or more test cases failed";