[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
+6 -2
View File
@@ -11,12 +11,13 @@ from sglang.test.send_one import BenchArgs, send_one_prompt
from sglang.test.test_utils import (
DEFAULT_URL_FOR_TEST,
CustomTestCase,
is_in_amd_ci,
is_in_ci,
popen_launch_server,
write_github_step_summary,
)
register_amd_ci(est_time=3600, suite="stage-c-test-large-8-gpu-amd-mi35x")
register_amd_ci(est_time=3600, suite="stage-c-test-large-8-gpu-amd")
KIMI_K2_MODEL_PATH = "moonshotai/Kimi-K2-Instruct-0905"
SERVER_LAUNCH_TIMEOUT = 3600
@@ -88,7 +89,10 @@ class TestKimiK2Instruct0905(CustomTestCase):
f"### test_bs_1_speed (Kimi-K2-Instruct-0905)\n"
f"{speed=:.2f} token/s\n"
)
self.assertGreater(speed, 45)
if is_in_amd_ci():
self.assertGreater(speed, 30)
else:
self.assertGreater(speed, 45)
if __name__ == "__main__":