Migrate backends tests to CI registration system (#16468)

This commit is contained in:
Alison Shao
2026-01-06 20:31:52 -08:00
committed by GitHub
parent 3be1e734ee
commit ce453fa43b
2 changed files with 4 additions and 2 deletions

View File

@@ -5,6 +5,7 @@ from types import SimpleNamespace
import requests
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.run_eval import run_eval
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST,
@@ -15,6 +16,9 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=190, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=169, suite="stage-b-test-small-1-gpu")
class TestTorchCompile(CustomTestCase):
@classmethod

View File

@@ -32,7 +32,6 @@ suites = {
TestFile("test_profile_merger.py", 8),
TestFile("test_profile_merger_http_api.py", 9),
TestFile("test_swa_unittest.py", 8),
TestFile("test_torch_compile.py", 190),
TestFile("test_utils_update_weights.py", 29),
TestFile("test_video_utils.py", 5),
TestFile("test_modelopt_export.py", 9),
@@ -142,7 +141,6 @@ suite_amd = {
TestFile("test_srt_engine.py", 261),
TestFile("test_start_profile.py", 60),
TestFile("test_swa_unittest.py", 10),
TestFile("test_torch_compile.py", 169),
# TestFile("test_torch_compile_moe.py", 210), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/13107
TestFile("test_type_based_dispatcher.py", 10),
TestFile("test_video_utils.py", 8),