[ROCm][CI] Prep Tests For Change To ROCM_ATTN As New Default Backend On ROCm (#36025)

Signed-off-by: Micah Williamson <micah.williamson@amd.com>
This commit is contained in:
Micah Williamson
2026-03-09 13:27:55 -05:00
committed by GitHub
parent 3fd03f1ec2
commit 4ff9b045fe
10 changed files with 32 additions and 10 deletions

View File

@@ -173,6 +173,9 @@ VLM_TEST_SETTINGS = {
marks=[
pytest.mark.core_model,
],
vllm_runner_kwargs={"attention_backend": "TRITON_ATTN"}
if current_platform.is_rocm()
else {},
),
"ultravox": VLMTestInfo(
models=["fixie-ai/ultravox-v0_5-llama-3_2-1b"],