From cd639c98c347c1addae8c65eb1f4d8e75b468fdb Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Thu, 11 Feb 2021 16:57:38 +0000 Subject: IVGCVSW-5685 Add CpuAcc specific configuration option numberOfThreads * Added --number-of-threads command line option to android-nn-driver !armnn:5068 Signed-off-by: Matthew Sloyan Change-Id: I248326bf3c0355a7a17b70cb1aac7b7976820c12 --- DriverOptions.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'DriverOptions.cpp') diff --git a/DriverOptions.cpp b/DriverOptions.cpp index c30a4406..3f9b76ee 100644 --- a/DriverOptions.cpp +++ b/DriverOptions.cpp @@ -38,6 +38,7 @@ DriverOptions::DriverOptions(armnn::Compute computeDevice, bool fp16Enabled) , m_FastMathEnabled(false) , m_ShouldExit(false) , m_SaveCachedNetwork(false) + , m_NumberOfThreads(0) { } @@ -51,6 +52,7 @@ DriverOptions::DriverOptions(const std::vector& backends, bool , m_FastMathEnabled(false) , m_ShouldExit(false) , m_SaveCachedNetwork(false) + , m_NumberOfThreads(0) { } @@ -63,6 +65,7 @@ DriverOptions::DriverOptions(int argc, char** argv) , m_FastMathEnabled(false) , m_ShouldExit(false) , m_SaveCachedNetwork(false) + , m_NumberOfThreads(0) { std::string unsupportedOperationsAsString; std::string clTunedParametersModeAsString; @@ -127,6 +130,12 @@ DriverOptions::DriverOptions(int argc, char** argv) " See also --cached-network-file", cxxopts::value(m_SaveCachedNetwork)->default_value("false")) + ("number-of-threads", + "Assign the number of threads used by the CpuAcc backend. " + "Input value must be between 1 and 64. " + "Default is set to 0 (Backend will decide number of threads to use).", + cxxopts::value(m_NumberOfThreads)->default_value("0")) + ("t,cl-tuned-parameters-file", "If non-empty, the given file will be used to load/save CL tuned parameters. " "See also --cl-tuned-parameters-mode", -- cgit v1.2.1