Browse Source

fix operator_info is null

tags/v1.2.0-rc1
Margaret_wangrui 4 years ago
parent
commit
a5fa4918f5
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/frontend/parallel/step_parallel.cc

+ 4
- 0
mindspore/ccsrc/frontend/parallel/step_parallel.cc View File

@@ -2220,6 +2220,10 @@ LossNodeInfo FindLossCNode(const FuncGraphPtr &func_graph) {
}
AnfNodePtr pre_node = return_node->input(1);
MS_EXCEPTION_IF_NULL(pre_node);
if (IsPrimitiveCNode(pre_node, prim::kPrimDepend)) {
pre_node = pre_node->cast<CNodePtr>()->input(1);
MS_EXCEPTION_IF_NULL(pre_node);
}

auto pre_cnode = pre_node->cast<CNodePtr>();
if (pre_cnode == nullptr || !IsValueNode<Primitive>(pre_cnode->input(0))) {


Loading…
Cancel
Save