Browse Source

!12434 fix operator_info is null

From: @Margaret_wangrui
Reviewed-by: 
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
2c5d19260e
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

@@ -2250,6 +2250,10 @@ LossNodeInfo FindLossCNode(const FuncGraphPtr &func_graph) {
} }
AnfNodePtr pre_node = return_node->input(1); AnfNodePtr pre_node = return_node->input(1);
MS_EXCEPTION_IF_NULL(pre_node); 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>(); auto pre_cnode = pre_node->cast<CNodePtr>();
if (pre_cnode == nullptr || !IsValueNode<Primitive>(pre_cnode->input(0))) { if (pre_cnode == nullptr || !IsValueNode<Primitive>(pre_cnode->input(0))) {


Loading…
Cancel
Save