diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2018-09-05 16:33:58 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:07 +0100 |
commit | 2b7a1581f18b88b55153d4edc5cee0e602fd1bfc (patch) | |
tree | cb5e1c662cbea2d3e6446d783fe181a60ab29c9c /tests/InferenceModel.hpp | |
parent | ecb56cd7ce8b45966b3fc6db89441c2c00d74975 (diff) | |
download | armnn-2b7a1581f18b88b55153d4edc5cee0e602fd1bfc.tar.gz |
MLCE-52 Where has PROFILING gone?
* Call the Print() function on the Profiler when profiling is enabled.
Change-Id: I4c06ce9bfa961ac8d81d02a516502337d1fbc968
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
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"); |