ci: decouple stage and runner for cuda registry (#25197)
This commit is contained in:
@@ -20,7 +20,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=74, suite="stage-b-test-1-gpu-small")
|
||||
register_cuda_ci(est_time=74, stage="stage-b", runner_config="1-gpu-small")
|
||||
register_amd_ci(est_time=32, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
_MODEL_NAME = "Qwen/Qwen3-0.6B"
|
||||
|
||||
@@ -18,7 +18,8 @@ from sglang.test.test_utils import (
|
||||
|
||||
register_cuda_ci(
|
||||
est_time=60,
|
||||
suite="stage-b-test-1-gpu-small",
|
||||
stage="stage-b",
|
||||
runner_config="1-gpu-small",
|
||||
)
|
||||
register_amd_ci(est_time=60, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
@@ -46,7 +46,7 @@ from sglang.test.test_utils import (
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# CI registration
|
||||
register_cuda_ci(est_time=113, suite="stage-b-test-1-gpu-small")
|
||||
register_cuda_ci(est_time=113, stage="stage-b", runner_config="1-gpu-small")
|
||||
|
||||
|
||||
# ============================================================================
|
||||
|
||||
@@ -33,7 +33,7 @@ from sglang.utils import wait_for_http_ready
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# CI registration - PD disaggregation requires 2 GPUs
|
||||
register_cuda_ci(est_time=65, suite="stage-b-test-2-gpu-large")
|
||||
register_cuda_ci(est_time=65, stage="stage-b", runner_config="2-gpu-large")
|
||||
|
||||
|
||||
class TestTraceDisaggregation(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user