Skip to content

Commit 453f19e

Browse files
authored
Fix libtorch nightly binaries links (#6972)
Fixes pytorch/pytorch#159880
1 parent bb05b79 commit 453f19e

File tree

1 file changed

+21
-4
lines changed

1 file changed

+21
-4
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,16 @@ def get_libtorch_install_command(
238238
if libtorch_config == "debug"
239239
else libtorch_variant
240240
)
241-
build_name = f"{prefix}-{_libtorch_variant}-latest.zip"
241+
242+
# Temporary until we release 2.8.0
243+
if (
244+
CURRENT_STABLE_VERSION == "2.7.1"
245+
and channel == RELEASE
246+
and devtoolset == "cxx11-abi"
247+
):
248+
build_name = f"{prefix}-{devtoolset}-{_libtorch_variant}-latest.zip"
249+
else:
250+
build_name = f"{prefix}-{_libtorch_variant}-latest.zip"
242251

243252
if os == MACOS_ARM64:
244253
arch = "arm64"
@@ -247,9 +256,17 @@ def get_libtorch_install_command(
247256
build_name = f"libtorch-macos-{arch}-{CURRENT_VERSION}.zip"
248257

249258
elif os == LINUX and (channel in (RELEASE, TEST)):
250-
build_name = (
251-
f"{prefix}-{_libtorch_variant}-{CURRENT_VERSION}%2B{desired_cuda}.zip"
252-
)
259+
if (
260+
CURRENT_STABLE_VERSION == "2.7.1"
261+
and channel == RELEASE
262+
and devtoolset == "cxx11-abi"
263+
):
264+
build_name = f"{prefix}-{devtoolset}-{_libtorch_variant}-{CURRENT_VERSION}%2B{desired_cuda}.zip"
265+
else:
266+
build_name = (
267+
f"{prefix}-{_libtorch_variant}-{CURRENT_VERSION}%2B{desired_cuda}.zip"
268+
)
269+
253270
elif os == WINDOWS and (channel in (RELEASE, TEST)):
254271
build_name = (
255272
f"{prefix}-shared-with-deps-debug-{CURRENT_VERSION}%2B{desired_cuda}.zip"

0 commit comments

Comments
 (0)