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

!18862 fix bug of UnifyMindIR repeatly match same node 

Merge pull request !18862 from yuchaojie/unify_ir
This commit is contained in:
i-robot 2021-06-25 08:39:45 +00:00 committed by Gitee
parent d47c8596bb dd6beb0a81
commit 36b6a96922
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/session/ascend_session.cc
View File

@ -389,8 +389,8 @@ GraphId AscendSession::CompileGraphImpl(NotNull<FuncGraphPtr> func_graph) {
auto root_graph = ConstructKernelGraph(func_graph, &all_graphs); auto root_graph = ConstructKernelGraph(func_graph, &all_graphs);
for (const auto &graph : all_graphs) { for (const auto &graph : all_graphs) {
graph->set_root_graph_id(root_graph->graph_id()); graph->set_root_graph_id(root_graph->graph_id());
UnifyMindIR(graph);
} }
UnifyMindIR(root_graph);
// Update Graph Dynamic Shape Attr // Update Graph Dynamic Shape Attr
UpdateAllGraphDynamicShapeAttr(all_graphs); UpdateAllGraphDynamicShapeAttr(all_graphs);
opt::BackendCommonOptimization(root_graph); opt::BackendCommonOptimization(root_graph);