[CI] Key scheduled CUDA suites by runner_config instead of hand-written jobs (#34186)
This commit is contained in:
@@ -21,7 +21,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=40, suite="nightly-1-gpu", nightly=True)
|
||||
register_cuda_ci(est_time=70, stage="nightly", runner_config="1-gpu-large")
|
||||
register_amd_ci(est_time=40, suite="nightly-amd-1-gpu", nightly=True)
|
||||
register_cpu_ci(est_time=225, suite="base-c-test-cpu")
|
||||
|
||||
|
||||
@@ -50,7 +50,7 @@ from sglang.test.test_utils import (
|
||||
run_distributed_test,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=30, suite="nightly-2-gpu", nightly=True)
|
||||
register_cuda_ci(est_time=30, stage="nightly", runner_config="2-gpu-large")
|
||||
register_amd_ci(est_time=60, suite="nightly-amd", nightly=True)
|
||||
|
||||
|
||||
|
||||
@@ -35,7 +35,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=300, suite="nightly-4-gpu", nightly=True)
|
||||
register_cuda_ci(est_time=290, stage="nightly", runner_config="4-gpu-h100")
|
||||
register_amd_ci(
|
||||
est_time=300,
|
||||
suite="nightly-amd-4-gpu",
|
||||
|
||||
@@ -9,6 +9,8 @@ Env knobs:
|
||||
SGLANG_PRECISION_COMMIT override sglang sha (7-40 hex) tagged on push
|
||||
SGLANG_PRECISION_HF_REPO required HF dataset repo for cross-runner
|
||||
baseline storage; see precision_baseline_store
|
||||
SGLANG_PRECISION_HF_TOKEN write token for that repo (not HF_TOKEN, which
|
||||
carries the runner's gated-model read token)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
@@ -47,7 +49,7 @@ try:
|
||||
except Exception: # pragma: no cover
|
||||
_hfs = None
|
||||
|
||||
register_cuda_ci(est_time=3600, suite="nightly-precision-8-gpu-h200", nightly=True)
|
||||
register_cuda_ci(est_time=3600, stage="nightly", runner_config="8-gpu-h200")
|
||||
|
||||
DEFAULT_MODELS_FOR_NIGHTLY_PRECISION = "zai-org/GLM-5.2-FP8"
|
||||
DEFAULT_DIFF_THRESHOLD = 1e-3
|
||||
|
||||
@@ -17,7 +17,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True)
|
||||
register_cuda_ci(est_time=240, stage="nightly", runner_config="1-gpu-large")
|
||||
register_amd_ci(est_time=120, suite="nightly-amd-1-gpu", nightly=True)
|
||||
register_cpu_ci(est_time=622, suite="base-c-test-cpu")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user