ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-19 20:40:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
3,219 Commits 748 Branches 38 Tags
b686143624f2d7154c3f5fb235df317100b22e45
Commit Graph

3 Commits

Author SHA1 Message Date
Damien Lejeune
b686143624 Adding SWA decode dispatcher to support GPT-OSS shape + update smoke test 2026-05-08 14:38:16 +00:00
Damien Lejeune
5afd97ff5b Adding SWA implementation + instances 2026-05-08 08:52:25 +00:00
root
4c5e290378 Add unified attention (42_unified_attention) and topk_softmax_decode 
Squashed from aghamari/unified-attention-decode-opt branch.

42_unified_attention: CK tile paged-KV attention kernel optimized for
decode with 4-tier dispatch (tiny/small/medium/large), 16x16 MFMA,
2D decode grid, head-group merging. Supports hdim=64 GQA-8 and
hdim=128 MHA with block_size=32.

topk_softmax_decode: fused topk + softmax kernel for M=1 MoE decode.

Made-with: Cursor
 2026-04-01 16:24:04 +00:00