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

2025-12-20 18:55:39 -08:00
parent 4b351f6b95
commit 9a3bdf2c95
8 changed files with 208 additions and 159 deletions
--- a/scripts/ci/slash_command_handler.py
+++ b/scripts/ci/slash_command_handler.py
@@ -144,7 +144,8 @@ def handle_rerun_stage(
    nvidia_stages = [
        "stage-a-test-1",
        "stage-b-test-small-1-gpu",
-        "stage-b-test-2-gpu",
+        "stage-b-test-large-1-gpu",
+        "stage-b-test-large-2-gpu",
        "multimodal-gen-test-1-gpu",
        "multimodal-gen-test-2-gpu",
        "quantization-test",