From 52de807dd7670d403810598f0a73d1c10988c62f Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Tue, 6 Jan 2026 21:04:48 -0800
Subject: [PATCH] Migrate profiling tests to test/registered/profiling/
 (#16459)

---
 test/{srt => registered/profiling}/test_profile_merger.py   | 4 ++++
 .../profiling}/test_profile_merger_http_api.py              | 4 ++++
 test/{srt => registered/profiling}/test_start_profile.py    | 4 ++++
 test/srt/run_suite.py                                       | 6 ------
 4 files changed, 12 insertions(+), 6 deletions(-)
 rename test/{srt => registered/profiling}/test_profile_merger.py (98%)
 rename test/{srt => registered/profiling}/test_profile_merger_http_api.py (96%)
 rename test/{srt => registered/profiling}/test_start_profile.py (98%)

diff --git a/test/srt/test_profile_merger.py b/test/registered/profiling/test_profile_merger.py
similarity index 98%
rename from test/srt/test_profile_merger.py
rename to test/registered/profiling/test_profile_merger.py
index 2a88b1915..412ae8046 100644
--- a/test/srt/test_profile_merger.py
+++ b/test/registered/profiling/test_profile_merger.py
@@ -15,6 +15,10 @@ import unittest
 
 from sglang.srt.managers.io_struct import ProfileReq, ProfileReqInput, ProfileReqType
 from sglang.srt.utils.profile_merger import ProfileMerger
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
+
+register_cuda_ci(est_time=8, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=8, suite="stage-b-test-small-1-gpu-amd")
 
 
 class TestProfileMerger(unittest.TestCase):
diff --git a/test/srt/test_profile_merger_http_api.py b/test/registered/profiling/test_profile_merger_http_api.py
similarity index 96%
rename from test/srt/test_profile_merger_http_api.py
rename to test/registered/profiling/test_profile_merger_http_api.py
index e293dae92..9bf656f26 100644
--- a/test/srt/test_profile_merger_http_api.py
+++ b/test/registered/profiling/test_profile_merger_http_api.py
@@ -2,6 +2,10 @@ import json
 import unittest
 
 from sglang.srt.managers.io_struct import ProfileReqInput
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
+
+register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=9, suite="stage-b-test-small-1-gpu-amd")
 
 
 class TestProfileMergerHTTPAPI(unittest.TestCase):
diff --git a/test/srt/test_start_profile.py b/test/registered/profiling/test_start_profile.py
similarity index 98%
rename from test/srt/test_start_profile.py
rename to test/registered/profiling/test_start_profile.py
index 5c235bf6f..1c1acc6cc 100644
--- a/test/srt/test_start_profile.py
+++ b/test/registered/profiling/test_start_profile.py
@@ -20,6 +20,7 @@ import requests
 
 from sglang.srt.environ import envs
 from sglang.srt.utils import kill_process_tree
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.test_utils import (
     DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -28,6 +29,9 @@ from sglang.test.test_utils import (
     popen_launch_server,
 )
 
+register_cuda_ci(est_time=41, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=60, suite="stage-b-test-small-1-gpu")
+
 OUTPUT_DIR = "./profiler_dir"
 
 
diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index a8a1722b7..e3dd9befb 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -27,9 +27,6 @@ suites = {
         TestFile("test_skip_tokenizer_init.py", 77),
         TestFile("test_srt_endpoint.py", 127),
         TestFile("test_srt_engine.py", 252),
-        TestFile("test_start_profile.py", 41),
-        TestFile("test_profile_merger.py", 8),
-        TestFile("test_profile_merger_http_api.py", 9),
         TestFile("test_utils_update_weights.py", 29),
         TestFile("test_video_utils.py", 5),
         TestFile("test_modelopt_export.py", 9),
@@ -129,15 +126,12 @@ suite_amd = {
         TestFile("test_model_hooks.py", 10),
         TestFile("test_multi_tokenizer.py", 345),
         TestFile("test_page_size.py", 60),
-        TestFile("test_profile_merger.py", 12),
-        TestFile("test_profile_merger_http_api.py", 15),
         TestFile("test_request_queue_validation.py", 70),
         TestFile("test_rope_rocm.py", 3),
         TestFile("test_server_args.py", 1),
         TestFile("test_skip_tokenizer_init.py", 117),
         TestFile("test_srt_endpoint.py", 130),
         TestFile("test_srt_engine.py", 261),
-        TestFile("test_start_profile.py", 60),
         # 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),