diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2022-03-28 10:55:48 +0100 |
---|---|---|
committer | KeithARM <keith.davis@arm.com> | 2022-04-07 10:03:52 +0000 |
commit | dfbec2d481ee60202ddf2f6400a4bcbbf6f23cc1 (patch) | |
tree | dab9cbfd37f006a1468cd50879bea32dee354a72 /tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | |
parent | fbd2b7ee376faf6909d3fd6d271b4364630348ce (diff) | |
download | armnn-dfbec2d481ee60202ddf2f6400a4bcbbf6f23cc1.tar.gz |
IVGCVSW-6828 Add a 'reuse I/O buffers' iteration mode to ExecuteNetwork
* Add shorthand argument for no print
* Add Execute network option to reuse buffers
* Add new synchronous execute method to reuse buffers
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Ia7ee99b2ba9a21043c9575d7546bf25208357141
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp index 681dc8a611..c84c79ea78 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp +++ b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp @@ -255,7 +255,7 @@ ProgramOptions::ProgramOptions() : m_CxxOptions{"ExecuteNetwork", "Add unsupported operators as stand-in layers (where supported by parser)", cxxopts::value<bool>(m_ExNetParams.m_ParseUnsupported)->default_value("false")->implicit_value("true")) - ("do-not-print-output", + ("N,do-not-print-output", "The default behaviour of ExecuteNetwork is to print the resulting outputs on the console. " "This behaviour can be changed by adding this flag to your command.", cxxopts::value<bool>(m_ExNetParams.m_DontPrintOutputs)->default_value("false")->implicit_value("true")) @@ -378,7 +378,11 @@ ProgramOptions::ProgramOptions() : m_CxxOptions{"ExecuteNetwork", ("MLGOTuningFilePath", "Path to tuning file. Enables use of CL MLGO tuning", - cxxopts::value<std::string>(m_ExNetParams.m_MLGOTuningFilePath)); + cxxopts::value<std::string>(m_ExNetParams.m_MLGOTuningFilePath)) + + ("R, reuse-buffers", + "If enabled then the IO buffers will be reused for each inference", + cxxopts::value<bool>(m_ExNetParams.m_ReuseBuffers)->default_value("false")->implicit_value("true")); m_CxxOptions.add_options("d) Profiling") ("a,enable-external-profiling", |