diff options
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r-- | DriverOptions.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp index 7271ac16..637ccd64 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -9,6 +9,7 @@ #include <set> #include <string> +#include <vector> namespace armnn_driver { @@ -17,10 +18,11 @@ class DriverOptions { public: DriverOptions(armnn::Compute computeDevice, bool fp16Enabled = false); + DriverOptions(const std::vector<armnn::BackendId>& backends, bool fp16Enabled); DriverOptions(int argc, char** argv); DriverOptions(DriverOptions&& other) = default; - armnn::Compute GetComputeDevice() const { return m_ComputeDevice; } + const std::vector<armnn::BackendId>& GetBackends() const { return m_Backends; } bool IsVerboseLoggingEnabled() const { return m_VerboseLogging; } const std::string& GetRequestInputsAndOutputsDumpDir() const { return m_RequestInputsAndOutputsDumpDir; } const std::set<unsigned int>& GetForcedUnsupportedOperations() const { return m_ForcedUnsupportedOperations; } @@ -30,7 +32,7 @@ public: bool GetFp16Enabled() const { return m_fp16Enabled; } private: - armnn::Compute m_ComputeDevice; + std::vector<armnn::BackendId> m_Backends; bool m_VerboseLogging; std::string m_RequestInputsAndOutputsDumpDir; std::set<unsigned int> m_ForcedUnsupportedOperations; |