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

copy ref map to final graph in RecurseCompileGraph

This commit is contained in:
Margaret_wangrui 2020-06-28 15:14:17 +08:00
parent 927278be44
commit 7676043baf
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

@ -1727,6 +1727,14 @@ void AscendSession::RecurseCompileGraph(NotNull<KernelGraphPtr> graph, const Not
continue; continue;
} }
RecurseCompileGraph(NOT_NULL(child_graph), memo); 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 } // namespace session