diff options
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r-- | src/armnn/LoadedNetwork.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp index 829405516a..45f94cbec7 100644 --- a/src/armnn/LoadedNetwork.hpp +++ b/src/armnn/LoadedNetwork.hpp @@ -102,7 +102,11 @@ public: private: - void AllocateWorkingMemory(std::lock_guard<std::mutex>& lock); + void AllocateWorkingMemory( +#if !defined(ARMNN_DISABLE_THREADS) + std::lock_guard<std::mutex>& lock +#endif + ); void AllocateAndExecuteConstantWorkloads(); void AllocateAndExecuteConstantWorkloadsAsync(); @@ -151,7 +155,9 @@ private: WorkloadQueue m_WorkloadQueue; WorkloadQueue m_OutputQueue; +#if !defined(ARMNN_DISABLE_THREADS) mutable std::mutex m_WorkingMemMutex; +#endif bool m_IsWorkingMemAllocated = false; |