p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2651 handle RecurseCompileGraph when one branch is Assign 

Merge pull request !2651 from Margaret_wangrui/copy_ref_map_to_final_graph_in_RecurseCompileGraph
This commit is contained in:
mindspore-ci-bot 2020-06-28 17:00:22 +08:00 committed by Gitee
parent 9b326fe8c5 7676043baf
commit e83c563022
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/session/ascend_session.cc
View File

@ -1735,6 +1735,14 @@ void AscendSession::RecurseCompileGraph(NotNull<KernelGraphPtr> graph, const Not
continue;
}
RecurseCompileGraph(NOT_NULL(child_graph), memo);
// copy ref map to final graph
auto child_ref_map = child_graph->GetRefMap();
for (auto &item : child_ref_map) {
if (graph->IsInRefOutputMap(item.first)) {
MS_LOG(EXCEPTION) << "The ref pair is already in final graph!";
}
graph->AddRefCorrespondPairs(item.first, item.second);
}
}
}
} // namespace session