ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cd7ba6e2e8a61e40d7ba7b62368f28aaed4d7d79
composable_kernel/include
History
root cd7ba6e2e8 Add unified attention (42_unified_attention) 
Squashed from aghamari/unified-attention-decode-opt branch.

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.

Made-with: Cursor
2026-04-01 16:39:15 +00:00
..
ck
[rocm-libraries] ROCm/rocm-libraries#6022 (commit 54b284a)
2026-03-31 15:19:43 +00:00
ck_tile
Add unified attention (42_unified_attention)
2026-04-01 16:39:15 +00:00
rapidjson
…