Migrate profiling tests to test/registered/profiling/ (#16459)

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):

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):

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"
 
 

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),