[Mlir-commits] [mlir] [MLIR][NVGPU] Introduce `nvgpu.wargroup.mma.store` Op for Hopper GPUs (PR #65441)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Fri Sep 29 09:04:38 PDT 2023
================
@@ -529,6 +530,39 @@ LogicalResult WarpgroupMmaOp::verify() {
return success();
}
+LogicalResult WarpgroupMmaStoreOp::verify() {
+ MemRefType dstMemrefType = getDstMemref().getType();
+ VectorType firstVtype = getMatrixD()
+ .front()
+ .getType()
+ .cast<WarpgroupAccumulatorType>()
+ .getFragmented();
+
+ int64_t totalFirstDimension = 0;
+ for (auto result : getMatrixD()) {
----------------
qcolombet wrote:
Nit: auto not clear here
https://github.com/llvm/llvm-project/pull/65441
More information about the Mlir-commits
mailing list