Skip to content

Commit d49a0d3

Browse files
committed
Add CUDA 12.9 builds to test channel and aarch64 matrix
Add CUDA 12.9 to the test channel in CUDA_ARCHES_DICT (for validating release candidate binaries) and add 12.9-aarch64 to CUDA_AARCH64_ARCHES. Also adds the corresponding CUDA_CUDNN_VERSIONS entry (12.9.1 / cudnn 9), matching how 12.9 was previously configured. Reference test assets are regenerated via 'python -m tools.tests.test_generate_binary_build_matrix --update-reference-files'. This regeneration also picks up a pre-existing staleness on this branch: the assets had stable_version 2.11.0 while the script already declares CURRENT_STABLE_VERSION = 2.12.0, which had left the 7 matrix tests red. The regenerated assets now pass.
1 parent b9e711f commit d49a0d3

8 files changed

Lines changed: 192 additions & 191 deletions

tools/scripts/generate_binary_build_matrix.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
}
3737
CUDA_ARCHES_DICT = {
3838
"nightly": ["12.6", "13.0", "13.2"],
39-
"test": ["12.6", "13.0", "13.2"],
39+
"test": ["12.6", "12.9", "13.0", "13.2"],
4040
"release": ["12.6", "12.8", "13.0"],
4141
}
4242

@@ -49,6 +49,7 @@
4949
CUDA_CUDNN_VERSIONS = {
5050
"12.6": {"cuda": "12.6.3", "cudnn": "9"},
5151
"12.8": {"cuda": "12.8.0", "cudnn": "9"},
52+
"12.9": {"cuda": "12.9.1", "cudnn": "9"},
5253
"13.0": {"cuda": "13.0.0", "cudnn": "9"},
5354
"13.2": {"cuda": "13.2.0", "cudnn": "9"},
5455
}
@@ -59,7 +60,7 @@
5960
"release": "13.0",
6061
}
6162

62-
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "13.0-aarch64", "13.2-aarch64"]
63+
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "12.9-aarch64", "13.0-aarch64", "13.2-aarch64"]
6364

6465
PACKAGE_TYPES = ["wheel", "libtorch"]
6566
CXX11_ABI = "cxx11-abi"

0 commit comments

Comments
 (0)