[CI] Key scheduled CUDA suites by runner_config instead of hand-written jobs (#34186)

2026-08-09 16:44:53 -07:00
parent 4a5d7d3c67
commit 7c90840bad
94 changed files with 531 additions and 869 deletions
@@ -128,8 +128,11 @@ def compute_partitions(
    in-source `est_time` / `(1.0, 0.0)`.
    `full_parallel=True` lifts the matrix-fanout throttle.
    """
-    # Allowlist: stages pr-test.yml dispatches. Stress / weekly /
-    # nightly-* live in test/registered/ but pr-test doesn't run them.
+    # Allowlist of the stages this workflow dispatches -- what keeps stress /
+    # weekly / nightly out, since CUDA scheduled suites no longer carry
+    # `nightly=True`. The nightly filter still matters for CPU: some tests sit on
+    # a dispatched suite with the flag set, so run_suite.py skips them and their
+    # est_time must not inflate the shard count.
    dispatched_suites = set(run_timeouts) | set(_BASE_A_OVERRIDES)
    suite_tests = defaultdict(list)
    for t in tests: