diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-05-19 20:52:00 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-05-26 14:09:49 +0100 |
commit | d8fb540568b29fd1d81a1cca667a1ad3e33ef5a1 (patch) | |
tree | 11754db57611c8069bfb3811eedd86b3304917ee /DriverOptions.cpp | |
parent | 8920cae4be95ef68295ca458514f0cc257b14f80 (diff) | |
download | android-nn-driver-d8fb540568b29fd1d81a1cca667a1ad3e33ef5a1.tar.gz |
IVGCVSW-5781 Add Async Support to Android-NN-Driver
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I1f13d04100fdb119495b9e3054425bf3babc59f1
Diffstat (limited to 'DriverOptions.cpp')
-rw-r--r-- | DriverOptions.cpp | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/DriverOptions.cpp b/DriverOptions.cpp index 42f7ea9e..5b67aa36 100644 --- a/DriverOptions.cpp +++ b/DriverOptions.cpp @@ -39,6 +39,8 @@ DriverOptions::DriverOptions(armnn::Compute computeDevice, bool fp16Enabled) , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) { } @@ -53,6 +55,8 @@ DriverOptions::DriverOptions(const std::vector<armnn::BackendId>& backends, bool , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) { } @@ -66,6 +70,8 @@ DriverOptions::DriverOptions(int argc, char** argv) , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) { std::string unsupportedOperationsAsString; std::string clTunedParametersModeAsString; @@ -154,7 +160,16 @@ DriverOptions::DriverOptions(int argc, char** argv) cxxopts::value<bool>(m_VerboseLogging)->default_value("false")) ("V,version", "Show version information", - cxxopts::value<bool>(showVersion)->default_value("false")); + cxxopts::value<bool>(showVersion)->default_value("false")) + + ("A,asyncModelExecution", "Enable AsynModel Execution", + cxxopts::value<bool>(m_EnableAsyncModelExecution)->default_value("false")) + + ("T,armnn-threads", + "Assign the number of threads used by ArmNN. " + "Input value must be at least 1. " + "Default is set to 1.", + cxxopts::value<unsigned int>(m_ArmnnNumberOfThreads)->default_value("1")); } catch (const std::exception& e) { |