ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
16ca5cb53237724f84c8a431938d8ff3faa4bb49
composable_kernel/example/ck_tile/01_fmha/codegen/ops
History
Jeff Huang c9f112b026 [FMHA] Support page_size=1 (linear layout) in batch prefill pipeline (#3545) 
- Enable page_size=1 support in batch prefill codegen (linear layout only).
- Implement per-token page lookup in `kv_offset_array_transform` for page_size=1 to handle 3D input tensors correctly.
- Relax `kPageBlockSize` alignment assertion for the page_size=1 case.
2026-01-13 12:04:43 +08:00
..
__init__.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
fmha_batch_prefill.py
[FMHA] Support page_size=1 (linear layout) in batch prefill pipeline (#3545)
2026-01-13 12:04:43 +08:00
fmha_bwd.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
fmha_fwd_appendkv.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00
fmha_fwd_splitkv.py
Add attention sink support for FMHA FWD (#3368)
2025-12-15 12:21:59 +08:00
fmha_fwd.py
[CK_TILE][FMHA] Add FP8 support for batch_prefill kernel (#3425)
2025-12-24 10:34:06 +08:00
fmha_pagedkv_prefill.py
Add attention sink support for FMHA FWD (#3368)
2025-12-15 12:21:59 +08:00