[all-commits] [llvm/llvm-project] 8fb09c: [mlir][gpu] Add GPU subgroup MMA extract and inser...

[Hsiangkai Wang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 8fb09c8d09bdbc1af98cb7138411029223731be8
      https://github.com/llvm/llvm-project/commit/8fb09c8d09bdbc1af98cb7138411029223731be8
  Author: Hsiangkai Wang <hsiangkai.wang at arm.com>
  Date:   2025-05-28 (Wed, 28 May 2025)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
    M mlir/lib/Conversion/GPUToSPIRV/WmmaOpsToSPIRV.cpp
    M mlir/test/Conversion/GPUToSPIRV/wmma-ops-to-spirv-khr-coop-matrix.mlir
    M mlir/test/Dialect/GPU/ops.mlir

  Log Message:
  -----------
  [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



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications