From d1a947fd487eb7674271f759e43507d10e932ab0 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Mon, 7 Feb 2022 13:12:24 +0000 Subject: Revert "Revert "IVGCVSW-6700 Enable import aligned host memory in android-nn-driver"" This reverts commit 8069603dc44b7673b356f66517cd8b25af8080f0. * Reason for revert: Try reenable import aligned host memory in android-nn-driver * Added a check to ArmNNDriverImpl.cpp to not call ExecuteWithDummyInputs with GpuAcc * Added new android-nn-driver driver options to enable / disable Import and Export * Import is disabled by default for now due to conv2d issues * Export is enabled by default !armnn:7147 Change-Id: I91110c58ebb3931d1c458e3774944e55c1250dd8 Signed-off-by: David Monahan --- DriverOptions.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'DriverOptions.hpp') diff --git a/DriverOptions.hpp b/DriverOptions.hpp index e1d25c48..ee68a945 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -42,6 +42,8 @@ public: unsigned int GetNumberOfThreads() const { return m_NumberOfThreads; } bool isAsyncModelExecutionEnabled() const { return m_EnableAsyncModelExecution; }; unsigned int getNoOfArmnnThreads() const { return m_ArmnnNumberOfThreads; }; + bool isImportEnabled() const { return m_EnableImport; }; + bool isExportEnabled() const { return m_EnableExport; }; private: std::vector m_Backends; @@ -63,6 +65,8 @@ private: unsigned int m_NumberOfThreads; bool m_EnableAsyncModelExecution; unsigned int m_ArmnnNumberOfThreads; + bool m_EnableImport; + bool m_EnableExport; }; } // namespace armnn_driver -- cgit v1.2.1