Skip to content

GitLab

Unverified Commit 684f2545 authored by Matthew Bonanni's avatar Matthew Bonanni Committed by GitHub
Browse files

Prefer FlashAttention MLA as default over FlashMLA (#27363) 


Signed-off-by: default avatarMatthew Bonanni <mbonanni@redhat.com>
parent e5534249
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/platforms/cuda.py
View file @ 684f2545
......@@ -55,15 +55,15 @@ def _get_backend_priorities(
return [
AttentionBackendEnum.CUTLASS_MLA,
AttentionBackendEnum.FLASHINFER_MLA,
AttentionBackendEnum.FLASHMLA,
AttentionBackendEnum.FLASH_ATTN_MLA,
AttentionBackendEnum.FLASHMLA,
AttentionBackendEnum.TRITON_MLA,
AttentionBackendEnum.FLASHMLA_SPARSE,
]
else:
return [
AttentionBackendEnum.FLASHMLA,
AttentionBackendEnum.FLASH_ATTN_MLA,
AttentionBackendEnum.FLASHMLA,
AttentionBackendEnum.FLASHINFER_MLA,
AttentionBackendEnum.TRITON_MLA,
AttentionBackendEnum.FLASHMLA_SPARSE,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment