ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-02 04:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4fde1646e534415221edf81146d41f85fbf33e63
composable_kernel/example/ck_tile/01_fmha/codegen/ops
History
Yi DING 4fde1646e5 [CK_TILE] FMHA BWD Optimization For GFX950 (#2628) 
* simplify fmha_bwd_kernel MakeKargs & dq_dram_window

* simply duplicate

* trload pipeline

* Try two-stage

* add prefetch

* optimize & iglp
2025-08-12 11:11:55 +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
[CK_TILE] Low CU utilization optimization for fMHA fwd kernels (#2402)
2025-07-09 22:01:33 +08:00
fmha_bwd.py
[CK_TILE] FMHA BWD Optimization For GFX950 (#2628)
2025-08-12 11:11:55 +08:00
fmha_fwd_appendkv.py
[CK_TILE][FMHA] Uncomment all the headdim, use optdim to control (#2539)
2025-07-28 17:16:32 +08:00
fmha_fwd_splitkv.py
[CK_TILE] Fix FMHA qr_async causing errors in FA (#2627)
2025-08-06 20:04:23 +08:00
fmha_fwd.py
[CK_TILE] Fix FMHA qr_async causing errors in FA (#2627)
2025-08-06 20:04:23 +08:00
fmha_pagedkv_prefill.py
ck tile pagedkv prefill (#2405)
2025-07-07 16:16:54 +08:00