[CI] Migrate Eagle 1-GPU tests to test/registered/ (#14529)

test/registered/spec/eagle/test_eagle_constrained_decoding.py

@@ -2,6 +2,7 @@ import unittest
 
 from sglang.srt.environ import envs
 from sglang.srt.utils import kill_process_tree
+from sglang.test.ci.ci_register import register_cuda_ci
 from sglang.test.kits.json_constrained_kit import TestJSONConstrainedMixin
 from sglang.test.kits.regex_constrained_kit import TestRegexConstrainedMixin
 from sglang.test.test_utils import (
@@ -13,6 +14,8 @@ from sglang.test.test_utils import (
     popen_launch_server,
 )
 
+register_cuda_ci(est_time=100, suite="stage-b-test-small-1-gpu")
+
 
 class TestEagleConstrainedDecoding(
     CustomTestCase, TestRegexConstrainedMixin, TestJSONConstrainedMixin

test/registered/spec/eagle/test_eagle_infer_a.py

@@ -7,6 +7,7 @@ import torch
 import sglang as sgl
 from sglang.srt.utils import kill_process_tree
 from sglang.srt.utils.hf_transformers_utils import get_tokenizer
+from sglang.test.ci.ci_register import register_cuda_ci
 from sglang.test.test_utils import (
     DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
     DEFAULT_EAGLE_TARGET_MODEL_FOR_TEST,
@@ -20,6 +21,8 @@ from sglang.test.test_utils import (
     popen_launch_server,
 )
 
+register_cuda_ci(est_time=470, suite="stage-b-test-small-1-gpu")
+
 torch_dtype = torch.float16
 prefill_tolerance = 5e-2
 decode_tolerance: float = 5e-2

test/registered/spec/eagle/test_eagle_infer_b.py

@@ -12,6 +12,7 @@ import numpy as np
 import requests
 
 from sglang.srt.utils import kill_process_tree
+from sglang.test.ci.ci_register import register_cuda_ci
 from sglang.test.few_shot_gsm8k import run_eval
 from sglang.test.test_utils import (
     DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
@@ -23,6 +24,8 @@ from sglang.test.test_utils import (
     run_logprob_check,
 )
 
+register_cuda_ci(est_time=473, suite="stage-b-test-small-1-gpu")
+
 
 class TestEAGLEServer(CustomTestCase):
     PROMPTS = [

test/registered/spec/eagle/test_eagle_infer_beta.py

@@ -3,6 +3,7 @@ from types import SimpleNamespace
 
 from sglang.srt.environ import envs
 from sglang.srt.utils import kill_process_tree
+from sglang.test.ci.ci_register import register_cuda_ci
 from sglang.test.few_shot_gsm8k import run_eval
 from sglang.test.kits.matched_stop_kit import MatchedStopMixin
 from sglang.test.kits.radix_cache_server_kit import run_radix_attention_test
@@ -15,6 +16,8 @@ from sglang.test.test_utils import (
     popen_launch_server,
 )
 
+register_cuda_ci(est_time=194, suite="stage-b-test-small-1-gpu")
+
 
 class TestEagleServerBase(CustomTestCase, MatchedStopMixin):
     max_running_requests = 64

test/registered/spec/utils/test_build_eagle_tree.py

@@ -6,6 +6,9 @@ from sglang.srt.speculative.eagle_utils import (
     build_tree_kernel_efficient,
     organize_draft_results,
 )
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=3, suite="stage-b-test-small-1-gpu")
 
 
 class TestBuildEagleTree(unittest.TestCase):

test/run_suite.py

@@ -17,7 +17,7 @@ HW_MAPPING = {
 PER_COMMIT_SUITES = {
     HWBackend.CPU: ["default"],
     HWBackend.AMD: ["stage-a-test-1"],
-    HWBackend.CUDA: ["stage-a-test-1"],
+    HWBackend.CUDA: ["stage-a-test-1", "stage-b-test-small-1-gpu"],
     HWBackend.NPU: [],
 }
 

test/srt/run_suite.py

@@ -7,7 +7,6 @@ from sglang.test.ci.ci_utils import TestFile, run_unittest_files
 # NOTE: please sort the test cases alphabetically by the test file name
 suites = {
     "per-commit-1-gpu": [
-        TestFile("test_eagle_constrained_decoding.py", 100),
         TestFile("debug_utils/test_tensor_dump_forward_hook.py", 9),
         TestFile("hicache/test_hicache_storage.py", 96),
         TestFile("hicache/test_hicache_variants.py", 368),
@@ -58,13 +57,9 @@ suites = {
         # TestFile("rl/test_update_weights_from_disk.py", 210),  # Temporarily disabled, see https://github.com/sgl-project/sglang/pull/13998
         TestFile("rl/test_update_weights_from_tensor.py", 195),
         TestFile("test_abort.py", 131),
-        TestFile("test_build_eagle_tree.py", 3),
         TestFile("test_chunked_prefill.py", 312),
         TestFile("test_create_kvindices.py", 7),
         TestFile("test_deterministic.py", 228),
-        TestFile("test_eagle_infer_a.py", 470),
-        TestFile("test_eagle_infer_b.py", 473),
-        TestFile("test_eagle_infer_beta.py", 194),
         TestFile("test_constrained_decoding.py", 111),
         TestFile("test_eval_fp8_accuracy.py", 250),
         TestFile("test_external_models.py", 30),