[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>
2026-09-21 16:39:31 +00:00
parent 7853700d4c
commit 4f14516790
1 changed files with 5 additions and 1 deletions
@@ -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",