From 518a1e4a438f47267235cff450d03efc8e7599d8 Mon Sep 17 00:00:00 2001 From: Declan-ARM Date: Wed, 27 Mar 2024 09:31:11 +0000 Subject: IVGCVSW-8298 Fix FSRCNN CI Failure * Skip inference if profiling is set to output details only Signed-off-by: Declan-ARM Change-Id: I9f35d8b1237bfa66790322a40f7ae6d09bbf6eb1 --- src/armnn/Profiling.cpp | 5 ----- 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() || -- cgit v1.2.1