[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