[AMD] Add 4-GPU test suite for MI325 runners (#20294)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user