From 507acd96d9247820642bef59cc2edbe3024a74a0 Mon Sep 17 00:00:00 2001
From: liuluobin <liuluobin@huawei.com>
Date: Wed, 26 Jul 2023 15:33:04 +0800
Subject: [PATCH] Fix bug in slice_activateion in cell reuse.

---
 .../parallel/pass/slice_activation_in_cell_share_recompute.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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 f8dd00e8a679..42c2c9c73fdd 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<parallel::Group> InferRepeatedRankList(const CNodePtr &cnode) {
   OperatorInfoPtr operator_info = cnode->user_data<parallel::OperatorInfo>();
   std::vector<parallel::TensorInfo> 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<parallel::Group>();
   }
   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
-- 
Gitee