ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 19:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
62e8f73545c49a3882d25e709d4a63fcdf5fd41e
composable_kernel/example/ck_tile/01_fmha/codegen
History
root 63821af1ff Add split-KV decode tiles (b16x32, b32x32) + fix num_splits heuristic 
Decode tiles for split-KV hdim=64: bm0=16/1-warp and bm0=32/2-warp.
Fix num_splits to use num_heads_kv (not num_heads_q) and target 4x SMs.

Performance unchanged (0.056ms) because:
1. Split+combine overhead dominates for short KV (31 pages)
2. Triton 3D's single-kernel split avoids combine kernel entirely

Made-with: Cursor
2026-04-01 18:49:16 +00:00
..
ops
Add split-KV decode tiles (b16x32, b32x32) + fix num_splits heuristic
2026-04-01 18:49:16 +00:00
__init__.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
arch.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
cmake_config.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
cpp_symbol_map.py
[rocm-libraries] ROCm/rocm-libraries#4368 (commit 17f7dfc)
2026-03-11 10:00:52 +00:00
utils.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00