zanzhang faf9be7de6 add chunked prefill into mha varlen

2025-05-20 22:00:35 +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

Use only qr_async pipeline for batch_prefill (#2195)

2025-05-15 11:47:29 -07:00

fmha_bwd.py

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163)

2025-05-13 12:19:25 +08:00

fmha_fwd_appendkv.py

Only generate specific hdim (#2120)

2025-04-24 18:52:58 +08:00

fmha_fwd_splitkv.py

Merge remote-tracking branch 'origin/fix_fa_codegen' into Fa_seqlen1_pass

2025-05-16 14:23:16 +08:00

fmha_fwd.py

add chunked prefill into mha varlen

2025-05-20 22:00:35 +08:00