Lianmin Zheng fba8eccd7e Log if cuda graph is used & extend cuda graph capture to cuda-graph-max-bs (#6201) ...

Co-authored-by: SangBin Cho <rkooo567@gmail.com>

2025-05-12 00:17:33 -07:00

..

triton_ops

Add typo checker in pre-commit (#6179)

2025-05-11 12:55:00 +08:00

base_attn_backend.py

Remove q concat in FA3 backend for DeepSeek decode (#5638)

2025-04-22 11:43:12 -07:00

cutlass_mla_backend.py

Cutlass MLA decode - fix dtype error (#5868)

2025-04-28 21:12:58 -07:00

double_sparsity_backend.py

Misc clean up; Remove the support of jump forward (#4032)

2025-03-03 07:02:14 -08:00

flashattention_backend.py

Add typo checker in pre-commit (#6179)

2025-05-11 12:55:00 +08:00

flashinfer_backend.py

Fix "Avoid computing lse in Ragged Prefill when there's no prefix match" (#5555)

2025-05-05 10:32:17 -07:00

flashinfer_mla_backend.py

opt flashinfer mla cat (#5822)

2025-05-08 23:17:14 -07:00

flashmla_backend.py

[Fix] Fix a bug for flashmla to run R1 model (#5875)

2025-04-29 01:03:13 -07:00

merge_state.py

feat: Add a unified merge_state API (#5428)

2025-05-05 10:32:33 -07:00

torch_native_backend.py

Feat/support encoder model (like bert) (#4887)

2025-04-17 01:50:48 -07:00

triton_backend.py

Feat/support encoder model (like bert) (#4887)

2025-04-17 01:50:48 -07:00

utils.py

Log if cuda graph is used & extend cuda graph capture to cuda-graph-max-bs (#6201)

2025-05-12 00:17:33 -07:00

vision.py

[Feature] Add FlashAttention3 as a backend for VisionAttention (#5764)

2025-05-08 10:01:19 -07:00