DefTruth 82653f6622 feat: Add a unified merge_state API (#5428)

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

..

triton_ops

feat: Add a unified merge_state API (#5428)

2025-05-05 10:32:33 -07: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

FA3 speed up: skip len operation and get batch size directly from forward batch (#5969)

2025-05-02 00:26:12 -07: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

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

2025-05-05 10:32:17 -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

Add Cutlass MLA attention backend (#5390)

2025-04-27 20:58:53 -07:00

vision.py

Add an assertion to enhance the robustness of the operator (#5736)

2025-04-26 18:09:12 -07:00