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",