diff options
Diffstat (limited to 'DriverOptions.cpp')
| -rw-r--r-- | DriverOptions.cpp | 34 |
1 files changed, 30 insertions, 4 deletions
diff --git a/DriverOptions.cpp b/DriverOptions.cpp index 42f7ea9e..2998917f 100644 --- a/DriverOptions.cpp +++ b/DriverOptions.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017, 2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -17,7 +17,6 @@ #include <cxxopts/cxxopts.hpp> #include <algorithm> -#include <cassert> #include <functional> #include <string> #include <sstream> @@ -39,6 +38,10 @@ DriverOptions::DriverOptions(armnn::Compute computeDevice, bool fp16Enabled) , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) + , m_EnableImport(false) + , m_EnableExport(false) { } @@ -53,6 +56,10 @@ DriverOptions::DriverOptions(const std::vector<armnn::BackendId>& backends, bool , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) + , m_EnableImport(false) + , m_EnableExport(false) { } @@ -66,6 +73,10 @@ DriverOptions::DriverOptions(int argc, char** argv) , m_ShouldExit(false) , m_SaveCachedNetwork(false) , m_NumberOfThreads(0) + , m_EnableAsyncModelExecution(false) + , m_ArmnnNumberOfThreads(1) + , m_EnableImport(false) + , m_EnableExport(false) { std::string unsupportedOperationsAsString; std::string clTunedParametersModeAsString; @@ -154,7 +165,22 @@ 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")) + + ("I,enableImport", "Enable Importing of input buffers", + cxxopts::value<bool>(m_EnableImport)->default_value("false")) + + ("E,enableExport", "Enable Exporting of output buffers", + cxxopts::value<bool>(m_EnableExport)->default_value("false")); } catch (const std::exception& e) { @@ -168,7 +194,7 @@ DriverOptions::DriverOptions(int argc, char** argv) { cxxopts::ParseResult result = optionsDesc.parse(argc, argv); } - catch (const cxxopts::OptionException& e) + catch (const cxxopts::exceptions::exception& e) { ALOGW("An exception occurred attempting to parse program options: %s", e.what()); std::cout << optionsDesc.help() << std::endl |