Skip to content

Commit 1969a06

Browse files
authored
Release 2.5 Use pypi for cuda 12.4 builds (#5776)
We will be releasing 12.4 builds to pypi. Hence use 12.4 rather then 12.1
1 parent 2b66760 commit 1969a06

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ def get_wheel_install_command(
310310
else:
311311
raise ValueError("Split build is not supported for this configuration. It is only supported for CUDA 11.8, 12.1, 12.4 on Linux nightly builds.")
312312
if channel == RELEASE and (not use_only_dl_pytorch_org) and (
313-
(gpu_arch_version == "12.1" and os == LINUX)
313+
(gpu_arch_version == "12.4" and os == LINUX)
314314
or (
315315
gpu_arch_type == CPU
316316
and os in [WINDOWS, MACOS, MACOS_ARM64]

0 commit comments

Comments
 (0)