diff options
author | Keith Davis <keith.davis@arm.com> | 2021-08-09 16:49:18 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-08-10 14:07:27 +0000 |
commit | f487486c843a38fced90229923433d09f99fc2e5 (patch) | |
tree | 63a2268196279a5c97c3b9dea86024f0bf4a751e /tests/ExecuteNetwork/ExecuteNetwork.cpp | |
parent | bcd860a30eba22bb2ba0943ad705734ce0ec5b23 (diff) | |
download | armnn-f487486c843a38fced90229923433d09f99fc2e5.tar.gz |
IVGCVSW-6292 Allow profiling details to be switched off during profiling
* Add switch for network details during profiling
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I8bd49fd58f0e0255598106e9ab36806ee78391d6
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index e757d2c992..64296d31b7 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -324,6 +324,7 @@ int MainImpl(const ExecuteNetworkParams& params, inferenceModelParams.m_MLGOTuningFilePath = params.m_MLGOTuningFilePath; inferenceModelParams.m_AsyncEnabled = params.m_Concurrent; inferenceModelParams.m_ThreadPoolSize = params.m_ThreadPoolSize; + inferenceModelParams.m_OutputDetailsToStdOut = params.m_OutputDetailsToStdOut; for(const std::string& inputName: params.m_InputNames) { @@ -768,6 +769,12 @@ int main(int argc, const char* argv[]) return EXIT_FAILURE; } + if (ProgramOptions.m_ExNetParams.m_OutputDetailsToStdOut && !ProgramOptions.m_ExNetParams.m_EnableProfiling) + { + ARMNN_LOG(fatal) << "You must enable profiling if you would like to output layer details"; + return EXIT_FAILURE; + } + // Create runtime std::shared_ptr<armnn::IRuntime> runtime(armnn::IRuntime::Create(ProgramOptions.m_RuntimeOptions)); |