tree db410ff9a1f7c569df7948f16db9417738e3bafa
parent ec7528cee889b6fe0c369420aff1094c333fd2b5
author Stanley Winata <68087699+raikonenfnu@users.noreply.github.com> 1730745053 -0800
committer GitHub <noreply@github.com> 1730745053 -0800
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJnKRLdCRC1aQ7uu5UhlAAA+AMQAEgBFH5GfYWLTq5yQNZZlwcY
 E8OzZOo/WlwXCFYKQA2umvkiTKJUoCVUrWJb5Q5p4gRa+PplhpV2wrcuErA0rk7D
 qA23JWIbyB+gTs0eFPCpVx5VtCz1gj06doTeSCW0No8HbToLqjZbHhxdGc6jgq9V
 IvS1/PV++qNDG0LBnMbTJJH3eb+Tl5BV0Xoig6QmozuoUaJB1pqlCDJci9hPxtMO
 TCjQH7R/vTbxrtgp2WQiJWPntA8X7OcuNAtOKOBxohcUB1kh2aU0ISEp9Z8n0bgH
 xsaTmBCr7DiY4aVW0WIXbPN99EkuhOz3Vs4gYKJoNygU5oCOEgOCk/ip38XMLWys
 U9/sfsUPPE+V1QlTx5zJj+A2H7uMiTevKVAY6+FjB0B0iSThIErEOs1J90aTG2rj
 PyeNGi31s3k2D3ds/TwGHst6eoW8cODn0AppL/VWlzcfZ8Oz1nxcYBx5LMo1XZFv
 fh18reF9HIB+OL4ZUS0ljfLfjySAqq/sIKC1A09ZC3FooNg23hN9y+AeCazo2fwe
 +yLzr3Ymqkq65PymQsYt/vh4lSzOFNYII18Hegjg1i5BUT3F9PrYh7XTLV0qN7Fn
 wOUaerL4htTtEZh3Jr4qd0QQmOCdLBouQr2Jv3nlrM3sAiNoYSqxIpF0rEGnDTBJ
 FP2o4dHJDZPyokS3oV28
 =MnSq
 -----END PGP SIGNATURE-----
 

 [LLVMGPU] Teach KernelConfig to set MMA schedules per op in LoweringConfig (#18984)

The main motivation for this change is to enable different
intrinsics/layout on different ops inside the same function/dispatch,
especially for attention. To that extend, we move the scheduling MMA
information such as mma_intrinsic, subgroup_m_count, and
subgroup_n_count, from the translation info attached to the function
onto the lowering_config per op. Here is a quick summary of things we
needed to do to achieve that:

1. Introduce setMmaKind, set/get subgroupMCount, and set/get
subgroupMCount on IREE::GPU::LoweringConfigAttr
2. Move configuring of QK matmul's schedule into KernelConfig from
LLVMGPUConfigureTensorLayout.
3. Now that qk and pv may have different intrinsic, update information
used to decide transposeIntrinsic and reuseIntrinsic in
LLVMGPUConfigureTensorLayout
4. Update a bunch of tests to use lowering config to configure MMAs now.

---------

Signed-off-by: Stanley Winata <stanley.winata@amd.com>