[mlir][nvvm] Introduce `nvvm.stmatrix` Op (#69467)
commit5316d19ed54d897acc0d1a5627379571fb07f0ac
authorGuray Ozen <guray.ozen@gmail.com>
Thu, 19 Oct 2023 08:26:28 +0000 (19 10:26 +0200)
committerGitHub <noreply@github.com>
Thu, 19 Oct 2023 08:26:28 +0000 (19 10:26 +0200)
tree1f2a622d90088b83d51c826e47e025f3d4e01ebd
parent5341d5465dbf0b35c64c54f200af8389a8b76aef
[mlir][nvvm] Introduce `nvvm.stmatrix` Op (#69467)

This PR adds `nvvm.stmatrix` Op to NVVM dialect. The Op collectively
store one or more matrices across all threads in a warp to the given
address location in shared memory.
mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp
mlir/test/Conversion/NVVMToLLVM/nvvm-to-llvm.mlir