ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0d13ef7329e0d4e9f3104fe3edd3b968a4dfd47b
composable_kernel/example/ck_tile/01_fmha/codegen
History
Jeff Huang 99b88be5fb [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.

[ROCm/composable_kernel commit: c9f112b026]
2026-01-13 12:04:43 +08:00
..
ops
[FMHA] Support page_size=1 (linear layout) in batch prefill pipeline (#3545)
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
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
[CK_TILE][FMHA] Integrate FAv2 & FAv3 (WIP) in the single fmha_fwd() API (#3153)
2025-12-05 10:31:12 +08:00
utils.py
chore(copyright): update copyright header for example directory (#3273)
2025-11-24 18:02:41 -08:00