diff options
author | Kevin May <kevin.may@arm.com> | 2021-05-26 16:01:08 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-05-26 16:19:38 +0100 |
commit | 94dd4dbbb2b59f47e38cf2f1280c4b25be45ea64 (patch) | |
tree | b179186bcac41484dad37bcee18a705a05e462f6 /tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | |
parent | b4b3ac91990eb5deaffca2300319f2ddf7aa0886 (diff) | |
download | armnn-94dd4dbbb2b59f47e38cf2f1280c4b25be45ea64.tar.gz |
IVGCVSW-6009 Enable creating thread pool with 1 thread
* Allow the user to use create a tread pool with a single thread
* This is in keeping with how the android-nn-driver was implemented
* Add it to ExecuteNetwork thread pool creation
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I05b8048a9e0e45ae11d2b585080af28d9d008d81
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp index 1f57f85252..25ddecf3ba 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp +++ b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp @@ -292,8 +292,8 @@ ProgramOptions::ProgramOptions() : m_CxxOptions{"ExecuteNetwork", ("thread-pool-size", "Number of Arm NN threads to use when running the network asynchronously via the Arm NN thread pool. " - "The default is set to 1", - cxxopts::value<size_t>(m_ExNetParams.m_ThreadPoolSize)->default_value("1")); + "The default is set to 0", + cxxopts::value<size_t>(m_ExNetParams.m_ThreadPoolSize)->default_value("0")); m_CxxOptions.add_options("c) Optimization") ("bf16-turbo-mode", @@ -461,7 +461,7 @@ void ProgramOptions::ParseOptions(int ac, const char* av[]) } // Set concurrent to true if the user expects to run inferences asynchronously - if (m_ExNetParams.m_SimultaneousIterations > 1) + if (m_ExNetParams.m_SimultaneousIterations > 1 || m_ExNetParams.m_ThreadPoolSize > 0) { m_ExNetParams.m_Concurrent = true; } |