From d5fd9767d426ca465eb660be062eecce51ad1097 Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Thu, 4 Apr 2019 13:33:10 +0100 Subject: IVGCVSW-2886 Support multiple backends in Android driver Change-Id: I4abe1f750801911570b6dc65c187b828c5929b5f Signed-off-by: Nattapat Chaimanowong --- DriverOptions.hpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'DriverOptions.hpp') diff --git a/DriverOptions.hpp b/DriverOptions.hpp index 7271ac16..637ccd64 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -9,6 +9,7 @@ #include #include +#include namespace armnn_driver { @@ -17,10 +18,11 @@ class DriverOptions { public: DriverOptions(armnn::Compute computeDevice, bool fp16Enabled = false); + DriverOptions(const std::vector& backends, bool fp16Enabled); DriverOptions(int argc, char** argv); DriverOptions(DriverOptions&& other) = default; - armnn::Compute GetComputeDevice() const { return m_ComputeDevice; } + const std::vector& GetBackends() const { return m_Backends; } bool IsVerboseLoggingEnabled() const { return m_VerboseLogging; } const std::string& GetRequestInputsAndOutputsDumpDir() const { return m_RequestInputsAndOutputsDumpDir; } const std::set& GetForcedUnsupportedOperations() const { return m_ForcedUnsupportedOperations; } @@ -30,7 +32,7 @@ public: bool GetFp16Enabled() const { return m_fp16Enabled; } private: - armnn::Compute m_ComputeDevice; + std::vector m_Backends; bool m_VerboseLogging; std::string m_RequestInputsAndOutputsDumpDir; std::set m_ForcedUnsupportedOperations; -- cgit v1.2.1