[AMD] CI - enable deepseekv3.2 on MI325-8gpu and merge perf/accuracy test suites into stage-b suites (#17633)

Co-authored-by: Bingxu Chen <Bingxu.Chen@amd.com>
This commit is contained in:
YC Tseng
2026-01-27 18:54:36 -08:00
committed by GitHub
co-authored by Bingxu Chen
parent 1c4616a034
commit 52bca42870
9 changed files with 91 additions and 233 deletions
@@ -14,6 +14,7 @@ from sglang.test.test_utils import (
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
DEFAULT_URL_FOR_TEST,
CustomTestCase,
is_in_amd_ci,
is_in_ci,
popen_launch_server,
write_github_step_summary,
@@ -71,7 +72,10 @@ class TestEvalAccuracyLarge(CustomTestCase):
f"### test_human_eval\n" f'{metrics["score"]=:.4f}\n'
)
self.assertGreater(metrics["score"], 0.64)
if is_in_amd_ci():
self.assertGreater(metrics["score"], 0.60)
else:
self.assertGreater(metrics["score"], 0.64)
def test_mgsm_en(self):
args = SimpleNamespace(
@@ -3,6 +3,7 @@ Usage:
python -m unittest test_moe_eval_accuracy_large.TestMoEEvalAccuracyLarge.test_mmlu
"""
import os
import unittest
from types import SimpleNamespace
@@ -14,6 +15,7 @@ from sglang.test.test_utils import (
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
DEFAULT_URL_FOR_TEST,
CustomTestCase,
is_in_amd_ci,
is_in_ci,
popen_launch_server,
write_github_step_summary,
@@ -28,6 +30,15 @@ class TestMoEEvalAccuracyLarge(CustomTestCase):
def setUpClass(cls):
cls.model = DEFAULT_MOE_MODEL_NAME_FOR_TEST
cls.base_url = DEFAULT_URL_FOR_TEST
# Disable AITER for AMD CI to ensure consistent results
env = None
if is_in_amd_ci():
env = os.environ.copy()
env["SGLANG_USE_AITER"] = "0"
env["SGLANG_USE_AITER_AR"] = "0"
env["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
cls.process = popen_launch_server(
cls.model,
cls.base_url,
@@ -38,6 +49,7 @@ class TestMoEEvalAccuracyLarge(CustomTestCase):
"--tp",
"2",
],
env=env,
)
@classmethod