aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LoadedNetwork.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r--src/armnn/LoadedNetwork.hpp8
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;