biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[ROCm][CI] Fix test_max_len.py for Rocm (#29916)

Browse Source 
Signed-off-by: charlifu <charlifu@amd.com>
Signed-off-by: Charlie Fu <Charlie.Fu@amd.com>
This commit is contained in:
Charlie Fu
2025-12-08 15:58:30 -06:00
committed by GitHub
parent ae0f69b16a
commit 6af70e11a0
5 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utils.py
View File

@@ -1225,9 +1225,9 @@ def get_attn_backend_list_based_on_platform() -> list[str]:
try:
import aiter # noqa: F401
attn_backend_list.append("FLASH_ATTN")
attn_backend_list.append("ROCM_AITER_FA")
except Exception:
print("Skip FLASH_ATTN on ROCm as aiter is not installed")
print("Skip ROCM_AITER_FA on ROCm as aiter is not installed")
return attn_backend_list
elif current_platform.is_xpu():