main

GGitHub[MLIR][NVVM] Update TMA Load Op (#156347)

fa366b4e创建于 2025年9月23日历史提交

文件	最后提交记录	最后更新时间
python	[MLIR][NVGPU] Fix nvgpu_arrive syntax in matmulBuilder.py (#113713) This patch updates the syntax for nvgpu_arrive Op in matmulBuilder.py. This fixes the compilation error for this test. For the warp-specialized matmul_kernel implementation, removing the WaitGroupSyncOp (after the mma-main-loop) fixes the hang observed. With these two fixes, the test compiles and executes successfully on an sm90a machine. Signed-off-by: Durgadoss R <durgadossr@nvidia.com>	1 年前
transform-dialect	[MLIR][NVVM] Update TMA Load Op (#156347) This patch includes im2col and gather mode support for the TMA Load Op. The lowering is also updated to intrinsics except when a Predicate is given. This completes the Blackwell additions on this Op. * NVVM Dialect has support for Shared::Cluster address-space now. So, this patch also updates the Op to use AS(7) instead of AS(3). The corresponding inline-ptx based unit tests are also updated. * lit tests are added for all combinations. Signed-off-by: Durgadoss R <durgadossr@nvidia.com>	10 个月前
cga_cluster.mlir	[mlir] Rename mlir-cpu-runner to mlir-runner (#123776) With the removal of mlir-vulkan-runner (as part of #73457) in e7e3c45bc70904e24e2b3221ac8521e67eb84668, mlir-cpu-runner is now the only runner for all CPU and GPU targets, and the "cpu" name has been misleading for some time already. This commit renames it to mlir-runner.	1 年前
gemm_f32_f16_f16_128x128x128.mlir	[mlir][nvgpu] Fix nvgpu integration test (#154748) Fix nvgpu mlir file integration test. This PR fixes the bug by removing memref.get_global and then using memref.view.	11 个月前
gemm_pred_f32_f16_f16_128x128x128.mlir	[mlir][nvgpu] Fix nvgpu integration test (#154748) Fix nvgpu mlir file integration test. This PR fixes the bug by removing memref.get_global and then using memref.view.	11 个月前
lit.local.cfg	[mlir] Nvidia Hopper TMA load integration test This work introduces sm90 integration testing and adds a single test. Depends on : D155825 D155680 D155563 D155453 Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D155838	2 年前
tma_load_128x128_stride_noswizzle.mlir	[mlir] Rename mlir-cpu-runner to mlir-runner (#123776) With the removal of mlir-vulkan-runner (as part of #73457) in e7e3c45bc70904e24e2b3221ac8521e67eb84668, mlir-cpu-runner is now the only runner for all CPU and GPU targets, and the "cpu" name has been misleading for some time already. This commit renames it to mlir-runner.	1 年前
tma_load_128x64_swizzle128b.mlir	[mlir] Rename mlir-cpu-runner to mlir-runner (#123776) With the removal of mlir-vulkan-runner (as part of #73457) in e7e3c45bc70904e24e2b3221ac8521e67eb84668, mlir-cpu-runner is now the only runner for all CPU and GPU targets, and the "cpu" name has been misleading for some time already. This commit renames it to mlir-runner.	1 年前
tma_load_64x64_swizzle128b.mlir	[MLIR][NVGPU] Use gpu.dynamic_shared_memory in tests (#133122) Reland #133051	1 年前
tma_load_64x8_8x128_noswizzle.mlir	[mlir] Rename mlir-cpu-runner to mlir-runner (#123776) With the removal of mlir-vulkan-runner (as part of #73457) in e7e3c45bc70904e24e2b3221ac8521e67eb84668, mlir-cpu-runner is now the only runner for all CPU and GPU targets, and the "cpu" name has been misleading for some time already. This commit renames it to mlir-runner.	1 年前