Skip to content

Commit dad2bab

Browse files
authored
[Dialect] Fix verifier for tmem scales (#7722)
This constraint is wrong
1 parent bf0f56d commit dad2bab

File tree

1 file changed

+0
-5
lines changed
  • lib/Dialect/TritonNvidiaGPU/IR

1 file changed

+0
-5
lines changed

lib/Dialect/TritonNvidiaGPU/IR/Ops.cpp

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -497,11 +497,6 @@ static LogicalResult verifyTMEMOperand(Operation *op, RankedTensorType type,
497497
MemDescType memdesc, StringRef regName) {
498498
if (type.getRank() != 2)
499499
return op->emitOpError(regName) << " must be a 2D tensor";
500-
if (isa<TensorMemoryScalesEncodingAttr>(memdesc.getEncoding()) &&
501-
!type.getElementType().isInteger(8)) {
502-
return op->emitOpError(regName)
503-
<< " expected to be a tensor of i8 for MMA scales encoding";
504-
}
505500
if (type.getEncoding()) {
506501
auto enc = dyn_cast<DistributedEncodingTrait>(type.getEncoding());
507502
if (!enc) {

0 commit comments

Comments
 (0)