diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc index e57370bf85..bd6e114004 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc @@ -63,7 +63,7 @@ static bool CheckStridedSlice(const CNodePtr &cnode) { << input->ToString(); } int base_number = 2; - if (shrink_axis_mask >= std::pow(base_number, input_dims - 1)) { + if (shrink_axis_mask >= std::pow(base_number, input_dims - 1) && input_dims > 1) { return false; } }