From 71d839dec7b7a6c875f58e3603e80648eb8dceb2 Mon Sep 17 00:00:00 2001 From: huanghui Date: Mon, 17 Aug 2020 14:36:18 +0800 Subject: [PATCH] fix do fission when UnsegmentSum's input0 shape size is 1 --- .../ascend/ir_fission/unsorted_segment_sum_fission.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/unsorted_segment_sum_fission.cc b/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/unsorted_segment_sum_fission.cc index 6fd81b537a..c8aaf3eeaa 100644 --- a/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/unsorted_segment_sum_fission.cc +++ b/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/unsorted_segment_sum_fission.cc @@ -94,6 +94,10 @@ const AnfNodePtr UnsortSegmentSumFission::Process(const FuncGraphPtr &graph, con return nullptr; } auto input0_shape = AnfAlgo::GetPrevNodeOutputInferShape(origin_node, 0); + if (input0_shape.size() < 2) { + MS_LOG(INFO) << "Input0's shape size less than 2, not optimize"; + return nullptr; + } if (input0_shape[input0_shape.size() - 1] != 1) { MS_LOG(INFO) << "UnsortedSegmentSum is not need fission. The last value of input0's shape is " << input0_shape[input0_shape.size() - 1];