Add 5090 dry run stage to PR test workflow (#17022)

2026-01-13 14:12:33 -08:00
parent 339915ce2b
commit b880607108
71 changed files with 158 additions and 3 deletions
@@ -227,6 +227,7 @@ def handle_rerun_stage(
        "stage-a-cpu-only",
        "stage-b-test-small-1-gpu",
        "stage-b-test-large-1-gpu",
+        "stage-b-test-small-1-gpu-5090",
        "stage-b-test-large-2-gpu",
        "stage-c-test-large-4-gpu",
        "stage-c-test-large-4-gpu-b200",