[Mlir-commits] [mlir] [MLIR][NVGPU] Adding `nvgpu.warpgroup.mma` Op for Hopper GPUs (PR #65440)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Mon Sep 11 02:43:55 PDT 2023


================
@@ -192,6 +192,15 @@ def NVGPU_WarpgroupMatrixDescriptor : NVGPU_Type<"WarpgroupMatrixDescriptor", "w
   let assemblyFormat = "`<` struct(params) `>`";
 }
 
+def NVGPU_WarpgroupResult : NVGPU_Type<"WarpgroupResult", "warpgroup.result", []> {
+  let parameters = (ins "Type":$tensor);
+  let assemblyFormat = "`<` struct(params) `>`";
+  let description = [{
+    It is fragmented result matrix from `nvgpu.warpgroup.mma`.
----------------
qcolombet wrote:

What is "it" in this context?
Maybe something more descriptive:
"This type represents a partial result of the expansion of `nvgpu.warpgroup.mma`?"

Putting an example would be great: Pattern IN -> produced pattern.

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


More information about the Mlir-commits mailing list