diff options
Diffstat (limited to 'mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp')
| -rw-r--r-- | mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp b/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp index 8474244c7d7c..697cb35a59a2 100644 --- a/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp +++ b/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp @@ -424,7 +424,7 @@ std::optional<InFlightDiagnostic> verifyTmaDescriptorWithMemref( int lastDimBytes = descMemref.getShape().back() * descMemref.getElementTypeBitWidth() / 8; - if (lastDimBytes % 16 != 0) { + if (lastDimBytes % kTMALastdimByte != 0) { return op->emitError() << "the bytes in the last dimension of the tensor " "map must be a multiple of 16"; } |