[CI] reduce CPU CI scope with base-c suite (#28623)
This commit is contained in:
@@ -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):
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -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):
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|||||||
@@ -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):
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
@@ -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
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user