main

文件	最后提交记录	最后更新时间
invalid.mlir	[MLIR][NVVM] Explicit Data Type for Output in wgmma.mma_async (#78713) The current implementation of nvvm.wgmma.mma_async Op deduces the data type of the output matrix from the data type of struct member, which can be non-intuitive, especially in cases where types like 2xf16 are packed into i32. This PR addresses this issue by improving the Op to include an explicit data type for the output matrix. The modified Op now includes an explicit data type for Matrix-D (<f16>), and looks as follows: %result = llvm.mlir.undef : !llvm.struct<(struct<(i32, i32, ... nvvm.wgmma.mma_async %descA, %descB, %result, #nvvm.shape<m = 64, n = 32, k = 16>, D [<f16>, #nvvm.wgmma_scale_out<zero>], A [<f16>, #nvvm.wgmma_scale_in<neg>, <col>], B [<f16>, #nvvm.wgmma_scale_in<neg>, <col>]	2 年前
nvvm-to-llvm.mlir	[MLIR][NVVM] Update mbarrier Ops to use AnyTypeOf[] (3/3) (#167567) This is a follow-up of PR #165558 and #165993. This patch updates the remaining two Ops to use the AnyTypeOf[] construct, completing the migration for the mbarrier family of Ops. mbarrier.arrive.expect_tx mbarrier.try_wait.parity Signed-off-by: Durgadoss R <durgadossr@nvidia.com>	8 个月前