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
+1 -1
View File
@@ -16,7 +16,7 @@ from sglang.test.test_deterministic_utils import (
)
from sglang.test.test_utils import is_in_amd_ci
register_cuda_ci(est_time=207, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=207, stage="stage-b", runner_config="1-gpu-large")
register_amd_ci(est_time=278, suite="stage-b-test-1-gpu-small-amd")
@@ -20,7 +20,7 @@ from sglang.test.test_utils import (
CustomTestCase,
)
register_cuda_ci(est_time=77, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=77, stage="stage-b", runner_config="1-gpu-small")
class TestEngineChildPids(CustomTestCase):
@@ -23,7 +23,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=107, suite="stage-b-test-2-gpu-large")
register_cuda_ci(est_time=107, stage="stage-b", runner_config="2-gpu-large")
PROMPT = (
+1 -1
View File
@@ -5,7 +5,7 @@ import torch
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.gpt_oss_common import BaseTestGptOss
register_cuda_ci(est_time=345, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=345, stage="stage-b", runner_config="1-gpu-small")
@unittest.skipIf(not torch.cuda.is_available(), "CUDA is not available")
+1 -1
View File
@@ -8,7 +8,7 @@ from sglang.srt.utils import get_device, is_hip
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import DEFAULT_SMALL_MODEL_NAME_FOR_TEST, CustomTestCase
register_cuda_ci(est_time=45, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=45, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=55, suite="stage-b-test-1-gpu-small-amd")
_is_hip = is_hip()
@@ -4,7 +4,7 @@ from unittest.mock import MagicMock, patch
from sglang.srt.server_args import ServerArgs
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=9, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=9, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=1, suite="stage-b-test-1-gpu-small-amd")
@@ -17,7 +17,7 @@ from sglang.test.test_utils import (
send_generate_requests,
)
register_cuda_ci(est_time=53, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=53, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=70, suite="stage-b-test-1-gpu-small-amd")
+1 -1
View File
@@ -28,7 +28,7 @@ from sglang.test.test_utils import (
run_logprob_check,
)
register_cuda_ci(est_time=134, suite="stage-b-test-1-gpu-small")
register_cuda_ci(est_time=134, stage="stage-b", runner_config="1-gpu-small")
register_amd_ci(est_time=130, suite="stage-b-test-1-gpu-small-amd")
+1 -1
View File
@@ -22,7 +22,7 @@ from sglang.test.test_utils import (
CustomTestCase,
)
register_cuda_ci(est_time=387, suite="stage-b-test-1-gpu-large")
register_cuda_ci(est_time=387, stage="stage-b", runner_config="1-gpu-large")
register_amd_ci(est_time=261, suite="stage-b-test-1-gpu-small-amd")