From 870b96c643388ae88dd4245b9169f526d6a8d49e Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 25 Mar 2022 21:24:56 +0000 Subject: IVGCVSW-6707 Enables a bare metal compile Change-Id: Icc2f83c5f27f413758fee3e5c1445e9fc44f42c8 Signed-off-by: Jim Flynn --- src/armnn/AsyncExecutionCallback.hpp | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'src/armnn/AsyncExecutionCallback.hpp') diff --git a/src/armnn/AsyncExecutionCallback.hpp b/src/armnn/AsyncExecutionCallback.hpp index 2ff73b3efb..9eab06b4fa 100644 --- a/src/armnn/AsyncExecutionCallback.hpp +++ b/src/armnn/AsyncExecutionCallback.hpp @@ -28,12 +28,17 @@ private: static InferenceId nextID; public: - AsyncExecutionCallback(std::queue& notificationQueue, - std::mutex& mutex, - std::condition_variable& condition) + AsyncExecutionCallback(std::queue& notificationQueue +#if !defined(ARMNN_DISABLE_THREADS) + , std::mutex& mutex + , std::condition_variable& condition +#endif + ) : m_NotificationQueue(notificationQueue) +#if !defined(ARMNN_DISABLE_THREADS) , m_Mutex(mutex) , m_Condition(condition) +#endif , m_InferenceId(++nextID) {} @@ -53,8 +58,10 @@ public: private: std::queue& m_NotificationQueue; +#if !defined(ARMNN_DISABLE_THREADS) std::mutex& m_Mutex; std::condition_variable& m_Condition; +#endif HighResolutionClock m_StartTime; HighResolutionClock m_EndTime; @@ -73,12 +80,14 @@ public: std::shared_ptr GetNotifiedCallback(); private: +#if !defined(ARMNN_DISABLE_THREADS) std::mutex m_Mutex; std::condition_variable m_Condition; +#endif std::unordered_map> m_Callbacks; std::queue m_NotificationQueue; }; } // namespace experimental -} // namespace armnn \ No newline at end of file +} // namespace armnn -- cgit v1.2.1