ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-01 20:21:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ead4447b20b597703e1e66a49801087aae92608d
composable_kernel/example/ck_tile/01_fmha/codegen/ops
History
Yi DING ead4447b20 [CK_TILE] FMHA BWD Enable Tile 16x192 (#2741) 
* 16x192

* Use buffer_load_lds for lse/d

* Dispatch & cleanup

* Avoid zeroing dq & fix

* fix
2025-08-28 18:54:18 +08:00
..
__init__.py
[CK_TILE] fmha forward split-kv + combine kernels (#1338)
2024-06-26 17:41:15 +08:00
fmha_batch_prefill.py
Support Wave32 in CK_TILE - Part 1 (#2594)
2025-08-18 10:08:31 -07:00
fmha_bwd.py
[CK_TILE] FMHA BWD Enable Tile 16x192 (#2741)
2025-08-28 18:54:18 +08:00
fmha_fwd_appendkv.py
Support Wave32 in CK_TILE - Part 1 (#2594)
2025-08-18 10:08:31 -07:00
fmha_fwd_splitkv.py
Support Wave32 in CK_TILE - Part 1 (#2594)
2025-08-18 10:08:31 -07:00
fmha_fwd.py
[CK_TILE] Update the fmhafwd dispatch logic (#2698)
2025-08-20 16:24:43 +08:00
fmha_pagedkv_prefill.py
Support Wave32 in CK_TILE - Part 1 (#2594)
2025-08-18 10:08:31 -07:00