ci: decouple stage and runner for cuda registry (#25197)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user