Fix weird shaped tmem

peterbell10 · peterbell10 · commit 4cd6abab5da8 · 2025-08-05T21:38:01.000+01:00
diff --git a/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp b/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp
@@ -253,12 +253,6 @@ LogicalResult TensorMemoryEncodingAttr::verify(
   if (CTASplitM < 1 || CTASplitN < 1) {
     return emitError() << "CTASplitM and CTASplitN must be greater than 0";
   }
-  if (blockM != 64 && blockM != 128) {
-    return emitError() << "blockM must be 64 or 128";
-  }
-  if (!llvm::isPowerOf2_32(blockN) || blockN > 512) {
-    return emitError() << "blockN must be a power of 2 and less than 512";
-  }
   if (!unpacked && blockN < 2) {
     return emitError() << "blockN must be at least 2 for packed tensor memory";
   }

Original file line number	Diff line number	Diff line change
`@@ -253,12 +253,6 @@ LogicalResult TensorMemoryEncodingAttr::verify(`
`253`	`253`	`if (CTASplitM < 1 \|\| CTASplitN < 1) {`
`254`	`254`	`return emitError() << "CTASplitM and CTASplitN must be greater than 0";`
`255`	`255`	`}`
`256`		`- if (blockM != 64 && blockM != 128) {`
`257`		`- return emitError() << "blockM must be 64 or 128";`
`258`		`- }`
`259`		`- if (!llvm::isPowerOf2_32(blockN) \|\| blockN > 512) {`
`260`		`- return emitError() << "blockN must be a power of 2 and less than 512";`
`261`		`- }`
`262`	`256`	`if (!unpacked && blockN < 2) {`
`263`	`257`	`return emitError() << "blockN must be at least 2 for packed tensor memory";`
`264`	`258`	`}`