diff --git a/test/srt/test_vision_openai_server_common.py b/python/sglang/test/vlm_utils.py similarity index 100% rename from test/srt/test_vision_openai_server_common.py rename to python/sglang/test/vlm_utils.py diff --git a/test/srt/test_vision_chunked_prefill.py b/test/registered/vlm/test_vision_chunked_prefill.py similarity index 98% rename from test/srt/test_vision_chunked_prefill.py rename to test/registered/vlm/test_vision_chunked_prefill.py index 7236bb2eb..f00d08d40 100644 --- a/test/srt/test_vision_chunked_prefill.py +++ b/test/registered/vlm/test_vision_chunked_prefill.py @@ -1,3 +1,7 @@ +from sglang.test.ci.ci_register import register_cuda_ci + +register_cuda_ci(est_time=150, suite="stage-b-test-small-1-gpu") + """ Usage: python3 -m unittest test_vision_chunked_prefill.TestVisionChunkedPrefill.test_chunked_prefill diff --git a/test/srt/test_vision_openai_server_a.py b/test/registered/vlm/test_vision_openai_server_a.py similarity index 97% rename from test/srt/test_vision_openai_server_a.py rename to test/registered/vlm/test_vision_openai_server_a.py index f9fb12466..fb57347a0 100644 --- a/test/srt/test_vision_openai_server_a.py +++ b/test/registered/vlm/test_vision_openai_server_a.py @@ -1,3 +1,7 @@ +from sglang.test.ci.ci_register import register_cuda_ci + +register_cuda_ci(est_time=890, suite="stage-b-test-small-1-gpu") + """ Usage: python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_mixed_batch @@ -7,8 +11,9 @@ python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_multi_ import unittest import openai -from test_vision_openai_server_common import * -from test_vision_openai_server_common import ( # DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,; DEFAULT_URL_FOR_TEST,; IMAGE_MAN_IRONING_URL,; popen_launch_server, + +from sglang.test.vlm_utils import * +from sglang.test.vlm_utils import ( AudioOpenAITestMixin, CustomTestCase, ImageOpenAITestMixin, diff --git a/test/srt/test_vlm_input_format.py b/test/registered/vlm/test_vlm_input_format.py similarity index 99% rename from test/srt/test_vlm_input_format.py rename to test/registered/vlm/test_vlm_input_format.py index 26242ee19..f97ea5c84 100644 --- a/test/srt/test_vlm_input_format.py +++ b/test/registered/vlm/test_vlm_input_format.py @@ -1,3 +1,7 @@ +from sglang.test.ci.ci_register import register_cuda_ci + +register_cuda_ci(est_time=430, suite="stage-b-test-small-1-gpu") + import json import unittest from io import BytesIO diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index c82a53f98..2b9456849 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -76,9 +76,6 @@ suites = { TestFile("test_torchao.py", 103), TestFile("test_utils_update_weights.py", 29), TestFile("test_video_utils.py", 5), - TestFile("test_vision_chunked_prefill.py", 150), - TestFile("test_vision_openai_server_a.py", 778), - TestFile("test_vlm_input_format.py", 166), TestFile("test_modelopt_export.py", 9), ], "per-commit-2-gpu": [ @@ -157,7 +154,6 @@ suites = { TestFile("test_eval_accuracy_large.py"), TestFile("test_gpt_oss_common.py"), TestFile("test_moe_eval_accuracy_large.py"), - TestFile("test_vision_openai_server_common.py"), TestFile("test_profile_v2.py"), TestFile("models/test_ministral3_models.py"), TestFile("test_mistral_large3_basic.py"),