Browse Source

fix bug of print scalar, not optimize depend when specializing

tags/v0.3.0-alpha
fary86 6 years ago
parent
commit
5138c10c6c
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      mindspore/ccsrc/pipeline/static_analysis/program_specialize.cc

+ 5
- 2
mindspore/ccsrc/pipeline/static_analysis/program_specialize.cc View File

@@ -648,9 +648,12 @@ AnfNodePtr FuncGraphSpecializer::BuildPossibleValueNode(const AnfNodePtr &origin
ValuePtr val = ival->BuildValue();
if (val->isa<AnyValue>()) {
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);
}
}



Loading…
Cancel
Save