ci: unify PR test suite naming (#21187)

This commit is contained in:
Lianmin Zheng
2026-03-23 00:18:45 -07:00
committed by GitHub
parent 3d312643b9
commit 814202704b
283 changed files with 554 additions and 554 deletions
@@ -12,8 +12,8 @@ from sglang.test.test_utils import (
DEFAULT_TARGET_MODEL_EAGLE3,
)
register_cuda_ci(est_time=50, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=50, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=50, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=50, suite="stage-b-test-1-gpu-small")
_is_hip = is_hip()
@@ -14,7 +14,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=100, suite="stage-b-test-large-1-gpu")
register_cuda_ci(est_time=100, suite="stage-b-test-1-gpu-large")
class TestEagleConstrainedDecoding(
@@ -22,7 +22,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=561, suite="stage-b-test-large-1-gpu")
register_cuda_ci(est_time=561, suite="stage-b-test-1-gpu-large")
torch_dtype = torch.float16
prefill_tolerance = 5e-2
@@ -22,7 +22,7 @@ from sglang.test.kits.radix_cache_server_kit import run_radix_attention_test
from sglang.test.server_fixtures.eagle_fixture import EagleServerBase
from sglang.test.test_utils import DEFAULT_TARGET_MODEL_EAGLE, run_logprob_check
register_cuda_ci(est_time=1100, suite="stage-b-test-large-1-gpu")
register_cuda_ci(est_time=1100, suite="stage-b-test-1-gpu-large")
class TestEAGLEServerBasic(EagleServerBase):
@@ -19,7 +19,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=283, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=283, suite="stage-b-test-1-gpu-small")
class TestEagleServerBase(CustomTestCase, MatchedStopMixin):
@@ -14,7 +14,7 @@ from sglang.test.test_utils import (
)
# Constrained decoding with EAGLE3 speculative reasoning (tp=2)
register_cuda_ci(est_time=60, suite="stage-b-test-large-2-gpu")
register_cuda_ci(est_time=60, suite="stage-b-test-2-gpu-large")
class ServerWithGrammar(CustomTestCase):
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=230, suite="stage-b-test-large-1-gpu")
register_cuda_ci(est_time=230, suite="stage-b-test-1-gpu-large")
GSM_DATASET_PATH = None
@@ -18,7 +18,7 @@ from sglang.test.test_utils import (
)
# Standalone speculative decoding tests (FA3, Triton, FlashInfer backends)
register_cuda_ci(est_time=308, suite="stage-b-test-large-1-gpu")
register_cuda_ci(est_time=308, suite="stage-b-test-1-gpu-large")
GSM_DATASET_PATH = None
@@ -9,8 +9,8 @@ from sglang.srt.speculative.eagle_utils import (
from sglang.srt.utils import get_device
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=3, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=3, suite="stage-b-test-small-1-gpu-amd")
register_cuda_ci(est_time=3, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=3, suite="stage-b-test-1-gpu-small-amd")
class TestBuildEagleTree(unittest.TestCase):
@@ -6,7 +6,7 @@ from sglang.srt.speculative.cpp_ngram.ngram_corpus import NgramCorpus
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=30, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=30, suite="stage-b-test-1-gpu-small")
def _make_corpus(match_type="BFS", **kwargs):