Browse Source

!7993 delete sub_graph_changed

Merge pull request !7993 from ZengZitao/delete_sub_graph_changed
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
8e5d76d632
1 changed files with 1 additions and 5 deletions
  1. +1
    -5
      mindspore/ccsrc/backend/optimizer/graph_kernel/arithmetic_simplify.cc

+ 1
- 5
mindspore/ccsrc/backend/optimizer/graph_kernel/arithmetic_simplify.cc View File

@@ -37,6 +37,7 @@ AnfNodePtr NewCNodeWithInfo(const AnfNodePtrList &inputs, const AnfNodePtr &ori_
} else { } else {
ResetKernelInfo(new_cnode, UNKNOWN_KERNEL_TYPE); ResetKernelInfo(new_cnode, UNKNOWN_KERNEL_TYPE);
} }

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


Loading…
Cancel
Save