From 516769391cafb5f96f0b3dabc5915b9b576c1ab3 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 14 Oct 2020 19:11:08 +0800
Subject: [PATCH] fix(mge/imperative): fix proxy graph comp_node finalize

GitOrigin-RevId: 22ad0d0fc612f3dd6d3060a344a0feeb072dd026
---
 imperative/src/impl/proxy_graph.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imperative/src/impl/proxy_graph.cpp b/imperative/src/impl/proxy_graph.cpp
index 4ce64ac9..0b5c96fb 100644
--- a/imperative/src/impl/proxy_graph.cpp
+++ b/imperative/src/impl/proxy_graph.cpp
@@ -364,6 +364,7 @@ class ProxyGraph::ProxyGraphImpl : public cg::ComputingGraph {
     ProxyGraph* m_owner;
     MemPool<VarNode> m_var_node_pool;
     std::vector<std::unique_ptr<OperatorNodeBase>> m_opr_refkeeper;
+    std::mutex m_opr_refkeeper_mtx;
     CompNode::UnorderedSet m_used_comp_node;
     VarReceiverInfo m_var_receiver_info;
 public:
@@ -431,7 +432,7 @@ public:
     }
 
     std::shared_ptr<void> on_comp_node_finalize() override {
-        // FIXME: mutex
+        MGB_LOCK_GUARD(m_opr_refkeeper_mtx);
         mgb_assert(!m_owner->m_cur_opr);
         // finalize would do sync first
         m_opr_refkeeper.clear();
-- 
GitLab