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 {
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);


Loading…
Cancel
Save