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

[CI] Reorganize stage-b 1-GPU tests for 5090 compatibility (#16826)

Browse Source
This commit is contained in:
Alison Shao
2026-01-15 15:23:35 -08:00
committed by GitHub
parent 8b22deef5b
commit 146b5fcc84
136 changed files with 277 additions and 404 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/spec/eagle/test_eagle3_basic.py
-1
View File
@@ -12,7 +12,6 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=50, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=87, suite="stage-b-test-small-1-gpu-5090")
class TestEagle3Basic(EagleServerBase):
test/registered/spec/eagle/test_eagle_constrained_decoding.py
+1 -1
View File
@@ -14,7 +14,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=100, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=100, suite="stage-b-test-large-1-gpu")
class TestEagleConstrainedDecoding(
test/registered/spec/eagle/test_eagle_infer_a.py
+1 -1
View File
@@ -22,7 +22,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=561, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=561, suite="stage-b-test-large-1-gpu")
torch_dtype = torch.float16
prefill_tolerance = 5e-2
test/registered/spec/eagle/test_eagle_infer_b.py
+1 -1
View File
@@ -17,7 +17,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-small-1-gpu")
register_cuda_ci(est_time=1100, suite="stage-b-test-large-1-gpu")
class TestEAGLEServerBasic(EagleServerBase):
test/registered/spec/eagle/test_eagle_infer_beta.py
-1
View File
@@ -17,7 +17,6 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=283, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=404, suite="stage-b-test-small-1-gpu-5090")
class TestEagleServerBase(CustomTestCase, MatchedStopMixin):
test/registered/spec/test_ngram_speculative_decoding.py
+1 -1
View File
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=230, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=230, suite="stage-b-test-large-1-gpu")
GSM_DATASET_PATH = None
test/registered/spec/test_standalone_speculative_decoding.py
+1 -1
View File
@@ -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-small-1-gpu")
register_cuda_ci(est_time=308, suite="stage-b-test-large-1-gpu")
GSM_DATASET_PATH = None
test/registered/spec/utils/test_build_eagle_tree.py
-1
View File
@@ -10,7 +10,6 @@ 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=2, suite="stage-b-test-small-1-gpu-5090")
class TestBuildEagleTree(unittest.TestCase):