Col_lin
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!19878 Clone graph if it is used by 2 partial 

Merge pull request !19878 from chenfei_mindspore/partial_eliminate_err_r1.3
This commit is contained in:
i-robot 2021-07-10 04:21:04 +00:00 committed by Gitee
parent 509cbe72ed 6d90feb0bd
commit ff5c7b6057
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mindspore/ccsrc/frontend/optimizer/irpass/partial_eliminate.h
View File

@ -117,14 +117,19 @@ class ChoicePartialEliminater : public AnfVisitor {
}
// check funcgraph should be used once only.
for (auto &fg_node : fg_list_) {
for (size_t i = 0; i < fg_list_.size(); i++) {
auto fg_node = fg_list_[i];
auto fg = GetValueNode<FuncGraphPtr>(fg_node);
MS_EXCEPTION_IF_NULL(fg);
if (fg->func_graph_cnodes_index().size() != 1) {
for (auto iter : fg->func_graph_cnodes_index()) {
MS_LOG(ERROR) << "fg user: " << iter.first->first->DebugString(1) << ", index: " << iter.first->second;
MS_LOG(DEBUG) << "fg user: " << iter.first->first->DebugString(1) << ", index: " << iter.first->second;
}
MS_LOG(EXCEPTION) << "fg is used multiple times: " << fg->ToString();
// If a graph is used by 2 or more partial nodes at the same time, clone the graph.
auto new_fg = BasicClone(fg);
auto new_fg_node = NewValueNode(new_fg);
fg->manager()->Replace(fg_node, new_fg_node);
fg_list_[i] = new_fg_node;
}
}
return true;