aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDeclan-ARM <decmce01@arm.com>2024-03-27 09:31:11 +0000
committerDeclan-ARM <decmce01@arm.com>2024-03-28 09:37:13 +0000
commit518a1e4a438f47267235cff450d03efc8e7599d8 (patch)
treeb3806b5639c951a895bf2c6708b81b67bb535c15
parent0cede65deff4d92505fb4aaeb97115d06b49bc3d (diff)
downloadarmnn-518a1e4a438f47267235cff450d03efc8e7599d8.tar.gz
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
-rw-r--r--src/armnn/Profiling.cpp5
-rw-r--r--tests/ExecuteNetwork/ExecuteNetwork.cpp6
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() ||