Browse Source

fix(mge/imperative): fix proxy graph comp_node finalize

GitOrigin-RevId: 22ad0d0fc6
release-1.4
Megvii Engine Team 4 years ago
parent
commit
ce7913e326
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/src/impl/proxy_graph.cpp

+ 2
- 1
imperative/src/impl/proxy_graph.cpp View File

@@ -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();


Loading…
Cancel
Save