mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix operator_info is null

This commit is contained in:
Margaret_wangrui 2021-02-22 09:50:37 +08:00
parent adfe6e1bc2
commit a5fa4918f5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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))) {