[AMD] Add 4-GPU test suite for MI325 runners (#20294)

This commit is contained in:
Michael
2026-03-24 14:04:49 -07:00
committed by GitHub
parent c4db64c16b
commit 6cb1c2d53d
12 changed files with 252 additions and 13 deletions
@@ -28,7 +28,7 @@ pytestmark = pytest.mark.filterwarnings(
)
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import (
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
DEFAULT_URL_FOR_TEST,
@@ -36,6 +36,12 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=300, suite="nightly-4-gpu", nightly=True)
register_amd_ci(
est_time=300,
suite="nightly-amd-4-gpu",
nightly=True,
disabled="TP=2 vs TP=4 numerical mismatch on AMD (comparator fails tolerance check)",
)
MODEL = "Qwen/Qwen3-30B-A3B"
BASELINE_TP = 2