IVGCVSW-8298 Fix FSRCNN CI Failure
* Skip inference if profiling is set to output details only
Signed-off-by: Declan-ARM <decmce01@arm.com>
Change-Id: I9f35d8b1237bfa66790322a40f7ae6d09bbf6eb1
diff --git a/src/armnn/Profiling.cpp b/src/armnn/Profiling.cpp
index 78afb05..7df6a9e 100644
--- a/src/armnn/Profiling.cpp
+++ b/src/armnn/Profiling.cpp
@@ -485,11 +485,6 @@
// 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 9f81eb1..02183a5 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -72,6 +72,12 @@
}
executor->PrintNetworkInfo();
+
+ if (programOptions.m_ExNetParams.m_OutputDetailsOnlyToStdOut)
+ {
+ return EXIT_SUCCESS;
+ }
+
outputResults = executor->Execute();
if (!programOptions.m_ExNetParams.m_ComparisonComputeDevices.empty() ||