diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc index 6b7db95e5d..b2788b4810 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc @@ -277,7 +277,7 @@ bool TbeKernelSelect::IsShapeMatchFormat(const std::vector &shape, const MS_LOG(INFO) << "Warning: Server not support format with C04 suffix."; return false; } - if (format == kOpFormat_FRAC_NZ && shape.size() < kShape2dDims) { + if (format == kOpFormat_FRAC_NZ && shape.size() > kShape2dDims) { return true; } // not support format: