diff options
| -rw-r--r-- | src/armnn/Profiling.cpp | 5 | ||||
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 6 |
2 files changed, 6 insertions, 5 deletions
diff --git a/src/armnn/Profiling.cpp b/src/armnn/Profiling.cpp index 78afb05611..7df6a9e255 100644 --- a/src/armnn/Profiling.cpp +++ b/src/armnn/Profiling.cpp @@ -485,11 +485,6 @@ 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 9f81eb1168..02183a5e5c 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -72,6 +72,12 @@ 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() || |