update test_spec_decode

charlifu · charlifu · commit 87c1f4d39d85 · 2025-12-03T17:08:35.000Z
Signed-off-by: charlifu &lt;charlifu@amd.com&gt;
diff --git a/tests/v1/e2e/test_spec_decode.py b/tests/v1/e2e/test_spec_decode.py
@@ -413,7 +413,7 @@ def test_eagle_correctness(
                 "multi-token eagle spec decode on current platform"
             )
 
-        if attn_backend == "FLASH_ATTN" and current_platform.is_rocm():
+        if attn_backend == "ROCM_AITER_FA" and current_platform.is_rocm():
             m.setenv("VLLM_ROCM_USE_AITER", "1")
 
         method, model_name, spec_model_name, tp_size = model_setup

Original file line number	Diff line number	Diff line change
`@@ -413,7 +413,7 @@ def test_eagle_correctness(`
`413`	`413`	`"multi-token eagle spec decode on current platform"`
`414`	`414`	`)`
`415`	`415`
`416`		`- if attn_backend == "FLASH_ATTN" and current_platform.is_rocm():`
	`416`	`+ if attn_backend == "ROCM_AITER_FA" and current_platform.is_rocm():`
`417`	`417`	`m.setenv("VLLM_ROCM_USE_AITER", "1")`
`418`	`418`
`419`	`419`	`method, model_name, spec_model_name, tp_size = model_setup`