[CI] reduce CPU CI scope with base-c suite (#28623)

This commit is contained in:
jundu
2026-06-24 16:18:06 +08:00
committed by GitHub
parent 4da23fdc35
commit e04ed05193
91 changed files with 96 additions and 91 deletions
@@ -36,7 +36,7 @@ from sglang.benchmark.datasets.sharegpt import sample_sharegpt_requests
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=40, suite="base-a-test-cpu") register_cpu_ci(est_time=40, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class _DummyTokenTensor: class _DummyTokenTensor:
@@ -34,7 +34,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestExecuteSubPlans: class TestExecuteSubPlans:
@@ -26,7 +26,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _make_meta( def _make_meta(
@@ -27,7 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _zigzag_order(cp_size: int) -> list[int]: def _zigzag_order(cp_size: int) -> list[int]:
@@ -27,7 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestComputeReordererPlans: class TestComputeReordererPlans:
@@ -18,7 +18,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestComputeAxisAlignerPlan: class TestComputeAxisAlignerPlan:
@@ -17,7 +17,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
_sglang_plugin = _SGLangPlugin() _sglang_plugin = _SGLangPlugin()
_megatron_plugin = _MegatronPlugin() _megatron_plugin = _MegatronPlugin()
@@ -11,7 +11,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestExecuteConcat: class TestExecuteConcat:
@@ -30,7 +30,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=15, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor: def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor:
@@ -24,7 +24,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestBuildTokenIndexSGLangThd: class TestBuildTokenIndexSGLangThd:
@@ -31,7 +31,7 @@ from sglang.srt.debug_utils.comparator.output_types import ReplicatedCheckResult
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _name_tensors( def _name_tensors(
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestNormalizeParallelInfo: class TestNormalizeParallelInfo:
@@ -20,7 +20,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis, parse_dims
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestComputeUnsharderPlan: class TestComputeUnsharderPlan:
@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestParseDim: class TestParseDim:
@@ -16,7 +16,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestSingletonDimUtilFilterOut: class TestSingletonDimUtilFilterOut:
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestFindDimIndex: class TestFindDimIndex:
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestDimConstants: class TestDimConstants:
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import DiffInfo
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=20, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=20, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestComputeTensorInfo: class TestComputeTensorInfo:
@@ -22,7 +22,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _make_stats(**overrides) -> TensorStats: def _make_stats(**overrides) -> TensorStats:
@@ -12,7 +12,7 @@ from sglang.srt.debug_utils.comparator.report_sink import report_sink
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
def _make_error_log(**overrides) -> ErrorLog: def _make_error_log(**overrides) -> ErrorLog:
@@ -17,7 +17,7 @@ from sglang.srt.debug_utils.comparator.meta_overrider import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
# ───────────────────── Unit: MetaOverrideRule ───────────────────── # ───────────────────── Unit: MetaOverrideRule ─────────────────────
@@ -45,7 +45,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair, _check_equal_lengths
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestCheckEqualLengths: class TestCheckEqualLengths:
@@ -4,7 +4,7 @@ from sglang.srt.debug_utils.comparator.preset import PRESETS, expand_preset
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=5, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestExpandPreset: class TestExpandPreset:
@@ -18,7 +18,7 @@ from sglang.srt.debug_utils.comparator.utils import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestCalcRelDiff: class TestCalcRelDiff:
@@ -11,7 +11,7 @@ from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchSpec
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module" SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
@@ -9,7 +9,7 @@ from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module" SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
@@ -6,7 +6,7 @@ from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchApplicati
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=10, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestApplyEdits: class TestApplyEdits:
@@ -23,7 +23,7 @@ from sglang.test.test_utils import (
register_cuda_ci(est_time=40, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=40, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=40, suite="nightly-amd-1-gpu", nightly=True) register_amd_ci(est_time=40, suite="nightly-amd-1-gpu", nightly=True)
register_cpu_ci(est_time=225, suite="base-b-test-cpu") register_cpu_ci(est_time=225, suite="base-c-test-cpu")
class TestCrashDump(CustomTestCase): class TestCrashDump(CustomTestCase):
@@ -26,7 +26,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=120, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=120, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=338, suite="base-b-test-cpu") register_cpu_ci(est_time=338, suite="base-c-test-cpu")
# ==================== Non-E2E Tests ==================== # ==================== Non-E2E Tests ====================
@@ -19,7 +19,7 @@ from sglang.test.test_utils import (
register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=120, suite="nightly-amd-1-gpu", nightly=True) register_amd_ci(est_time=120, suite="nightly-amd-1-gpu", nightly=True)
register_cpu_ci(est_time=622, suite="base-b-test-cpu") register_cpu_ci(est_time=622, suite="base-c-test-cpu")
class BaseTestSoftWatchdog: class BaseTestSoftWatchdog:
@@ -12,7 +12,7 @@ from sglang.srt.parser.reasoning_parser import KimiK2Detector as KimiK2Reasoning
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(5, "base-a-test-cpu") register_cpu_ci(5, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
def _make_tool(name, parameters=None): def _make_tool(name, parameters=None):
+1 -1
View File
@@ -8,7 +8,7 @@ from sglang.test.kv_canary.e2e_base import (
) )
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=1, suite="base-b-test-cpu") register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestCanaryE2EBase(CustomTestCase): class TestCanaryE2EBase(CustomTestCase):
@@ -8,7 +8,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.kv_canary.e2e_base import CanaryE2EBase from sglang.test.kv_canary.e2e_base import CanaryE2EBase
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=5, suite="base-b-test-cpu") register_cpu_ci(est_time=5, suite="base-c-test-cpu")
_GOOD_LINE: str = SwaDivergenceLog( _GOOD_LINE: str = SwaDivergenceLog(
@@ -29,7 +29,7 @@ from sglang.test.ci.ci_register import (
register_cuda_ci(est_time=200, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=200, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=200, suite="nightly-amd-1-gpu", nightly=True) register_amd_ci(est_time=200, suite="nightly-amd-1-gpu", nightly=True)
register_cpu_ci(est_time=6, suite="base-b-test-cpu") register_cpu_ci(est_time=6, suite="base-c-test-cpu")
register_xpu_ci(est_time=10, suite="stage-a-test-1-gpu-xpu") register_xpu_ci(est_time=10, suite="stage-a-test-1-gpu-xpu")
+1 -1
View File
@@ -18,7 +18,7 @@ from sglang.test.ci.ci_register import (
register_cuda_ci(est_time=30, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=30, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=30, suite="nightly-amd-1-gpu", nightly=True) register_amd_ci(est_time=30, suite="nightly-amd-1-gpu", nightly=True)
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class MockTokenizerManager: class MockTokenizerManager:
@@ -11,7 +11,7 @@ from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=29, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=29, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=45, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=45, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=203, suite="base-b-test-cpu") register_cpu_ci(est_time=203, suite="base-c-test-cpu")
class TestExternalModels(CustomTestCase): class TestExternalModels(CustomTestCase):
@@ -13,7 +13,7 @@ from sglang.test.server_fixtures.default_fixture import DefaultServerBase
register_cuda_ci(est_time=48, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=48, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=48, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=48, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=320, suite="base-b-test-cpu") register_cpu_ci(est_time=320, suite="base-c-test-cpu")
class TestTransformersBackendEval(DefaultServerBase): class TestTransformersBackendEval(DefaultServerBase):
@@ -26,7 +26,7 @@ register_cuda_ci(
runner_config="1-gpu-small", runner_config="1-gpu-small",
) )
register_amd_ci(est_time=60, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=60, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=179, suite="base-b-test-cpu") register_cpu_ci(est_time=179, suite="base-c-test-cpu")
_MODEL_NAME = "Qwen/Qwen3-0.6B" _MODEL_NAME = "Qwen/Qwen3-0.6B"
@@ -6,7 +6,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=15, suite="base-a-test-cpu") register_cpu_ci(est_time=15, suite="base-a-test-cpu")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestQuantLogString(CustomTestCase): class TestQuantLogString(CustomTestCase):
@@ -10,7 +10,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
@@ -19,7 +19,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
_HTTP_SCOPE = { _HTTP_SCOPE = {
"type": "http", "type": "http",
@@ -22,7 +22,7 @@ from sglang.test.test_utils import (
register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True)
register_amd_ci(est_time=120, suite="nightly-amd-1-gpu", nightly=True) register_amd_ci(est_time=120, suite="nightly-amd-1-gpu", nightly=True)
register_cpu_ci(est_time=184, suite="base-b-test-cpu") register_cpu_ci(est_time=184, suite="base-c-test-cpu")
class TestRoutingKeyScheduling(CustomTestCase): class TestRoutingKeyScheduling(CustomTestCase):
@@ -9,7 +9,7 @@ from sglang.test.simple_eval_gsm8k import get_one_example
from sglang.test.simple_eval_mixed_prefix_gsm8k import MixedPrefixGSM8KEval from sglang.test.simple_eval_mixed_prefix_gsm8k import MixedPrefixGSM8KEval
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=5, suite="base-b-test-cpu") register_cpu_ci(est_time=5, suite="base-c-test-cpu")
def _write_synthetic_dataset(path: str, n: int) -> None: def _write_synthetic_dataset(path: str, n: int) -> None:
@@ -28,7 +28,7 @@ from sglang.srt.constrained.reasoner_grammar_backend import ReasonerGrammarObjec
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(2.0, "base-a-test-cpu") register_cpu_ci(2.0, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
def _make_scheduler(grammar_backend_name="none", skip_tokenizer=False): def _make_scheduler(grammar_backend_name="none", skip_tokenizer=False):
@@ -16,7 +16,7 @@ from sglang.srt.constrained.torch_ops.token_filter_torch_ops import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(2.0, "base-a-test-cpu") register_cpu_ci(2.0, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class _DummyTokenizer: class _DummyTokenizer:
@@ -15,7 +15,7 @@ from sglang.srt.constrained.utils import is_legacy_structural_tag
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(1.0, "base-a-test-cpu") register_cpu_ci(1.0, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestIsLegacyStructuralTag(unittest.TestCase): class TestIsLegacyStructuralTag(unittest.TestCase):
@@ -57,7 +57,7 @@ from sglang.srt.managers.io_struct import EmbeddingReqInput
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=10, suite="base-a-test-cpu") register_cpu_ci(est_time=10, suite="base-a-test-cpu")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
# Mock TokenizerManager for embedding tests # Mock TokenizerManager for embedding tests
@@ -32,7 +32,7 @@ from sglang.srt.function_call.qwen3_coder_detector import Qwen3CoderDetector
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=15, suite="base-a-test-cpu") register_cpu_ci(est_time=15, suite="base-a-test-cpu")
register_cpu_ci(est_time=61, suite="base-b-test-cpu") register_cpu_ci(est_time=61, suite="base-c-test-cpu")
class TestPythonicDetector(unittest.TestCase): class TestPythonicDetector(unittest.TestCase):
@@ -19,7 +19,7 @@ from sglang.srt.function_call.utils import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(5, "base-a-test-cpu") register_cpu_ci(5, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestJsonSchemaConstraint(unittest.TestCase): class TestJsonSchemaConstraint(unittest.TestCase):
@@ -24,7 +24,7 @@ from sglang.srt.function_call.json_array_parser import JsonArrayParser
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(5, "base-a-test-cpu") register_cpu_ci(5, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestParallelToolCalls(unittest.TestCase): class TestParallelToolCalls(unittest.TestCase):
@@ -10,7 +10,7 @@ from sglang.srt.function_call.core_types import StreamingParseResult
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(5, "base-a-test-cpu") register_cpu_ci(5, "base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class DummyDetector(BaseFormatDetector): class DummyDetector(BaseFormatDetector):
@@ -14,7 +14,7 @@ from sglang.srt.layers.moe.moe_runner.aiter import (
from sglang.srt.layers.moe.moe_runner.base import MoeRunnerConfig from sglang.srt.layers.moe.moe_runner.base import MoeRunnerConfig
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=5, suite="base-b-test-cpu") register_cpu_ci(est_time=5, suite="base-c-test-cpu")
def _runner_input(): def _runner_input():
@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-large") register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-large")
register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestGenerateReqInputNormalization(CustomTestCase): class TestGenerateReqInputNormalization(CustomTestCase):
@@ -18,7 +18,7 @@ from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=2, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=2, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestPrefillAdder(CustomTestCase): class TestPrefillAdder(CustomTestCase):
@@ -10,7 +10,7 @@ from sglang.test.ci.ci_register import (
register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestProfileMergerHTTPAPI(unittest.TestCase): class TestProfileMergerHTTPAPI(unittest.TestCase):
@@ -13,7 +13,7 @@ from sglang.srt.managers.scheduler_components.flush_wrapper import (
) )
register_cpu_ci(est_time=14, suite="base-a-test-cpu") register_cpu_ci(est_time=14, suite="base-a-test-cpu")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestSchedulerFlushCache(unittest.TestCase): class TestSchedulerFlushCache(unittest.TestCase):
@@ -12,7 +12,7 @@ from sglang.srt.managers.scheduler import Scheduler
from sglang.srt.managers.scheduler_components.pool_stats_observer import PoolStats from sglang.srt.managers.scheduler_components.pool_stats_observer import PoolStats
register_cpu_ci(est_time=15, suite="base-a-test-cpu") register_cpu_ci(est_time=15, suite="base-a-test-cpu")
register_cpu_ci(est_time=9, suite="base-b-test-cpu") register_cpu_ci(est_time=9, suite="base-c-test-cpu")
class TestSchedulerPauseGeneration(unittest.TestCase): class TestSchedulerPauseGeneration(unittest.TestCase):
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import unittest import unittest
from unittest.mock import MagicMock from unittest.mock import MagicMock
@@ -20,7 +20,7 @@ from sglang.test.ci.ci_register import (
register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=8, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestSLRUAccuracy(unittest.TestCase): class TestSLRUAccuracy(unittest.TestCase):
+1 -1
View File
@@ -11,7 +11,7 @@ from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class PixtralVisionConfig: class PixtralVisionConfig:
@@ -7,7 +7,7 @@ from unittest.mock import MagicMock, patch
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=60, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=60, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestCpuMonitor(unittest.TestCase): class TestCpuMonitor(unittest.TestCase):
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import asyncio import asyncio
import unittest import unittest
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import unittest import unittest
@@ -7,7 +7,7 @@ from sglang.srt.observability.utils import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestMetricsUtils(unittest.TestCase): class TestMetricsUtils(unittest.TestCase):
@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import asyncio import asyncio
import json import json
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import unittest import unittest
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
@@ -19,7 +19,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestFimPosition(CustomTestCase): class TestFimPosition(CustomTestCase):
@@ -33,7 +33,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestConversationGetPrompt(CustomTestCase): class TestConversationGetPrompt(CustomTestCase):
@@ -15,7 +15,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestEvent(CustomTestCase): class TestEvent(CustomTestCase):
@@ -10,7 +10,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestTemplateContentFormatDetection(CustomTestCase): class TestTemplateContentFormatDetection(CustomTestCase):
@@ -5,7 +5,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
# Simulated model output that contains think tags (e.g. from DeepSeek-R1) # Simulated model output that contains think tags (e.g. from DeepSeek-R1)
THINK_OUTPUT = ( THINK_OUTPUT = (
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import json import json
import unittest import unittest
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=9, suite="base-a-test-cpu") register_cpu_ci(est_time=9, suite="base-a-test-cpu")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
import unittest import unittest
from unittest.mock import MagicMock from unittest.mock import MagicMock
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=9, suite="base-a-test-cpu") register_cpu_ci(est_time=9, suite="base-a-test-cpu")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
import unittest import unittest
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
@@ -3,7 +3,7 @@
from sglang.test.ci.ci_register import register_cpu_ci, register_xpu_ci from sglang.test.ci.ci_register import register_cpu_ci, register_xpu_ci
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
register_xpu_ci(est_time=10, suite="stage-a-test-1-gpu-xpu") register_xpu_ci(est_time=10, suite="stage-a-test-1-gpu-xpu")
import unittest import unittest
@@ -23,7 +23,7 @@ from sglang.test.test_utils import (
) )
register_cpu_ci(est_time=10, suite="base-a-test-cpu") register_cpu_ci(est_time=10, suite="base-a-test-cpu")
register_cpu_ci(est_time=12, suite="base-b-test-cpu") register_cpu_ci(est_time=12, suite="base-c-test-cpu")
# Mock get_device() so all tests run on CPU-only CI runners # Mock get_device() so all tests run on CPU-only CI runners
_mock_device = patch("sglang.srt.server_args.get_device", return_value="cuda") _mock_device = patch("sglang.srt.server_args.get_device", return_value="cuda")
+1 -1
View File
@@ -7,7 +7,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestRunaiUtils(CustomTestCase): class TestRunaiUtils(CustomTestCase):
@@ -1,7 +1,7 @@
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
import unittest import unittest
@@ -11,7 +11,7 @@ from sglang.srt.utils.auth import AuthLevel, decide_request_auth
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestHttpServerAdminAuth(unittest.TestCase): class TestHttpServerAdminAuth(unittest.TestCase):
@@ -11,7 +11,7 @@ from sglang.srt.utils.json_response import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestJSONResponseUtils(unittest.TestCase): class TestJSONResponseUtils(unittest.TestCase):
@@ -12,7 +12,7 @@ from sglang.srt.utils.patch_tokenizer import (
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True) register_cpu_ci(est_time=30, suite="base-a-test-cpu", nightly=True)
register_cpu_ci(est_time=53, suite="base-b-test-cpu") register_cpu_ci(est_time=53, suite="base-c-test-cpu")
class TestPatchTokenizerEndToEndTest(unittest.TestCase): class TestPatchTokenizerEndToEndTest(unittest.TestCase):
@@ -23,7 +23,7 @@ from sglang.test.ci.ci_register import (
register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=9, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestProfileMerger(unittest.TestCase): class TestProfileMerger(unittest.TestCase):
@@ -25,7 +25,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=9, suite="base-a-test-cpu") register_cpu_ci(est_time=9, suite="base-a-test-cpu")
register_cpu_ci(est_time=9, suite="base-b-test-cpu") register_cpu_ci(est_time=9, suite="base-c-test-cpu")
def healthy_worker(): def healthy_worker():
+1 -1
View File
@@ -11,7 +11,7 @@ from sglang.srt.utils.log_utils import create_log_targets, log_json
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=6, suite="base-a-test-cpu") register_cpu_ci(est_time=6, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
_LOG_PREFIX_RE = re.compile(r"^\[\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\] ") _LOG_PREFIX_RE = re.compile(r"^\[\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\] ")
@@ -29,7 +29,7 @@ from sglang.test.test_utils import (
# Note: AMD registration removed - test_model_file_verifier fails on AMD # Note: AMD registration removed - test_model_file_verifier fails on AMD
register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True) register_cuda_ci(est_time=120, suite="nightly-1-gpu", nightly=True)
register_cpu_ci(est_time=540, suite="base-b-test-cpu") register_cpu_ci(est_time=540, suite="base-c-test-cpu")
MODEL_NAME = "Qwen/Qwen3-0.6B" MODEL_NAME = "Qwen/Qwen3-0.6B"
@@ -7,7 +7,7 @@ from sglang.srt.utils.network import NetworkAddress, is_zmq_endpoint_ipv6
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
# Mock get_device() so ServerArgs tests run on CPU-only CI runners # Mock get_device() so ServerArgs tests run on CPU-only CI runners
_mock_device = patch("sglang.srt.server_args.get_device", return_value="cuda") _mock_device = patch("sglang.srt.server_args.get_device", return_value="cuda")
+1 -1
View File
@@ -16,7 +16,7 @@ from sglang.test.test_utils import CustomTestCase
from sglang.utils import normalize_base_url, release_port, reserve_port from sglang.utils import normalize_base_url, release_port, reserve_port
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class TestTryBindSocket(CustomTestCase): class TestTryBindSocket(CustomTestCase):
@@ -12,7 +12,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cpu_ci
from sglang.utils import TypeBasedDispatcher from sglang.utils import TypeBasedDispatcher
register_amd_ci(est_time=10, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=10, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
class TestTypeBasedDispatcher(unittest.TestCase): class TestTypeBasedDispatcher(unittest.TestCase):
+1 -1
View File
@@ -12,7 +12,7 @@ from sglang.test.test_utils import run_doctests
register_cuda_ci(est_time=11, stage="base-b", runner_config="1-gpu-small") register_cuda_ci(est_time=11, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=20, suite="stage-b-test-1-gpu-small-amd") register_amd_ci(est_time=20, suite="stage-b-test-1-gpu-small-amd")
register_cpu_ci(est_time=8, suite="base-b-test-cpu") register_cpu_ci(est_time=8, suite="base-c-test-cpu")
def test_resolve_evs_config(): def test_resolve_evs_config():
+1 -1
View File
@@ -6,7 +6,7 @@ from sglang.srt.utils import sample_video_frames
from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.ci.ci_register import register_cpu_ci
register_cpu_ci(est_time=7, suite="base-a-test-cpu") register_cpu_ci(est_time=7, suite="base-a-test-cpu")
register_cpu_ci(est_time=7, suite="base-b-test-cpu") register_cpu_ci(est_time=7, suite="base-c-test-cpu")
class DummyVideo: class DummyVideo:
+6 -1
View File
@@ -29,7 +29,12 @@ HW_MAPPING = {
# (label-gated; pr-test-extra.yml). Tests are tagged per-commit regardless; # (label-gated; pr-test-extra.yml). Tests are tagged per-commit regardless;
# pr-test-extra.yml's `run-ci-extra` PR label decides whether extra-* dispatches. # pr-test-extra.yml's `run-ci-extra` PR label decides whether extra-* dispatches.
PER_COMMIT_SUITES = { PER_COMMIT_SUITES = {
HWBackend.CPU: ["base-a-test-cpu", "base-b-test-cpu", "base-b-test-cpu-arm64"], HWBackend.CPU: [
"base-a-test-cpu",
"base-b-test-cpu",
"base-c-test-cpu",
"base-b-test-cpu-arm64",
],
HWBackend.AMD: [ HWBackend.AMD: [
"stage-a-test-1-gpu-small-amd", "stage-a-test-1-gpu-small-amd",
"stage-b-test-1-gpu-small-amd", "stage-b-test-1-gpu-small-amd",