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

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