Skip to content

Commit 4b7817c

Browse files
authored
[Misc] Add missing _Backend enums (#19081)
Signed-off-by: nicklucche <nlucches@redhat.com>
1 parent d00dd65 commit 4b7817c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vllm/platforms/interface.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,8 @@ class _Backend(enum.Enum):
4747
TORCH_SDPA = enum.auto()
4848
FLASHINFER = enum.auto()
4949
TRITON_MLA = enum.auto() # Supported by V1
50+
TRITON_MLA_VLLM_V1 = enum.auto()
51+
FLASHMLA_VLLM_V1 = enum.auto()
5052
FLASHMLA = enum.auto() # Supported by V1
5153
HPU_ATTN = enum.auto()
5254
PALLAS = enum.auto()

0 commit comments

Comments
 (0)