From ce453fa43b4a80f0d76e1e3d7497f6bf504c484c Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Tue, 6 Jan 2026 20:31:52 -0800 Subject: [PATCH] Migrate backends tests to CI registration system (#16468) --- test/{srt => registered/backends}/test_torch_compile.py | 4 ++++ test/srt/run_suite.py | 2 -- 2 files changed, 4 insertions(+), 2 deletions(-) rename test/{srt => registered/backends}/test_torch_compile.py (91%) diff --git a/test/srt/test_torch_compile.py b/test/registered/backends/test_torch_compile.py similarity index 91% rename from test/srt/test_torch_compile.py rename to test/registered/backends/test_torch_compile.py index 36815718a..5ba812bd9 100644 --- a/test/srt/test_torch_compile.py +++ b/test/registered/backends/test_torch_compile.py @@ -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 diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 5f43c5c85..b3993bf27 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -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),