aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/Graph.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/Graph.h')
-rw-r--r--arm_compute/graph/Graph.h7
1 files changed, 5 insertions, 2 deletions
diff --git a/arm_compute/graph/Graph.h b/arm_compute/graph/Graph.h
index 878976f0af..92507e74e9 100644
--- a/arm_compute/graph/Graph.h
+++ b/arm_compute/graph/Graph.h
@@ -35,10 +35,13 @@
#include <map>
#include <memory>
#include <string>
-#include <thread>
#include <utility>
#include <vector>
+#ifndef BARE_METAL
+#include <thread>
+#endif /* BARE_METAL */
+
namespace arm_compute
{
namespace graph
@@ -231,7 +234,7 @@ private:
template <typename NT, typename... Ts>
inline NodeID Graph::add_node(Ts &&... args)
{
- std::lock_guard<arm_compute::Mutex> lock(_mtx);
+ arm_compute::lock_guard<arm_compute::Mutex> lock(_mtx);
// Create node
NodeID nid = _nodes.size();