minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Migrate 4-GPU/8-GPU workflow jobs to stage-c and add CI registry decorators (#17299)

Browse Source
This commit is contained in:
Alison Shao
2026-01-31 22:37:22 -08:00
committed by GitHub
parent 95180484e9
commit a0bae4c343
39 changed files with 189 additions and 284 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/srt/run_suite.py
+7 -60
View File
@@ -7,65 +7,13 @@ import tabulate
from sglang.test.ci.ci_utils import TestFile, run_unittest_files
# NOTE: please sort the test cases alphabetically by the test file name
# NOTE: per-commit-4-gpu, per-commit-8-gpu-h200, per-commit-8-gpu-h20, per-commit-4-gpu-b200,
# per-commit-4-gpu-gb200, per-commit-4-gpu-deepep, and per-commit-8-gpu-h200-deepep suites
# have been migrated to stage-c suites in test/registered/ using the CI registry system.
suites = {
"per-commit-4-gpu": [
TestFile("models/test_qwen3_next_models.py", 350),
TestFile("models/test_qwen3_next_models_mtp.py", 500),
TestFile("test_gpt_oss_4gpu.py", 300),
TestFile("test_multi_instance_release_memory_occupation.py", 64),
TestFile("test_pp_single_node.py", 500),
TestFile("test_epd_disaggregation.py", 150),
],
"per-commit-8-gpu-h200": [
TestFile("test_deepseek_v3_basic.py", 275),
TestFile("test_deepseek_v3_mtp.py", 275),
TestFile("test_disaggregation_hybrid_attention.py", 400),
TestFile("models/test_kimi_k2_models.py", 200),
TestFile("test_deepseek_v32_basic.py", 360),
TestFile("test_deepseek_v32_mtp.py", 360),
TestFile("models/test_mimo_models.py", 200),
],
"per-commit-8-gpu-h20": [
TestFile("quant/test_w4a8_deepseek_v3.py", 520),
TestFile("test_disaggregation_different_tp.py", 600),
TestFile("test_disaggregation_pp.py", 180),
TestFile("test_disaggregation_dp_attention.py", 155),
],
"per-commit-4-gpu-b200": [
TestFile("test_deepseek_v3_fp4_4gpu.py", 1500),
TestFile("test_fp8_blockwise_gemm.py", 280),
TestFile("test_gpt_oss_4gpu.py", 300),
TestFile("test_nvfp4_gemm.py", 360),
TestFile("test_deepseek_v32_fp4_4gpu.py", 600),
],
# "per-commit-8-gpu-b200": [
# TestFile("test_mistral_large3_basic.py", 275), # Moved to nightly - large model
# ],
"per-commit-4-gpu-gb200": [
TestFile("test_deepseek_v3_cutedsl_4gpu.py", 1800),
TestFile("test_disaggregation_aarch64.py", 300),
],
"per-commit-4-gpu-deepep": [
TestFile("ep/test_deepep_small.py", 531),
TestFile("ep/test_mooncake_ep_small.py", 660),
],
"per-commit-8-gpu-h200-deepep": [
TestFile("ep/test_deepep_large.py", 563),
],
# quantization_test suite migrated to test/registered/quant/
"__not_in_ci__": [
TestFile("test_release_memory_occupation.py", 200), # Temporarily disabled
TestFile("models/test_dummy_grok_models.py"),
TestFile("test_profile_v2.py"),
TestFile("models/test_ministral3_models.py"),
TestFile("test_mistral_large3_basic.py"),
TestFile("test_prefill_delayer.py"),
TestFile("test_fla_layernorm_guard.py"),
TestFile(
"models/test_qwen3_next_models_pcg.py"
), # Disabled: intermittent failures, see #17039
TestFile("ep/test_moriep_small.py"),
],
# All CUDA tests migrated to test/registered/
"__not_in_ci__": [],
}
# Add AMD tests
@@ -89,9 +37,8 @@ suite_amd = {
# TestFile("test_wave_attention_backend.py", 150), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/11127
# The time estimation for `test_int4fp8_moe.py` assumes `mistralai/Mixtral-8x7B-Instruct-v0.1` is already cached (running on 1xMI300X).
],
"per-commit-4-gpu-amd": [
TestFile("test_pp_single_node.py", 150),
],
# per-commit-4-gpu-amd migrated to test/registered/distributed/ using the CI registry system
"per-commit-4-gpu-amd": [],
# NOTE: AMD nightly suites (nightly-amd, nightly-amd-vlm, nightly-amd-8-gpu)
# have been migrated to test/registered/amd/nightly/ and are now managed
# by test/run_suite.py using the registry system.