From a5fa4918f5d732c7adc45a6f403f40ac452d03ce Mon Sep 17 00:00:00 2001
From: Margaret_wangrui <wangrui178@huawei.com>
Date: Mon, 22 Feb 2021 09:50:37 +0800
Subject: [PATCH] fix operator_info is null

---
 mindspore/ccsrc/frontend/parallel/step_parallel.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/ccsrc/frontend/parallel/step_parallel.cc b/mindspore/ccsrc/frontend/parallel/step_parallel.cc
index d5bb19c3f1..97ca59b9fa 100644
--- 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))) {