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

!1368 don't eliminate nop node in getnext-memcpy elimination 

Merge pull request !1368 from laiyongqiang/getnext_memcpy_elimination_bugfix
This commit is contained in:
mindspore-ci-bot 2020-05-23 14:33:43 +08:00 committed by Gitee
parent 9fe6074c3f 1c44a6edbe
commit 910178e4d5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/pre_activate/ascend/enhancer/getnext_memcpy_elimination.cc
View File

@ -56,9 +56,12 @@ const AnfNodePtr GetnextMemcpyElimination::Process(const FuncGraphPtr &graph, co
return nullptr;
}
// 3. next_node has only one input which is memcpy's output
// 3. next_node is not nop node and it has only one input which is memcpy's output
for (auto &item : next_nodes) {
auto next_node = item.first->cast<CNodePtr>();
if (opt::IsNopNode(next_node)) {
return nullptr;
}
if (next_node->inputs().size() != 2) {
MS_LOG(DEBUG) << "next node has more than one input";
return nullptr;