ci: decouple stage and runner for cuda registry (#25197)

This commit is contained in:
Liangsheng Yin
2026-05-13 17:28:21 -07:00
committed by GitHub
parent dd98153d17
commit 22d3f3996c
261 changed files with 388 additions and 293 deletions
@@ -14,7 +14,7 @@ from sglang.test.test_utils import (
)
# RadixAttention server integration tests
register_cuda_ci(est_time=100, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=100, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=100, suite="stage-b-test-1-gpu-small-amd")
@@ -11,7 +11,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=55, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=55, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=55, suite="stage-b-test-1-gpu-small-amd")
MODEL = DEFAULT_SMALL_MODEL_NAME_FOR_TEST
@@ -6,7 +6,7 @@ from sglang.test.server_fixtures.default_fixture import DefaultServerBase
MODEL = "openai/gpt-oss-20b"
register_cuda_ci(est_time=151, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=151, stage="stage-b", runner_config="1-gpu-large")
class TestSWARadixCacheKL(KLDivergenceMixin, DefaultServerBase):
@@ -35,7 +35,7 @@ MAMBA_TRACK_INTERVAL = 128
SWA_MODEL = "openai/gpt-oss-20b"
FULL_MODEL = "Qwen/Qwen3-32B"
register_cuda_ci(est_time=760, suite="stage-c-test-4-gpu-h100")
register_cuda_ci(est_time=760, stage="stage-c", runner_config="4-gpu-h100")
class UnifiedRadixTreeTestMixin: