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