Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
charlifu wants to merge 10 commits into
base: main
Choose a base branch
from
+15 −8
Open

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

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
0074958
use ROCM_AITER_FA for the flash attn on rocm
charlifu Dec 2, 2025
e5f1f4a
Update tests/utils.py
charlifu Dec 2, 2025
d6f3c79
use ROCM_AITER_FA on test
charlifu Dec 2, 2025
d38904b
Merge branch 'main' into amd_ci/fix_test_max_len
charlifu Dec 3, 2025
4776f66
update test_spec_decode
charlifu Dec 3, 2025
e4d4d9e
update test_spec_decode
charlifu Dec 3, 2025
0f2b4b9
fix test_eagle.py
charlifu Dec 4, 2025
cb573c6
Merge branch 'main' into amd_ci/fix_test_max_len
charlifu Dec 5, 2025
a0b984c
fix test_basic_correctness
charlifu Dec 5, 2025
e420c26
Merge branch 'main' into amd_ci/fix_test_max_len
micah-wil Dec 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion tests/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1225,7 +1225,7 @@ 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")

Expand Down
Loading