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

Fix checking nop for a cnode whose first input is not a value node

This commit is contained in:
yujianfeng 2020-10-14 17:29:14 +08:00
parent 83bdc34716
commit dc954d4c96
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/optimizer/common/helper.cc
View File

@ -405,6 +405,14 @@ bool IsNopNode(const AnfNodePtr &node) {
}
CNodePtr cnode = node->cast<CNodePtr>();
MS_EXCEPTION_IF_NULL(cnode);
if (cnode->inputs().empty()) {
return false;
}
auto input0 = cnode->input(0);
MS_EXCEPTION_IF_NULL(input0);
if (!input0->isa<ValueNode>()) {
return false;
}
bool is_nop_node = false;
if (AnfAlgo::HasNodeAttr("nop_op", cnode)) {
is_nop_node = AnfAlgo::GetNodeAttr<bool>(cnode, "nop_op");