Browse Source

!7625 fix bug of nz format

Merge pull request !7625 from lianliguang/master
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
00d40d36c0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc View File

@@ -277,7 +277,7 @@ bool TbeKernelSelect::IsShapeMatchFormat(const std::vector<size_t> &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:


Loading…
Cancel
Save