[clang] [MLIR][NVGPU] Introduce `nvgpu.wargroup.mma.store` Op for Hopper GPUs (PR #65441)
via cfe-commits
cfe-commits at lists.llvm.org
Wed Oct 4 07:20:32 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:
We still have the `auto` keyword here. I'd like to see the explicit type.
https://github.com/llvm/llvm-project/pull/65441
More information about the cfe-commits
mailing list