[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>
djramic committed
Sep 21, 2026 at 18:39 UTC
4f1451679088e5832bce0965a254295c854aaa09
1 file changed
+5
-1
tests/models/multimodal/generation/test_vit_cudagraph.py
+5
-1
@@ -236,7 +236,11 @@ MODEL_CONFIGS: dict[str, VitCudagraphTestConfig] = {
236
},
237
vllm_runner_kwargs={
238
"load_format": "dummy",
239
- "attention_backend": "FLASHMLA_SPARSE_DSV41",
239
+ "attention_backend": (
240
+ "ROCM_FLASHMLA_SPARSE_DSV4"
241
+ if current_platform.is_rocm()
242
+ else "FLASHMLA_SPARSE_DSV41"
243
+ ),
244
"hf_overrides": partial(
245
dummy_hf_overrides,
246
model_arch="DeepseekV41ForCausalLM",