diff options
-rw-r--r-- | src/armnn/Profiling.cpp | 5 | ||||
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 6 |
2 files changed, 5 insertions, 6 deletions
diff --git a/src/armnn/Profiling.cpp b/src/armnn/Profiling.cpp index 7df6a9e255..78afb05611 100644 --- a/src/armnn/Profiling.cpp +++ b/src/armnn/Profiling.cpp @@ -485,6 +485,11 @@ void ProfilerImpl::Print(std::ostream& outStream) const // Restores previous precision settings. outStream.flags(oldFlags); outStream.precision(oldPrecision); + + if (m_DetailsToStdOutMethod == ProfilingDetailsMethod::DetailsOnly) + { + exit(0); + } } void ProfilerImpl::AnalyzeEventsAndWriteResults(std::ostream& outStream) const diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 02183a5e5c..9f81eb1168 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -72,12 +72,6 @@ int main(int argc, const char* argv[]) } executor->PrintNetworkInfo(); - - if (programOptions.m_ExNetParams.m_OutputDetailsOnlyToStdOut) - { - return EXIT_SUCCESS; - } - outputResults = executor->Execute(); if (!programOptions.m_ExNetParams.m_ComparisonComputeDevices.empty() || |