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
@@ -24,7 +24,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=87, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=87, stage="stage-b", runner_config="1-gpu-large")
def remove_prefix(text: str, prefix: str) -> str:
@@ -33,7 +33,8 @@ from sglang.test.test_utils import (
register_cuda_ci(
est_time=122,
suite="stage-b-test-1-gpu-large",
stage="stage-b",
runner_config="1-gpu-large",
)
NUM_TURNS = 150
@@ -21,7 +21,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=691, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=691, stage="stage-b", runner_config="1-gpu-large")
LOGPROB_PROMPTS = [
"The quick brown fox jumps over the lazy dog.",
@@ -24,7 +24,7 @@ from test_streaming_session import ( # noqa: E402
TestStreamingSessionAbortLeakRepro,
)
register_cuda_ci(est_time=519, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=519, stage="stage-b", runner_config="1-gpu-large")
SWA_MODEL = "openai/gpt-oss-20b"