ci: unify PR test suite naming (#21187)
This commit is contained in:
Lianmin Zheng
@@ -28,7 +28,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis, TokenLayou
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestExecuteSubPlans:
|
||||
|
||||
@@ -25,7 +25,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _make_meta(
|
||||
|
||||
@@ -22,7 +22,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _zigzag_order(cp_size: int) -> list[int]:
|
||||
|
||||
@@ -24,7 +24,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestComputeReordererPlans:
|
||||
|
||||
@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.log_sink import log_sink
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestComputeAxisAlignerPlan:
|
||||
|
||||
@@ -19,7 +19,7 @@ from sglang.srt.debug_utils.comparator.log_sink import LogSink
|
||||
from sglang.srt.debug_utils.comparator.output_types import ErrorLog, InfoLog
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_sglang_plugin = _SGLangPlugin()
|
||||
_megatron_plugin = _MegatronPlugin()
|
||||
|
||||
@@ -16,7 +16,7 @@ from sglang.srt.debug_utils.comparator.aligner.token_aligner.smart.types import
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_sglang_plugin = _SGLangPlugin()
|
||||
_megatron_plugin = _MegatronPlugin()
|
||||
|
||||
@@ -9,7 +9,7 @@ from sglang.srt.debug_utils.comparator.aligner.token_aligner.concat_steps import
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestExecuteConcat:
|
||||
|
||||
@@ -25,7 +25,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor:
|
||||
|
||||
@@ -23,7 +23,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import TokenLayout
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=30, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestBuildTokenIndexSGLangThd:
|
||||
|
||||
+1
-1
@@ -14,7 +14,7 @@ from sglang.srt.debug_utils.comparator.aligner.token_aligner.smart.aux_plugins i
|
||||
)
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _save_pt(
|
||||
|
||||
@@ -27,7 +27,7 @@ from sglang.srt.debug_utils.comparator.dims_spec import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _name_tensors(
|
||||
|
||||
@@ -9,7 +9,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import AxisInfo
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestNormalizeParallelInfo:
|
||||
|
||||
@@ -19,7 +19,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestComputeUnsharderPlan:
|
||||
|
||||
@@ -12,7 +12,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestParseDim:
|
||||
|
||||
@@ -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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestSingletonDimUtilFilterOut:
|
||||
|
||||
@@ -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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestFindDimIndex:
|
||||
|
||||
@@ -9,7 +9,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestDimConstants:
|
||||
|
||||
@@ -14,7 +14,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.comparator import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=20, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestComputeTensorInfo:
|
||||
|
||||
@@ -58,7 +58,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
||||
from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_DEFAULT_PERCENTILE_LINES: list[str] = [
|
||||
" [blue]p1 [/] -1.8000 -1.8000 [dim]+0.00e+00[/]",
|
||||
|
||||
@@ -21,7 +21,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _make_stats(**overrides) -> TensorStats:
|
||||
|
||||
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.srt.debug_utils.dump_loader import ValueWithMeta
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _save_tensor(
|
||||
|
||||
@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.bundle_matcher import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _make_row(
|
||||
|
||||
@@ -21,7 +21,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
|
||||
)
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _render_rich(renderable: object) -> str:
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.srt.debug_utils.comparator.dp_utils import (
|
||||
from sglang.srt.debug_utils.dump_loader import ValueWithMeta
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=15, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _make_sglang_meta(
|
||||
|
||||
@@ -7,7 +7,7 @@ import torch
|
||||
from sglang.srt.debug_utils.dump_loader import read_tokenizer_path
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _save_pt(
|
||||
|
||||
@@ -30,7 +30,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
|
||||
from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper, _RecomputeStatus
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=30, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_FIXED_EXP_NAME = "my_exp_name"
|
||||
|
||||
|
||||
@@ -11,7 +11,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _make_error_log(**overrides) -> ErrorLog:
|
||||
|
||||
@@ -23,7 +23,7 @@ import torch
|
||||
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=60, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=60, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_PUBLISH_DIR: Path = Path("/tmp/comparator_manual_verify")
|
||||
_PNG_MAGIC: bytes = b"\x89PNG"
|
||||
|
||||
@@ -16,7 +16,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
# ───────────────────── Unit: MetaOverrideRule ─────────────────────
|
||||
|
||||
@@ -44,7 +44,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestCheckEqualLengths:
|
||||
|
||||
@@ -55,7 +55,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
|
||||
from sglang.srt.debug_utils.comparator.utils import Pair
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
def _render_rich(renderable: object) -> str:
|
||||
|
||||
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.comparator import (
|
||||
)
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=30, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
_PNG_MAGIC: bytes = b"\x89PNG"
|
||||
|
||||
|
||||
@@ -3,7 +3,7 @@ import pytest
|
||||
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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=5, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestExpandPreset:
|
||||
|
||||
@@ -17,7 +17,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="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=10, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestCalcRelDiff:
|
||||
|
||||
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.visualizer.preprocessing import (
|
||||
)
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||
register_cpu_ci(est_time=30, suite="stage-a-test-cpu", nightly=True)
|
||||
|
||||
|
||||
class TestPreprocessTensor:
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(
|
||||
est_time=0,
|
||||
suite="stage-a-cpu-only",
|
||||
suite="stage-a-test-cpu",
|
||||
nightly=True,
|
||||
disabled="helper module, no tests",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user