diff options
| author | Hsiangkai Wang <hsiangkai.wang@arm.com> | 2025-05-28 20:40:17 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-05-28 20:40:17 +0100 |
| commit | 8fb09c8d09bdbc1af98cb7138411029223731be8 (patch) | |
| tree | e9674999052cce46225405fd22fe20bc035c1aaf /llvm/lib/Transforms/Utils/LoopUtils.cpp | |
| parent | d8235affcf45fbea5ee4670ae2ac89c7f519b49c (diff) | |
| download | llvm-8fb09c8d09bdbc1af98cb7138411029223731be8.zip llvm-8fb09c8d09bdbc1af98cb7138411029223731be8.tar.gz llvm-8fb09c8d09bdbc1af98cb7138411029223731be8.tar.bz2 | |
[mlir][gpu] Add GPU subgroup MMA extract and insert operations (#139048)
- Introduced `gpu.subgroup_mma_extract` operation to extract values from
`!gpu.mma_matrix` by invocation and indices.
- Introduced `gpu.subgroup_mma_insert` operation to insert values into
`!gpu.mma_matrix` by invocation and indices.
- Updated the conversion patterns to SPIR-V for both extract and insert
operations.
- Added test cases to validate the new operations in the GPU to SPIR-V
conversion.
RFC:
https://discourse.llvm.org/t/rfc-add-gpu-operations-to-permute-data-in-2-loaded-mma-matrix/86148?u=hsiangkai
Diffstat (limited to 'llvm/lib/Transforms/Utils/LoopUtils.cpp')
0 files changed, 0 insertions, 0 deletions