[CI] Add batch invariant test to ci (#27842)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye
2025-11-21 11:20:33 -05:00
committed by GitHub
parent 711241c13c
commit 1f400c58b8
3 changed files with 16 additions and 1 deletions

View File

@@ -6,6 +6,7 @@ import random
import pytest
import torch
from vllm.attention.utils.fa_utils import flash_attn_supports_mla
from vllm.platforms import current_platform
skip_unsupported = pytest.mark.skipif(
@@ -18,7 +19,7 @@ BACKENDS: list[str] = [
"FLASHINFER",
]
if current_platform.is_cuda() and current_platform.is_device_capability(90):
if flash_attn_supports_mla():
BACKENDS.append("FLASH_ATTN_MLA")
DEFAULT_MODEL = "Qwen/Qwen3-1.7B"