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
@@ -19,7 +19,7 @@ from sglang.srt.configs.model_config import ModelConfig
from sglang.srt.model_loader.loader import ModelOptModelLoader
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=11, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=11, stage="stage-b", runner_config="1-gpu-large")
register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
# Note: PYTHONPATH=python should be set when running tests
@@ -30,7 +30,7 @@ CALIBRATION_BATCH_SIZE = 36
CALIBRATION_NUM_SAMPLES = 512
DEFAULT_DEVICE = "cuda:0"
register_cuda_ci(est_time=11, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=11, stage="stage-b", runner_config="1-gpu-small")
class TestModelOptModelLoader(CustomTestCase):