From 8dfc7485c6a40f182d230495064dd3c15915dcd8 Mon Sep 17 00:00:00 2001 From: lianliguang Date: Thu, 22 Oct 2020 15:49:04 +0800 Subject: [PATCH] update mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc. fix bug of nz filter --- .../kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: