Repository navigation
Drop CUDA 12.6 support - #24740
Drop CUDA 12.6 support#24740Gasoonjia wants to merge 2 commits into
Conversation
PyTorch no longer publishes cu126 nightlies (torchao's cu126 channel stops at dev20260907), so the 12.6 CI rows can no longer install their pinned deps. - install_utils: remove (12, 6) from SUPPORTED_CUDA_VERSIONS. - cuda.yml: drop 12.6 from the build matrix (13.0 and 13.4 remain, and 13.0 is used by every model e2e job). - vulkan.yml: move both real-GPU jobs from the 12.6 image to 13.2. - Remove the cu126 GPU arch lists, the 12.6 Windows cross-compile mapping, and 12.6 cases in CI unit tests. - Docs/comments: drop 12.6 mentions.
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/24740
Note: Links to docs will display an error until the docs builds have been completed. ❌ 2 Cancelled Jobs, 4 Unrelated Failures, 2 Unclassified FailuresAs of commit 0cc0ce7 with merge base 97971ca ( UNCLASSIFIED FAILURES - DrCI could not classify the following jobs because the workflow did not run on the merge base. The failures may be pre-existing on trunk or introduced by this PR:
CANCELLED JOBS - The following jobs were cancelled. Please retry:
BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
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)fromSUPPORTED_CUDA_VERSIONS. Supported versions are now 13.0, 13.2, 13.4.CI
.github/workflows/vulkan.yml: movetest-vulkan-nvidiaand the Vulkan LLM e2e job from thecuda12.6image tocuda13.2. Thepytorch/almalinux-builder:cuda13.2image exists and is current..github/workflows/cuda.yml: drop12.6from thetest-cuda-buildsmatrix. Coverage that remains: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.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.pyCUDA 12 runtime package mapping (nvidia-cuda-runtime-cu12): this is keyed by CUDA major, not 12.6.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 passedexecutorch_cuda_arch_list: cu130/cu132/cu134 still resolve, cu126 now returns 64 (unknown train)12.6/cu126/(12, 6): two hits left, both kept on purpose: thenvidia-cuda-runtime-cu12==12.6.77example in a parsing comment, and theCUDA 12.6+comment on the unchanged ptxas branch incuda_backend.py