From 01ce557c8d3ea837555ec91ef8595c39dad1f9d8 Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Fri, 10 Jul 2020 16:55:29 +0800 Subject: [PATCH] Fix random ci fail Signed-off-by: zhoufeng --- mindspore/ccsrc/session/kernel_graph.cc | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/session/kernel_graph.cc b/mindspore/ccsrc/session/kernel_graph.cc index 2b719ade05..c8cc6fbbee 100644 --- a/mindspore/ccsrc/session/kernel_graph.cc +++ b/mindspore/ccsrc/session/kernel_graph.cc @@ -748,6 +748,10 @@ bool KernelGraph::RemoveValueNodeFromGraph(const ValueNodePtr &value_node) { void KernelGraph::ReplaceNode(NotNull old_anf_node, NotNull new_anf_node) { MS_EXCEPTION_IF_NULL(inputs_); + { + std::queue seed_nodes; + UpdateNodeEdgeList(&seed_nodes); + } auto it = node_output_edges_.find(old_anf_node); if (it != node_output_edges_.end()) { const auto &outputs = it->second; @@ -778,8 +782,10 @@ void KernelGraph::ReplaceNode(NotNull old_anf_node, NotNull seed_nodes; + UpdateNodeEdgeList(&seed_nodes); + } // update graph inputs in child graph auto it_real_inputs = std::find_if(real_inputs_.begin(), real_inputs_.end(), [&old_anf_node](const std::pair> &n) -> bool {