diff --git a/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc b/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc index 767e4e2235..8e97db9e16 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/reshape_base.cc @@ -42,6 +42,9 @@ int ReshapeBaseCPUKernel::RunImpl(int task_id) { size_t start_index = task_id * cal_max_num_per_thread_; auto cur_in_ptr = input_ptr_ + start_index; auto cur_out_ptr = output_ptr_ + start_index; + if (start_index > in_tensors_.front()->Size()) { + return RET_OK; + } size_t data_size = in_tensors_.front()->Size() - start_index; data_size = data_size > cal_max_num_per_thread_ ? cal_max_num_per_thread_ : data_size; memcpy(cur_out_ptr, cur_in_ptr, data_size); diff --git a/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc b/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc index a0d27eb277..aeed56df4a 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/strided_slice.cc @@ -110,6 +110,9 @@ int StridedSliceCPUKernel::FastRunImpl(int task_id) { uint8_t *cur_in_ptr = input_ptr_ + (caled_num * in_shape[split_axis_] + begin_index) * inner_size_; uint8_t *cur_out_ptr = output_ptr_ + caled_num * out_shape[split_axis_] * inner_size_; int cur_outer = outer_ - caled_num; + if (cur_outer <= 0) { + return RET_OK; + } if (cur_outer > cal_num_per_thread_) { cur_outer = cal_num_per_thread_; } @@ -120,6 +123,9 @@ int StridedSliceCPUKernel::FastRunImpl(int task_id) { uint8_t *cur_in_ptr = input_ptr_ + (caled_num * param_->strides_[split_axis_] + begin_index) * inner_size_; uint8_t *cur_out_ptr = output_ptr_ + caled_num * inner_size_; int cal_axis_num = out_shape[split_axis_] - caled_num; + if (cal_axis_num <= 0) { + return RET_OK; + } if (cal_axis_num > cal_num_per_thread_) { cal_axis_num = cal_num_per_thread_; } diff --git a/mindspore/lite/test/models_caffe.cfg b/mindspore/lite/test/models_caffe.cfg index d414e9b2aa..64dc0a0ccd 100644 --- a/mindspore/lite/test/models_caffe.cfg +++ b/mindspore/lite/test/models_caffe.cfg @@ -71,3 +71,5 @@ ml_text_division ml_video_edit_person_divison ml_video_edit_img_segment ml_video_edit_video_segment_gauss_adaptis_part1 +ml_video_edit_Mnet +ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 \ No newline at end of file diff --git a/mindspore/lite/test/models_caffe_fp16.cfg b/mindspore/lite/test/models_caffe_fp16.cfg index 85c058e68e..c4fa87bf7b 100644 --- a/mindspore/lite/test/models_caffe_fp16.cfg +++ b/mindspore/lite/test/models_caffe_fp16.cfg @@ -66,3 +66,5 @@ PoseNet_dla_17_x512 5 ml_location_scene_division 8 ml_tabel_recog 0.1 ml_text_division 12 +ml_video_edit_Mnet 11 # Further analysis in the future +ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5 \ No newline at end of file diff --git a/mindspore/lite/test/models_npu.cfg b/mindspore/lite/test/models_npu.cfg index 51f3934f10..d9fe2f5502 100644 --- a/mindspore/lite/test/models_npu.cfg +++ b/mindspore/lite/test/models_npu.cfg @@ -32,3 +32,5 @@ ml_video_edit_style_transfer_candy.onnx 11 ml_video_edit_style_transfer_gongnongbing.onnx 10 ml_video_edit_style_transfer_starry.onnx 11 porseg_tmp.onnx 1 2 +ml_video_edit_Mnet 1.5 +ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5 \ No newline at end of file