From 4f1451679088e5832bce0965a254295c854aaa09 Mon Sep 17 00:00:00 2001
From: djramic <djoramic@amd.com>
Date: Mon, 21 Sep 2026 18:39:31 +0200
Subject: [PATCH] [ROCm][CI] Use ROCm backend for DeepSeek V4.1 ViT test
 (#57931)

Signed-off-by: Djordje Ramic <djoramic@amd.com>
Co-authored-by: Andreas Karatzas <akaratza@amd.com>
---
 tests/models/multimodal/generation/test_vit_cudagraph.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/models/multimodal/generation/test_vit_cudagraph.py b/tests/models/multimodal/generation/test_vit_cudagraph.py
index 9bda7ec85d..544623319f 100644
--- a/tests/models/multimodal/generation/test_vit_cudagraph.py
+++ b/tests/models/multimodal/generation/test_vit_cudagraph.py
@@ -236,7 +236,11 @@ MODEL_CONFIGS: dict[str, VitCudagraphTestConfig] = {
         },
         vllm_runner_kwargs={
             "load_format": "dummy",
-            "attention_backend": "FLASHMLA_SPARSE_DSV41",
+            "attention_backend": (
+                "ROCM_FLASHMLA_SPARSE_DSV4"
+                if current_platform.is_rocm()
+                else "FLASHMLA_SPARSE_DSV41"
+            ),
             "hf_overrides": partial(
                 dummy_hf_overrides,
                 model_arch="DeepseekV41ForCausalLM",