Yi DING 8e1eb0c1ee [CK_TILE] FMHA BWD Decode Pipeline (#2643) ...

* Fix distr

* Duplicate block_fmha_bwd_dq_dk_dv_pipeline_trload_kr_ktr_vr

* decode 16x16 o2

2025-08-12 17:02:52 +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 Decode Pipeline (#2643)

2025-08-12 17:02:52 +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