ci: merge CPU default and stage-a-cpu-only suites (#20992)
This commit is contained in:
@@ -825,8 +825,6 @@ jobs:
|
|||||||
CONTINUE_ON_ERROR_FLAG="--continue-on-error"
|
CONTINUE_ON_ERROR_FLAG="--continue-on-error"
|
||||||
fi
|
fi
|
||||||
python3 run_suite.py --hw cuda --suite stage-a-test-1 $CONTINUE_ON_ERROR_FLAG
|
python3 run_suite.py --hw cuda --suite stage-a-test-1 $CONTINUE_ON_ERROR_FLAG
|
||||||
# temporarily put backend-independent cpu tests here
|
|
||||||
python3 run_suite.py --hw cpu --suite default $CONTINUE_ON_ERROR_FLAG
|
|
||||||
|
|
||||||
- uses: ./.github/actions/upload-cuda-coredumps
|
- uses: ./.github/actions/upload-cuda-coredumps
|
||||||
if: always()
|
if: always()
|
||||||
|
|||||||
@@ -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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestExecuteSubPlans:
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _make_meta(
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _zigzag_order(cp_size: int) -> list[int]:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=10, suite="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestComputeReordererPlans:
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestComputeAxisAlignerPlan:
|
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.srt.debug_utils.comparator.output_types import ErrorLog, InfoLog
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
_sglang_plugin = _SGLangPlugin()
|
_sglang_plugin = _SGLangPlugin()
|
||||||
_megatron_plugin = _MegatronPlugin()
|
_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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
_sglang_plugin = _SGLangPlugin()
|
_sglang_plugin = _SGLangPlugin()
|
||||||
_megatron_plugin = _MegatronPlugin()
|
_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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestExecuteConcat:
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _named(tensor: torch.Tensor, names: list[str]) -> torch.Tensor:
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestBuildTokenIndexSGLangThd:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=15, suite="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _save_pt(
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _name_tensors(
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestNormalizeParallelInfo:
|
class TestNormalizeParallelInfo:
|
||||||
|
|||||||
@@ -14,7 +14,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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestComputeUnsharderPlan:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=5, suite="default", nightly=True)
|
register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestParseDim:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=5, suite="default", nightly=True)
|
register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestSingletonDimUtilFilterOut:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=5, suite="default", nightly=True)
|
register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestFindDimIndex:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=5, suite="default", nightly=True)
|
register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestDimConstants:
|
class TestDimConstants:
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.comparator import (
|
|||||||
from sglang.srt.debug_utils.comparator.tensor_comparator.types import DiffInfo
|
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="default", nightly=True)
|
register_cpu_ci(est_time=20, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestComputeTensorStats:
|
class TestComputeTensorStats:
|
||||||
|
|||||||
@@ -55,7 +55,7 @@ from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
|||||||
from sglang.srt.debug_utils.comparator.utils import Pair
|
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=10, suite="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
# Snapshot strings below are intentionally spelled out in full per test.
|
# Snapshot strings below are intentionally spelled out in full per test.
|
||||||
|
|||||||
@@ -21,7 +21,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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _make_stats(**overrides) -> TensorStats:
|
def _make_stats(**overrides) -> TensorStats:
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.comparator.log_sink import LogSink
|
|||||||
from sglang.srt.debug_utils.comparator.output_types import ErrorLog
|
from sglang.srt.debug_utils.comparator.output_types import ErrorLog
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _save_tensor(
|
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.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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _make_row(
|
def _make_row(
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ from sglang.srt.debug_utils.comparator.output_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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _render_rich(renderable: object) -> str:
|
def _render_rich(renderable: object) -> str:
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ from sglang.srt.debug_utils.comparator.dp_utils import (
|
|||||||
from sglang.srt.debug_utils.dump_loader import ValueWithMeta
|
from sglang.srt.debug_utils.dump_loader import ValueWithMeta
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _make_sglang_meta(
|
def _make_sglang_meta(
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import torch
|
|||||||
from sglang.srt.debug_utils.dump_loader import read_tokenizer_path
|
from sglang.srt.debug_utils.dump_loader import read_tokenizer_path
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _save_pt(
|
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.srt.debug_utils.dumper import DumperConfig, _Dumper, _RecomputeStatus
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
_FIXED_EXP_NAME = "my_exp_name"
|
_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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _make_error_log(**overrides) -> ErrorLog:
|
def _make_error_log(**overrides) -> ErrorLog:
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ import torch
|
|||||||
|
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=60, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
_PUBLISH_DIR: Path = Path("/tmp/comparator_manual_verify")
|
_PUBLISH_DIR: Path = Path("/tmp/comparator_manual_verify")
|
||||||
_PNG_MAGIC: bytes = b"\x89PNG"
|
_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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=10, suite="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
# ───────────────────── Unit: MetaOverrideRule ─────────────────────
|
# ───────────────────── 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.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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestCheckEqualLengths:
|
class TestCheckEqualLengths:
|
||||||
|
|||||||
@@ -49,7 +49,7 @@ from sglang.srt.debug_utils.comparator.output_types import (
|
|||||||
from sglang.srt.debug_utils.comparator.utils import Pair
|
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=10, suite="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
def _render_rich(renderable: object) -> str:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=30, suite="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
_PNG_MAGIC: bytes = b"\x89PNG"
|
_PNG_MAGIC: bytes = b"\x89PNG"
|
||||||
|
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import pytest
|
|||||||
from sglang.srt.debug_utils.comparator.preset import PRESETS, expand_preset
|
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="default", nightly=True)
|
register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestExpandPreset:
|
class TestExpandPreset:
|
||||||
|
|||||||
@@ -17,7 +17,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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestCalcRelDiff:
|
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
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(est_time=30, suite="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestPreprocessTensor:
|
class TestPreprocessTensor:
|
||||||
|
|||||||
@@ -7,7 +7,10 @@ from typing import Optional
|
|||||||
from sglang.test.ci.ci_register import register_cpu_ci
|
from sglang.test.ci.ci_register import register_cpu_ci
|
||||||
|
|
||||||
register_cpu_ci(
|
register_cpu_ci(
|
||||||
est_time=0, suite="default", nightly=True, disabled="helper module, no tests"
|
est_time=0,
|
||||||
|
suite="stage-a-cpu-only",
|
||||||
|
nightly=True,
|
||||||
|
disabled="helper module, no tests",
|
||||||
)
|
)
|
||||||
|
|
||||||
from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
from sglang.srt.debug_utils.comparator.tensor_comparator.types import (
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from sglang.srt.debug_utils.source_patcher.code_patcher import (
|
|||||||
from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchSpec
|
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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
|
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
|
||||||
|
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import yaml
|
|||||||
from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper
|
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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
|
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
|
||||||
|
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from sglang.srt.debug_utils.source_patcher.source_editor import apply_edits
|
|||||||
from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchApplicationError
|
from sglang.srt.debug_utils.source_patcher.types import EditSpec, PatchApplicationError
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestApplyEdits:
|
class TestApplyEdits:
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ from sglang.srt.debug_utils.dump_comparator import (
|
|||||||
from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper
|
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=30, suite="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
# ----------------------------- Unit tests -----------------------------
|
# ----------------------------- Unit tests -----------------------------
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ from sglang.srt.debug_utils.dump_loader 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="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestReadMeta:
|
class TestReadMeta:
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ from sglang.srt.debug_utils.schedule_simulator import (
|
|||||||
from sglang.test.ci.ci_register import register_cpu_ci
|
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="default", nightly=True)
|
register_cpu_ci(est_time=120, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
# ==================== Non-E2E Tests ====================
|
# ==================== Non-E2E Tests ====================
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ from sglang.srt.function_call.kimik2_detector import (
|
|||||||
from sglang.srt.parser.reasoning_parser import KimiK2Detector as KimiK2ReasoningDetector
|
from sglang.srt.parser.reasoning_parser import KimiK2Detector as KimiK2ReasoningDetector
|
||||||
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, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
def _make_tool(name, parameters=None):
|
def _make_tool(name, parameters=None):
|
||||||
|
|||||||
@@ -18,7 +18,7 @@ from sglang.srt.function_call.pythonic_detector import PythonicDetector
|
|||||||
from sglang.srt.function_call.qwen3_coder_detector import Qwen3CoderDetector
|
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(1.0, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class TestPythonicDetector(unittest.TestCase):
|
class TestPythonicDetector(unittest.TestCase):
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from sglang.srt.function_call.glm47_moe_detector import (
|
|||||||
)
|
)
|
||||||
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, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class TestGlm47MoeDetector(unittest.TestCase):
|
class TestGlm47MoeDetector(unittest.TestCase):
|
||||||
|
|||||||
@@ -18,7 +18,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(1.0, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class TestJsonSchemaConstraint(unittest.TestCase):
|
class TestJsonSchemaConstraint(unittest.TestCase):
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ from sglang.srt.entrypoints.openai.protocol import Function, Tool
|
|||||||
from sglang.srt.function_call.json_array_parser import JsonArrayParser
|
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(1.0, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class TestParallelToolCalls(unittest.TestCase):
|
class TestParallelToolCalls(unittest.TestCase):
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from sglang.srt.function_call.base_format_detector import BaseFormatDetector
|
|||||||
from sglang.srt.function_call.core_types import StreamingParseResult
|
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(1.0, "default")
|
register_cpu_ci(1.0, "stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class DummyDetector(BaseFormatDetector):
|
class DummyDetector(BaseFormatDetector):
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import unittest
|
|||||||
|
|
||||||
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="default", nightly=True)
|
register_cpu_ci(est_time=60, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestCpuMonitor(unittest.TestCase):
|
class TestCpuMonitor(unittest.TestCase):
|
||||||
|
|||||||
@@ -10,7 +10,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="default", nightly=True)
|
register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True)
|
||||||
|
|
||||||
|
|
||||||
class TestPatchTokenizerEndToEndTest(unittest.TestCase):
|
class TestPatchTokenizerEndToEndTest(unittest.TestCase):
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from pathlib import Path
|
|||||||
from sglang.srt.utils.log_utils import create_log_targets, log_json
|
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=1, suite="default")
|
register_cpu_ci(est_time=1, suite="stage-a-cpu-only")
|
||||||
|
|
||||||
|
|
||||||
class TestLogUtils(unittest.TestCase):
|
class TestLogUtils(unittest.TestCase):
|
||||||
|
|||||||
+1
-1
@@ -22,7 +22,7 @@ HW_MAPPING = {
|
|||||||
|
|
||||||
# Per-commit test suites (run on every PR)
|
# Per-commit test suites (run on every PR)
|
||||||
PER_COMMIT_SUITES = {
|
PER_COMMIT_SUITES = {
|
||||||
HWBackend.CPU: ["default", "stage-a-cpu-only"],
|
HWBackend.CPU: ["stage-a-cpu-only"],
|
||||||
HWBackend.AMD: [
|
HWBackend.AMD: [
|
||||||
"stage-a-test-1-amd",
|
"stage-a-test-1-amd",
|
||||||
"stage-b-test-small-1-gpu-amd",
|
"stage-b-test-small-1-gpu-amd",
|
||||||
|
|||||||
Reference in New Issue
Block a user