[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
@@ -34,7 +34,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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(
@@ -27,7 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
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=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]:
@@ -27,7 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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:
@@ -17,7 +17,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
_sglang_plugin = _SGLangPlugin()
_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
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:
@@ -30,7 +30,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
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=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:
@@ -24,7 +24,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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(
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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:
@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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:
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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:
@@ -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
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:
@@ -22,7 +22,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
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
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:
@@ -17,7 +17,7 @@ from sglang.srt.debug_utils.comparator.meta_overrider import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
# ───────────────────── 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
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:
@@ -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
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:
@@ -18,7 +18,7 @@ from sglang.srt.debug_utils.comparator.utils import (
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=1, suite="base-b-test-cpu")
register_cpu_ci(est_time=1, suite="base-c-test-cpu")
class TestCalcRelDiff: