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

delete sub_graph_changed to avoid resetkernelinfo cause error

This commit is contained in:
zengzitao 2020-10-29 20:41:32 +08:00
parent 32d6ed3e28
commit 6665684a18
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/backend/optimizer/graph_kernel/arithmetic_simplify.cc
View File

@ -37,6 +37,7 @@ AnfNodePtr NewCNodeWithInfo(const AnfNodePtrList &inputs, const AnfNodePtr &ori_
} else {
ResetKernelInfo(new_cnode, UNKNOWN_KERNEL_TYPE);
}
func_graph->AddNode(new_cnode);
return new_cnode;
}
@ -768,17 +769,12 @@ bool ArithmeticSimplify::Run(const FuncGraphPtr &func_graph) {
mng_sub = Manage(sub_graph, false);
sub_graph->set_manager(mng_sub);
}
bool sub_graph_changed = false;
for (auto node_sub : sub_graph->GetOrderedCnodes()) {
auto new_node = TrySimplify(node_sub);
if (new_node != nullptr) {
sub_graph_changed = true;
PERFORM_REPLACE(node_sub->cast<AnfNodePtr>(), new_node, sub_graph, replaced);
}
}
if (sub_graph_changed) {
ResetKernelInfo(node, AKG_KERNEL);
}
} else {
auto new_node = TrySimplify(node);
PERFORM_REPLACE(node->cast<AnfNodePtr>(), new_node, func_graph, replaced);