MLCE-52 Where has PROFILING gone?
* Call the Print() function on the Profiler when profiling is enabled.
Change-Id: I4c06ce9bfa961ac8d81d02a516502337d1fbc968
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");
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";