You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
Commit f0e8589
Browse filesBrowse the repository at this point in the historyBrowse files
## Summary
PyTorch no longer publishes cu126 nightlies. The torchao cu126 channel
stops at `0.19.0.dev20260907`, so the 12.6 CI rows can't install the
current pins. This removes CUDA 12.6 support.
**Support**
- `install_utils.py`: remove `(12, 6)` from `SUPPORTED_CUDA_VERSIONS`.
Supported versions are now 13.0, 13.2, 13.4.
**CI**
- `.github/workflows/vulkan.yml`: move `test-vulkan-nvidia` and the
Vulkan LLM e2e job from the `cuda12.6` image to `cuda13.2`. The
`pytorch/almalinux-builder:cuda13.2` image exists and is current.
- `.github/workflows/cuda.yml`: drop `12.6` from the `test-cuda-builds`
matrix. Coverage that remains:
- build matrix rows for 13.0 and 13.4
- 13.0 on every model e2e / benchmark job (10+ jobs)
- 13.2 on the Windows e2e jobs (`cuda-windows.yml`) and the published
wheel rows (`filter_cuda_matrix.py`: cu132 / cu134)
- `.ci/scripts/wheel/cuda_arch_list.sh`: remove the cu126 GPU arch
lists. A cu126 row now fails as an unknown train.
- `.ci/docker/common/install_cuda_windows_cross_compile.sh`: remove the
12.6 entry.
- `.ci/scripts/tests/test_cuda_workflow.py`,
`.ci/scripts/tests/test_cu134_dependencies.py`: drop the 12.6 cases.
**Docs / comments**
- `docs/source/backends/cuda/cuda-overview.md`: the source build now
lists CUDA 13.0 instead of 12.6.
- Update the 12.6 / cu126 examples in `install_utils.py`, `setup.py`,
`filter_cuda_matrix.py`, `test_shared_libraries.py`,
`cuda_benchmark.py`.
**Intentionally unchanged**
- `backends/cuda/cuda_backend.py` `_setup_cuda_environment_for_fatbin`:
the CUDA 12 ptxas branch handles any user torch build on CUDA 12.x below
12.9, not just a 12.6 support row.
- `setup.py` / `test_shared_libraries.py` CUDA 12 runtime package
mapping (`nvidia-cuda-runtime-cu12`): this is keyed by CUDA major, not
12.6.
- Windows cross-compile entries for 12.8 / 12.9.
## Test plan
- `pytest .ci/scripts/tests/test_cu134_dependencies.py
.ci/scripts/tests/test_cuda_workflow.py
.ci/scripts/tests/test_filter_cuda_matrix.py`: 49 passed
- `executorch_cuda_arch_list`: cu130/cu132/cu134 still resolve, cu126
now returns 64 (unknown train)
- Both workflow YAMLs parse; flake8 clean
- Repo-wide grep for `12.6` / `cu126` / `(12, 6)`: two hits left, both
kept on purpose: the `nvidia-cuda-runtime-cu12==12.6.77` example in a
parsing comment, and the `CUDA 12.6+` comment on the unchanged ptxas
branch in `cuda_backend.py`
- CI (Vulkan real-GPU jobs on the 13.2 image)
---------
Co-authored-by: Digant Desai <digantdesai@meta.com>
0 commit comments