diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 1d516842a..c1220f2cf 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -825,8 +825,6 @@ jobs: CONTINUE_ON_ERROR_FLAG="--continue-on-error" fi 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 if: always() diff --git a/test/registered/debug_utils/comparator/aligner/entrypoint/test_executor.py b/test/registered/debug_utils/comparator/aligner/entrypoint/test_executor.py index a6b6bad71..fe10e02d5 100644 --- a/test/registered/debug_utils/comparator/aligner/entrypoint/test_executor.py +++ b/test/registered/debug_utils/comparator/aligner/entrypoint/test_executor.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) class TestExecuteSubPlans: diff --git a/test/registered/debug_utils/comparator/aligner/entrypoint/test_planner.py b/test/registered/debug_utils/comparator/aligner/entrypoint/test_planner.py index 139adeaf9..c8dfc9b89 100644 --- a/test/registered/debug_utils/comparator/aligner/entrypoint/test_planner.py +++ b/test/registered/debug_utils/comparator/aligner/entrypoint/test_planner.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) def _make_meta( diff --git a/test/registered/debug_utils/comparator/aligner/reorderer/test_executor.py b/test/registered/debug_utils/comparator/aligner/reorderer/test_executor.py index 811cc3d68..47a5a5f40 100644 --- a/test/registered/debug_utils/comparator/aligner/reorderer/test_executor.py +++ b/test/registered/debug_utils/comparator/aligner/reorderer/test_executor.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _zigzag_order(cp_size: int) -> list[int]: diff --git a/test/registered/debug_utils/comparator/aligner/reorderer/test_planner.py b/test/registered/debug_utils/comparator/aligner/reorderer/test_planner.py index 1ecb0dcd1..f429f2ba5 100644 --- a/test/registered/debug_utils/comparator/aligner/reorderer/test_planner.py +++ b/test/registered/debug_utils/comparator/aligner/reorderer/test_planner.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) class TestComputeReordererPlans: diff --git a/test/registered/debug_utils/comparator/aligner/test_axis_aligner.py b/test/registered/debug_utils/comparator/aligner/test_axis_aligner.py index ccff155ad..d811602cf 100644 --- a/test/registered/debug_utils/comparator/aligner/test_axis_aligner.py +++ b/test/registered/debug_utils/comparator/aligner/test_axis_aligner.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) class TestComputeAxisAlignerPlan: diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_loader.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_loader.py index 3ebc24d83..50ce2fb3b 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_loader.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_loader.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) _sglang_plugin = _SGLangPlugin() _megatron_plugin = _MegatronPlugin() diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_plugins.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_plugins.py index e53473738..76e34cdf4 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_plugins.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_aux_plugins.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) _sglang_plugin = _SGLangPlugin() _megatron_plugin = _MegatronPlugin() diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_concat_steps.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_concat_steps.py index c9bc2c238..2d89dd914 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_concat_steps.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_concat_steps.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) class TestExecuteConcat: diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_executor.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_executor.py index ec4325642..de8b9aa01 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_executor.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_executor.py @@ -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="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: diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_planner.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_planner.py index 180780c4e..c55673e0c 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_planner.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_planner.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True) class TestBuildTokenIndexSGLangThd: diff --git a/test/registered/debug_utils/comparator/aligner/token_aligner/test_thd_seq_lens_loader.py b/test/registered/debug_utils/comparator/aligner/token_aligner/test_thd_seq_lens_loader.py index 6e04d6289..e72171315 100644 --- a/test/registered/debug_utils/comparator/aligner/token_aligner/test_thd_seq_lens_loader.py +++ b/test/registered/debug_utils/comparator/aligner/token_aligner/test_thd_seq_lens_loader.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) def _save_pt( diff --git a/test/registered/debug_utils/comparator/aligner/unsharder/test_executor.py b/test/registered/debug_utils/comparator/aligner/unsharder/test_executor.py index e467b23f3..1e8e9d0e1 100644 --- a/test/registered/debug_utils/comparator/aligner/unsharder/test_executor.py +++ b/test/registered/debug_utils/comparator/aligner/unsharder/test_executor.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _name_tensors( diff --git a/test/registered/debug_utils/comparator/aligner/unsharder/test_parallel_info.py b/test/registered/debug_utils/comparator/aligner/unsharder/test_parallel_info.py index 2388e9552..5c82d6d88 100644 --- a/test/registered/debug_utils/comparator/aligner/unsharder/test_parallel_info.py +++ b/test/registered/debug_utils/comparator/aligner/unsharder/test_parallel_info.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) class TestNormalizeParallelInfo: diff --git a/test/registered/debug_utils/comparator/aligner/unsharder/test_planner.py b/test/registered/debug_utils/comparator/aligner/unsharder/test_planner.py index bb2044c88..c535dc215 100644 --- a/test/registered/debug_utils/comparator/aligner/unsharder/test_planner.py +++ b/test/registered/debug_utils/comparator/aligner/unsharder/test_planner.py @@ -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.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: diff --git a/test/registered/debug_utils/comparator/dims_spec/test_dim_parser.py b/test/registered/debug_utils/comparator/dims_spec/test_dim_parser.py index d04a6fb1c..adffcf9fb 100644 --- a/test/registered/debug_utils/comparator/dims_spec/test_dim_parser.py +++ b/test/registered/debug_utils/comparator/dims_spec/test_dim_parser.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True) class TestParseDim: diff --git a/test/registered/debug_utils/comparator/dims_spec/test_dims_parser.py b/test/registered/debug_utils/comparator/dims_spec/test_dims_parser.py index ab1e17d76..22f5d3156 100644 --- a/test/registered/debug_utils/comparator/dims_spec/test_dims_parser.py +++ b/test/registered/debug_utils/comparator/dims_spec/test_dims_parser.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True) class TestSingletonDimUtilFilterOut: diff --git a/test/registered/debug_utils/comparator/dims_spec/test_tensor_naming.py b/test/registered/debug_utils/comparator/dims_spec/test_tensor_naming.py index 2719760a8..b5156a344 100644 --- a/test/registered/debug_utils/comparator/dims_spec/test_tensor_naming.py +++ b/test/registered/debug_utils/comparator/dims_spec/test_tensor_naming.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True) class TestFindDimIndex: diff --git a/test/registered/debug_utils/comparator/dims_spec/test_types.py b/test/registered/debug_utils/comparator/dims_spec/test_types.py index bc0e627b7..1792a9e55 100644 --- a/test/registered/debug_utils/comparator/dims_spec/test_types.py +++ b/test/registered/debug_utils/comparator/dims_spec/test_types.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True) class TestDimConstants: diff --git a/test/registered/debug_utils/comparator/tensor_comparator/test_comparator.py b/test/registered/debug_utils/comparator/tensor_comparator/test_comparator.py index 426336121..76d197d03 100644 --- a/test/registered/debug_utils/comparator/tensor_comparator/test_comparator.py +++ b/test/registered/debug_utils/comparator/tensor_comparator/test_comparator.py @@ -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.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: diff --git a/test/registered/debug_utils/comparator/tensor_comparator/test_formatter.py b/test/registered/debug_utils/comparator/tensor_comparator/test_formatter.py index 1fb37ecbb..057cbcfa1 100644 --- a/test/registered/debug_utils/comparator/tensor_comparator/test_formatter.py +++ b/test/registered/debug_utils/comparator/tensor_comparator/test_formatter.py @@ -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.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. diff --git a/test/registered/debug_utils/comparator/tensor_comparator/test_types.py b/test/registered/debug_utils/comparator/tensor_comparator/test_types.py index 17ddcedb9..593e3b110 100644 --- a/test/registered/debug_utils/comparator/tensor_comparator/test_types.py +++ b/test/registered/debug_utils/comparator/tensor_comparator/test_types.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _make_stats(**overrides) -> TensorStats: diff --git a/test/registered/debug_utils/comparator/test_bundle_comparator.py b/test/registered/debug_utils/comparator/test_bundle_comparator.py index 5254b1e86..52c19040a 100644 --- a/test/registered/debug_utils/comparator/test_bundle_comparator.py +++ b/test/registered/debug_utils/comparator/test_bundle_comparator.py @@ -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.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( diff --git a/test/registered/debug_utils/comparator/test_bundle_matcher.py b/test/registered/debug_utils/comparator/test_bundle_matcher.py index cab011061..65a56f31c 100644 --- a/test/registered/debug_utils/comparator/test_bundle_matcher.py +++ b/test/registered/debug_utils/comparator/test_bundle_matcher.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) def _make_row( diff --git a/test/registered/debug_utils/comparator/test_display.py b/test/registered/debug_utils/comparator/test_display.py index ebbf7bb0f..6e29efb4a 100644 --- a/test/registered/debug_utils/comparator/test_display.py +++ b/test/registered/debug_utils/comparator/test_display.py @@ -20,7 +20,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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _render_rich(renderable: object) -> str: diff --git a/test/registered/debug_utils/comparator/test_dp_utils.py b/test/registered/debug_utils/comparator/test_dp_utils.py index 7fe2c81e5..62a391c1b 100644 --- a/test/registered/debug_utils/comparator/test_dp_utils.py +++ b/test/registered/debug_utils/comparator/test_dp_utils.py @@ -11,7 +11,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="default", nightly=True) +register_cpu_ci(est_time=15, suite="stage-a-cpu-only", nightly=True) def _make_sglang_meta( diff --git a/test/registered/debug_utils/comparator/test_dump_loader.py b/test/registered/debug_utils/comparator/test_dump_loader.py index 0938cf60d..7e88fc6ac 100644 --- a/test/registered/debug_utils/comparator/test_dump_loader.py +++ b/test/registered/debug_utils/comparator/test_dump_loader.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _save_pt( diff --git a/test/registered/debug_utils/comparator/test_entrypoint.py b/test/registered/debug_utils/comparator/test_entrypoint.py index 2ab49899f..87e68f093 100644 --- a/test/registered/debug_utils/comparator/test_entrypoint.py +++ b/test/registered/debug_utils/comparator/test_entrypoint.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True) _FIXED_EXP_NAME = "my_exp_name" diff --git a/test/registered/debug_utils/comparator/test_log_sink.py b/test/registered/debug_utils/comparator/test_log_sink.py index 348ea168a..66c054f52 100644 --- a/test/registered/debug_utils/comparator/test_log_sink.py +++ b/test/registered/debug_utils/comparator/test_log_sink.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _make_error_log(**overrides) -> ErrorLog: diff --git a/test/registered/debug_utils/comparator/test_manually_verify.py b/test/registered/debug_utils/comparator/test_manually_verify.py index 439f1cfa3..dd237cd00 100644 --- a/test/registered/debug_utils/comparator/test_manually_verify.py +++ b/test/registered/debug_utils/comparator/test_manually_verify.py @@ -23,7 +23,7 @@ import torch 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") _PNG_MAGIC: bytes = b"\x89PNG" diff --git a/test/registered/debug_utils/comparator/test_meta_overrider.py b/test/registered/debug_utils/comparator/test_meta_overrider.py index bd8beabad..afbacea32 100644 --- a/test/registered/debug_utils/comparator/test_meta_overrider.py +++ b/test/registered/debug_utils/comparator/test_meta_overrider.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) # ───────────────────── Unit: MetaOverrideRule ───────────────────── diff --git a/test/registered/debug_utils/comparator/test_model_validation.py b/test/registered/debug_utils/comparator/test_model_validation.py index 096559eaa..bbf22ce95 100644 --- a/test/registered/debug_utils/comparator/test_model_validation.py +++ b/test/registered/debug_utils/comparator/test_model_validation.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) class TestCheckEqualLengths: diff --git a/test/registered/debug_utils/comparator/test_output_types.py b/test/registered/debug_utils/comparator/test_output_types.py index 5eb316015..897c005ac 100644 --- a/test/registered/debug_utils/comparator/test_output_types.py +++ b/test/registered/debug_utils/comparator/test_output_types.py @@ -49,7 +49,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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) def _render_rich(renderable: object) -> str: diff --git a/test/registered/debug_utils/comparator/test_per_token_visualizer.py b/test/registered/debug_utils/comparator/test_per_token_visualizer.py index a9c9c1733..31b55d137 100644 --- a/test/registered/debug_utils/comparator/test_per_token_visualizer.py +++ b/test/registered/debug_utils/comparator/test_per_token_visualizer.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True) _PNG_MAGIC: bytes = b"\x89PNG" diff --git a/test/registered/debug_utils/comparator/test_preset.py b/test/registered/debug_utils/comparator/test_preset.py index d7f746790..5a06497c5 100644 --- a/test/registered/debug_utils/comparator/test_preset.py +++ b/test/registered/debug_utils/comparator/test_preset.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=5, suite="stage-a-cpu-only", nightly=True) class TestExpandPreset: diff --git a/test/registered/debug_utils/comparator/test_utils.py b/test/registered/debug_utils/comparator/test_utils.py index 2b74a2462..74adc6081 100644 --- a/test/registered/debug_utils/comparator/test_utils.py +++ b/test/registered/debug_utils/comparator/test_utils.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=10, suite="stage-a-cpu-only", nightly=True) class TestCalcRelDiff: diff --git a/test/registered/debug_utils/comparator/test_visualizer.py b/test/registered/debug_utils/comparator/test_visualizer.py index 8a01187bc..d498b5dd8 100644 --- a/test/registered/debug_utils/comparator/test_visualizer.py +++ b/test/registered/debug_utils/comparator/test_visualizer.py @@ -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="default", nightly=True) +register_cpu_ci(est_time=30, suite="stage-a-cpu-only", nightly=True) class TestPreprocessTensor: diff --git a/test/registered/debug_utils/comparator/testing_helpers.py b/test/registered/debug_utils/comparator/testing_helpers.py index 9af13d034..9c2b6e8ac 100644 --- a/test/registered/debug_utils/comparator/testing_helpers.py +++ b/test/registered/debug_utils/comparator/testing_helpers.py @@ -7,7 +7,10 @@ from typing import Optional from sglang.test.ci.ci_register import 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 ( diff --git a/test/registered/debug_utils/source_patcher/test_code_patcher.py b/test/registered/debug_utils/source_patcher/test_code_patcher.py index a2f0687c4..d647cef75 100644 --- a/test/registered/debug_utils/source_patcher/test_code_patcher.py +++ b/test/registered/debug_utils/source_patcher/test_code_patcher.py @@ -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.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" diff --git a/test/registered/debug_utils/source_patcher/test_dumper_integration.py b/test/registered/debug_utils/source_patcher/test_dumper_integration.py index 377f0e811..cd15b3d69 100644 --- a/test/registered/debug_utils/source_patcher/test_dumper_integration.py +++ b/test/registered/debug_utils/source_patcher/test_dumper_integration.py @@ -8,7 +8,7 @@ import yaml from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper 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" diff --git a/test/registered/debug_utils/source_patcher/test_source_editor.py b/test/registered/debug_utils/source_patcher/test_source_editor.py index 9204acd9d..616e289e0 100644 --- a/test/registered/debug_utils/source_patcher/test_source_editor.py +++ b/test/registered/debug_utils/source_patcher/test_source_editor.py @@ -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.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: diff --git a/test/registered/debug_utils/test_dump_comparator.py b/test/registered/debug_utils/test_dump_comparator.py index 9cb2e2362..82115dcdb 100644 --- a/test/registered/debug_utils/test_dump_comparator.py +++ b/test/registered/debug_utils/test_dump_comparator.py @@ -14,7 +14,7 @@ from sglang.srt.debug_utils.dump_comparator import ( from sglang.srt.debug_utils.dumper import DumperConfig, _Dumper 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 ----------------------------- diff --git a/test/registered/debug_utils/test_dump_loader.py b/test/registered/debug_utils/test_dump_loader.py index 3062f6b77..dccd15fee 100644 --- a/test/registered/debug_utils/test_dump_loader.py +++ b/test/registered/debug_utils/test_dump_loader.py @@ -15,7 +15,7 @@ from sglang.srt.debug_utils.dump_loader import ( ) 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: diff --git a/test/registered/debug_utils/test_schedule_simulator.py b/test/registered/debug_utils/test_schedule_simulator.py index 0366ec5d1..b4c5d7dfa 100644 --- a/test/registered/debug_utils/test_schedule_simulator.py +++ b/test/registered/debug_utils/test_schedule_simulator.py @@ -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.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 ==================== diff --git a/test/registered/function_call/test_kimik2_detector.py b/test/registered/function_call/test_kimik2_detector.py index 2d89c2af2..a2ec93ea9 100644 --- a/test/registered/function_call/test_kimik2_detector.py +++ b/test/registered/function_call/test_kimik2_detector.py @@ -11,7 +11,7 @@ from sglang.srt.function_call.kimik2_detector import ( from sglang.srt.parser.reasoning_parser import KimiK2Detector as KimiK2ReasoningDetector 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): diff --git a/test/registered/unit/function_call/test_function_call_parser.py b/test/registered/unit/function_call/test_function_call_parser.py index 23f1a163a..0ba19cc13 100644 --- a/test/registered/unit/function_call/test_function_call_parser.py +++ b/test/registered/unit/function_call/test_function_call_parser.py @@ -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.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): diff --git a/test/registered/unit/function_call/test_glm47_moe_detector.py b/test/registered/unit/function_call/test_glm47_moe_detector.py index c9e4e8d89..788580177 100644 --- a/test/registered/unit/function_call/test_glm47_moe_detector.py +++ b/test/registered/unit/function_call/test_glm47_moe_detector.py @@ -10,7 +10,7 @@ from sglang.srt.function_call.glm47_moe_detector import ( ) 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): diff --git a/test/registered/unit/function_call/test_json_schema_constraint.py b/test/registered/unit/function_call/test_json_schema_constraint.py index 6a7131cac..444864897 100644 --- a/test/registered/unit/function_call/test_json_schema_constraint.py +++ b/test/registered/unit/function_call/test_json_schema_constraint.py @@ -18,7 +18,7 @@ from sglang.srt.function_call.utils import ( ) 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): diff --git a/test/registered/unit/function_call/test_parallel_tool_calls.py b/test/registered/unit/function_call/test_parallel_tool_calls.py index 34aa48410..8019065df 100644 --- a/test/registered/unit/function_call/test_parallel_tool_calls.py +++ b/test/registered/unit/function_call/test_parallel_tool_calls.py @@ -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.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): diff --git a/test/registered/unit/function_call/test_unknown_tool_name.py b/test/registered/unit/function_call/test_unknown_tool_name.py index 79a50f88a..922d5f61b 100644 --- a/test/registered/unit/function_call/test_unknown_tool_name.py +++ b/test/registered/unit/function_call/test_unknown_tool_name.py @@ -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.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): diff --git a/test/registered/unit/observability/test_cpu_monitor.py b/test/registered/unit/observability/test_cpu_monitor.py index 2d2fb2340..fe84c0413 100644 --- a/test/registered/unit/observability/test_cpu_monitor.py +++ b/test/registered/unit/observability/test_cpu_monitor.py @@ -3,7 +3,7 @@ import unittest 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): diff --git a/test/registered/unit/utils/test_patch_tokenizer.py b/test/registered/unit/utils/test_patch_tokenizer.py index 669318141..824b65959 100644 --- a/test/registered/unit/utils/test_patch_tokenizer.py +++ b/test/registered/unit/utils/test_patch_tokenizer.py @@ -10,7 +10,7 @@ from sglang.srt.utils.patch_tokenizer import ( ) 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): diff --git a/test/registered/utils/test_log_utils.py b/test/registered/utils/test_log_utils.py index 551f97f55..cfefe8f43 100644 --- a/test/registered/utils/test_log_utils.py +++ b/test/registered/utils/test_log_utils.py @@ -9,7 +9,7 @@ from pathlib import Path from sglang.srt.utils.log_utils import create_log_targets, log_json 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): diff --git a/test/run_suite.py b/test/run_suite.py index aba22bafd..5b5d1e1c6 100644 --- a/test/run_suite.py +++ b/test/run_suite.py @@ -22,7 +22,7 @@ HW_MAPPING = { # Per-commit test suites (run on every PR) PER_COMMIT_SUITES = { - HWBackend.CPU: ["default", "stage-a-cpu-only"], + HWBackend.CPU: ["stage-a-cpu-only"], HWBackend.AMD: [ "stage-a-test-1-amd", "stage-b-test-small-1-gpu-amd",