fix operator_info is null

4 years ago · a5fa4918f5
parent adfe6e1bc2
commit a5fa4918f5
1 changed files with 4 additions and 0 deletions
--- a/mindspore/ccsrc/frontend/parallel/step_parallel.cc
+++ b/mindspore/ccsrc/frontend/parallel/step_parallel.cc
@ -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))) {