[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:31 PDT 2023


================
@@ -727,4 +727,24 @@ def NVGPU_WarpgroupMmaOp : NVGPU_Op<"warpgroup.mma"> {
   let hasVerifier = 1;
 }
 
+def NVGPU_WarpgroupMmaStoreOp : NVGPU_Op<"warpgroup.mma.store"> {
+  let description = [{
+    The `nvgpu.warpgroup.mma.store` op performs the store of fragmented result 
+    in $matrixD to give memref. 
----------------
qcolombet wrote:

Maybe I missed something, but I don't see the nit fixed here: "to the given memref"

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


More information about the cfe-commits mailing list