diff --git a/mindspore/ccsrc/frontend/parallel/pass/slice_activation_in_cell_share_recompute.cc b/mindspore/ccsrc/frontend/parallel/pass/slice_activation_in_cell_share_recompute.cc index f8dd00e8a67914788ca1a83b0c3208d491721dc6..42c2c9c73fdd1090c944767ae10c30a7a872aee2 100644 --- a/mindspore/ccsrc/frontend/parallel/pass/slice_activation_in_cell_share_recompute.cc +++ b/mindspore/ccsrc/frontend/parallel/pass/slice_activation_in_cell_share_recompute.cc @@ -50,7 +50,8 @@ std::vector InferRepeatedRankList(const CNodePtr &cnode) { OperatorInfoPtr operator_info = cnode->user_data(); std::vector output_info = operator_info->outputs_tensor_info(); if (output_info.size() != 1) { - MS_LOG(EXCEPTION) << "The output_info size is wrong, node is" << cnode->DebugString(); + MS_LOG(WARNING) << "The output_info size is wrong, node is" << cnode->DebugString(); + return std::vector(); } auto tensor_layout = output_info[0].tensor_layout(); auto tensor_map = tensor_layout.origin_tensor_map(); @@ -229,6 +230,7 @@ void SliceReuseRecomputedActivationNodes(const FuncGraphPtr &graph) { auto slice_cnode = CreateSliceNode(slice_pos_node, groups); if (!slice_cnode) { MS_LOG(INFO) << "Create slice failed"; + continue; } manager->SetEdge(node, i, slice_cnode); // create depend for slice