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

fix bug of print scalar, not optimize depend when specializing

This commit is contained in:
fary86 2020-05-22 09:12:44 +08:00
parent 0ca626ef6c
commit 5138c10c6c
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/pipeline/static_analysis/program_specialize.cc
View File

@ -648,9 +648,12 @@ AnfNodePtr FuncGraphSpecializer::BuildPossibleValueNode(const AnfNodePtr &origin
ValuePtr val = ival->BuildValue(); ValuePtr val = ival->BuildValue();
if (val->isa<AnyValue>()) { if (val->isa<AnyValue>()) {
return nullptr; return nullptr;
} else {
return BuildValueNode(val, ival);
} }
// keep primitive 'depend' not to be optimized
if (IsPrimitiveCNode(origin_node, prim::kPrimDepend)) {
return nullptr;
}
return BuildValueNode(val, ival);
} }
} }