[Mlir-commits] [mlir] [mlir][nvgpu] Remove strict verifiers on `warpgroup.generate.descriptor` (PR #69935)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Tue Oct 24 08:59:58 PDT 2023


================
@@ -375,15 +375,9 @@ LogicalResult WarpgroupGenerateDescriptorOp::verify() {
   MemRefType memrefType = getTensor().getType();
   MemRefType tensorMapType = getTensorMap().getType().getTensor();
 
-  if (memrefType != tensorMapType)
-    return emitError() << "memref and tensor map type mismatch";
-
   if (!memrefType.hasStaticShape() || !tensorMapType.hasStaticShape())
     return emitError() << "supports only static shapes";
 
-  if (memrefType.getRank() != 2)
-    return emitError() << "supports only 2d memref is supported for now";
----------------
qcolombet wrote:

Since this is supposed to be used to feed into wgmma operations, why do we need to support more than 2ds?

(Sorry for the dumb questions x).)

https://github.com/llvm/llvm-project/pull/69935


More information about the Mlir-commits mailing list