composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00

Files

Linjun-AMD ecda0fe2e9 [CK_TILE][FMHA] Add new tile size for async (#3586 )

* add new tile size for async

Signed-off-by: Linjun-AMD <Jun.Lin@amd.com>

* Update example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* fix lse error

Signed-off-by: Linjun-AMD <Jun.Lin@amd.com>

---------

Signed-off-by: Linjun-AMD <Jun.Lin@amd.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

[ROCm/composable_kernel commit: f3aafb9555]

2026-01-19 15:22:33 -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] Enable page size 16 for batch prefill kernel (#3568 )

2026-01-15 22:11:44 +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 )