File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
include/mlir/Dialect/NVGPU/IR Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -46,8 +46,8 @@ constexpr unsigned kMaxTMATensorDimension = 5;
4646// / to be traversed along each of the kMaxTMATensorDimension (tensorRank)
4747// / dimensions, must be non-zero and less than or equal to 256.
4848constexpr unsigned kMaxTMADimension = 256 ;
49- // / Last dimension of 2D+ TMA must be 128 bytes
50- constexpr unsigned kMaxTMALastdimByte = 128 ;
49+ // / The bytes in the last dimension of the tensor map must be a multiple of 16.
50+ constexpr unsigned kTMALastdimByte = 16 ;
5151
5252#define GET_ATTRDEF_CLASSES
5353#include " mlir/Dialect/NVGPU/IR/NVGPUAttrDefs.h.inc"
Original file line number Diff line number Diff line change @@ -424,7 +424,7 @@ std::optional<InFlightDiagnostic> verifyTmaDescriptorWithMemref(
424424
425425 int lastDimBytes =
426426 descMemref.getShape ().back () * descMemref.getElementTypeBitWidth () / 8 ;
427- if (lastDimBytes % 16 != 0 ) {
427+ if (lastDimBytes % kTMALastdimByte != 0 ) {
428428 return op->emitError () << " the bytes in the last dimension of the tensor "
429429 " map must be a multiple of 16" ;
430430 }
You can’t perform that action at this time.
0 commit comments