diff --git a/vllm/platforms/interface.py b/vllm/platforms/interface.py index 5c4f7a2f7dc7..c7a627262357 100644 --- a/vllm/platforms/interface.py +++ b/vllm/platforms/interface.py @@ -47,6 +47,8 @@ class _Backend(enum.Enum): TORCH_SDPA = enum.auto() FLASHINFER = enum.auto() TRITON_MLA = enum.auto() # Supported by V1 + TRITON_MLA_VLLM_V1 = enum.auto() + FLASHMLA_VLLM_V1 = enum.auto() FLASHMLA = enum.auto() # Supported by V1 HPU_ATTN = enum.auto() PALLAS = enum.auto()