Migrate sampling tests to test/registered/sampling/ (#16455)
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user