chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Migrate sampling tests to test/registered/sampling/ (#16455)

Browse Source
This commit is contained in:
Alison Shao
2026-01-05 23:43:20 -08:00
committed by GitHub
parent 5e5b1183ed
commit 176266f358
4 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/srt/test_original_logprobs.py → test/registered/sampling/test_original_logprobs.py
View File

@@ -14,6 +14,11 @@ The test covers the following scenarios:
"""
import os
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
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")
import random
import unittest

4
test/srt/test_penalty.py → test/registered/sampling/test_penalty.py
View File

@@ -7,6 +7,10 @@ from concurrent.futures import ThreadPoolExecutor
import requests
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=82, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=180, suite="stage-b-test-small-1-gpu")
from sglang.test.test_utils import (
DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,

4
test/srt/test_pytorch_sampling_backend.py → test/registered/sampling/test_pytorch_sampling_backend.py
View File

@@ -4,6 +4,10 @@ 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
register_cuda_ci(est_time=66, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=66, suite="stage-b-test-small-1-gpu")
from sglang.test.run_eval import run_eval
from sglang.test.test_utils import (
DEFAULT_MODEL_NAME_FOR_TEST,

6
test/srt/run_suite.py
View File

@@ -48,12 +48,9 @@ suites = {
TestFile("test_multi_tokenizer.py", 230),
TestFile("test_no_chunked_prefill.py", 108),
TestFile("test_no_overlap_scheduler.py", 217),
TestFile("test_original_logprobs.py", 41),
TestFile("test_page_size.py", 60),
TestFile("test_penalty.py", 82),
TestFile("test_prefill_adder.py", 1),
TestFile("test_priority_scheduling.py", 130),
TestFile("test_pytorch_sampling_backend.py", 66),
TestFile("test_request_queue_validation.py", 47),
TestFile("test_retract_decode.py", 259),
TestFile("test_score_api.py", 260),
@@ -189,14 +186,11 @@ suite_amd = {
TestFile("test_model_hooks.py", 10),
TestFile("test_multi_tokenizer.py", 345),
TestFile("test_no_chunked_prefill.py", 108),
TestFile("test_original_logprobs.py", 60),
TestFile("test_page_size.py", 60),
TestFile("test_penalty.py", 180),
TestFile("test_prefill_adder.py", 2),
TestFile("test_priority_scheduling.py", 195),
TestFile("test_profile_merger.py", 12),
TestFile("test_profile_merger_http_api.py", 15),
TestFile("test_pytorch_sampling_backend.py", 66),
TestFile("test_request_queue_validation.py", 70),
TestFile("test_retract_decode.py", 450),
TestFile("test_rope_rocm.py", 3),