[CI] Migrate CUDA Graph tests to test/registered/cuda_graph/ (#15436)
This commit is contained in:
@@ -88,8 +88,6 @@ suites = {
|
||||
TestFile("test_original_logprobs.py", 41),
|
||||
TestFile("test_page_size.py", 60),
|
||||
TestFile("test_penalty.py", 82),
|
||||
TestFile("test_piecewise_cuda_graph_1_gpu_a.py", 460),
|
||||
TestFile("test_piecewise_cuda_graph_1_gpu_b.py", 480),
|
||||
TestFile("test_priority_scheduling.py", 130),
|
||||
TestFile("test_pytorch_sampling_backend.py", 66),
|
||||
TestFile("test_radix_attention.py", 105),
|
||||
@@ -140,7 +138,6 @@ suites = {
|
||||
TestFile("test_dp_attention.py", 350),
|
||||
TestFile("test_load_weights_from_remote_instance.py", 72),
|
||||
TestFile("test_patch_torch.py", 19),
|
||||
TestFile("test_piecewise_cuda_graph_2_gpu.py", 400),
|
||||
TestFile("test_eagle_dp_attention.py", 200),
|
||||
],
|
||||
"per-commit-4-gpu": [
|
||||
|
||||
Reference in New Issue
Block a user