[CI] Migrate CUDA Graph tests to test/registered/cuda_graph/ (#15436)

This commit is contained in:
Alison Shao
2025-12-20 18:55:39 -08:00
committed by GitHub
parent 4b351f6b95
commit 9a3bdf2c95
8 changed files with 208 additions and 159 deletions
-3
View File
@@ -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": [