use ROCM_AITER_FA on test

charlifu · charlifu · commit d6f3c7951421 · 2025-12-02T19:57:39.000Z
Signed-off-by: charlifu &lt;charlifu@amd.com&gt;
diff --git a/tests/v1/spec_decode/test_max_len.py b/tests/v1/spec_decode/test_max_len.py
@@ -47,7 +47,7 @@ def test_eagle_max_len(
                 "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")
 
         llm = LLM(

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ def test_eagle_max_len(`
`47`	`47`	`"multi-token eagle spec decode on current platform"`
`48`	`48`	`)`
`49`	`49`
`50`		`- if attn_backend == "FLASH_ATTN" and current_platform.is_rocm():`
	`50`	`+ if attn_backend == "ROCM_AITER_FA" and current_platform.is_rocm():`
`51`	`51`	`m.setenv("VLLM_ROCM_USE_AITER", "1")`
`52`	`52`
`53`	`53`	`llm = LLM(`