Skip to content

Commit f93982b

Browse files
committed
Don't advertise CUDA 13.4 on the getting-started page
`update-quick-start-module.yml` in pytorch/pytorch.github.io calls `generate_binary_build_matrix.yml@main` with `getting-started: true` for linux/windows/macos-arm64 on the nightly and release channels, and publishes the result as the pytorch.org install selector. Because 13.4 is in `CUDA_ARCHES_DICT["nightly"]`, that bot has started adding a `cu134` option -- see pytorch/pytorch.github.io#2130, which introduces `pip3 install --pre torch torchvision --index-url .../nightly/cu134`. 13.4 is built and validated but is not ready to be offered to users, so gate it out of the getting-started matrix only. `initialize_globals` already took a `getting_started` argument that was never used; this gives it a purpose. The filter follows the existing `CUDA_ARCHES_NO_WINDOWS` convention right above it. To re-enable 13.4 on the page later, drop it from `CUDA_ARCHES_NO_GETTING_STARTED` -- no other change needed. Note `CUDA_AARCH64_ARCHES` is deliberately untouched: the getting-started workflow never requests `linux-aarch64`, and that list is a module constant with no per-channel dict behind it, so filtering it in place would leak into later calls in the same process. Test plan: - `python -m tools.tests.test_generate_binary_build_matrix` - 11 tests pass. - `--update-reference-files` produces no asset changes. - Getting-started matrix (`--getting-started true`), cu134 entry count, before -> after, across all six jobs the quick-start workflow runs: - linux/nightly: 2 -> 0 (the wheel + libtorch entries #2130 was adding) - linux/release, windows/*, macos-arm64/*: 0 -> 0 (unchanged) - Non-getting-started matrices are byte-identical (md5) for {linux, linux-aarch64, windows, macos-arm64} x {nightly, test, release}, so builds and binary validation are unaffected. - Verified the filter does not leak across calls: after a `getting_started=True` call, a subsequent default call still sees `['12.6', '13.0', '13.2', '13.4']`. Authored with the assistance of Claude Code.
1 parent d607479 commit f93982b

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

tools/scripts/generate_binary_build_matrix.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,13 @@
7575
# CUDA_ARCHES_NO_WINDOWS in pytorch/pytorch.
7676
CUDA_ARCHES_NO_WINDOWS = ["13.4"]
7777

78+
# CUDA versions that are built and validated, but must not be advertised on the
79+
# pytorch.org getting-started page yet. Without this, update-quick-start-module
80+
# in pytorch/pytorch.github.io publishes an install selector for a CUDA version
81+
# users cannot rely on. Only affects the getting-started matrix; nightly, test
82+
# and release builds and their validation are unchanged.
83+
CUDA_ARCHES_NO_GETTING_STARTED = ["13.4"]
84+
7885
PACKAGE_TYPES = ["wheel", "libtorch"]
7986
CXX11_ABI = "cxx11-abi"
8087
RELEASE = "release"
@@ -181,6 +188,10 @@ def initialize_globals(
181188
CURRENT_VERSION = CURRENT_STABLE_VERSION
182189

183190
CUDA_ARCHES = CUDA_ARCHES_DICT[channel]
191+
if getting_started:
192+
CUDA_ARCHES = [
193+
arch for arch in CUDA_ARCHES if arch not in CUDA_ARCHES_NO_GETTING_STARTED
194+
]
184195
ROCM_ARCHES = ROCM_ARCHES_DICT[channel]
185196
if build_python_only:
186197
# Only select the oldest version of python if building a python only package

0 commit comments

Comments
 (0)