29
29
CUDA_ARCHES_DICT = {
30
30
"nightly" : ["12.6" , "12.8" , "12.9" ],
31
31
"test" : ["12.6" , "12.8" , "12.9" ],
32
- "release" : ["11.8 " , "12.6 " , "12.8 " ],
32
+ "release" : ["12.6 " , "12.8 " , "12.9 " ],
33
33
}
34
+
34
35
ROCM_ARCHES_DICT = {
35
36
"nightly" : ["6.3" , "6.4" ],
36
37
"test" : ["6.3" , "6.4" ],
37
- "release" : ["6.2.4 " , "6.3 " ],
38
+ "release" : ["6.3 " , "6.4 " ],
38
39
}
39
40
40
41
CUDA_CUDNN_VERSIONS = {
47
48
STABLE_CUDA_VERSIONS = {
48
49
"nightly" : "12.8" ,
49
50
"test" : "12.8" ,
50
- "release" : "12.6 " ,
51
+ "release" : "12.8 " ,
51
52
}
52
53
53
54
CUDA_AARCH64_ARCHES = ["12.9-aarch64" ]
77
78
78
79
CURRENT_NIGHTLY_VERSION = "2.9.0"
79
80
CURRENT_CANDIDATE_VERSION = "2.8.0"
80
- CURRENT_STABLE_VERSION = "2.7.1 "
81
+ CURRENT_STABLE_VERSION = "2.8.0 "
81
82
CURRENT_VERSION = CURRENT_STABLE_VERSION
82
83
83
84
# By default use Nightly for CUDA arches
@@ -228,7 +229,6 @@ def get_libtorch_install_command(
228
229
channel : str ,
229
230
gpu_arch_type : str ,
230
231
libtorch_variant : str ,
231
- devtoolset : str ,
232
232
desired_cuda : str ,
233
233
libtorch_config : str ,
234
234
) -> str :
@@ -238,16 +238,7 @@ def get_libtorch_install_command(
238
238
if libtorch_config == "debug"
239
239
else libtorch_variant
240
240
)
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"
241
+ build_name = f"{ prefix } -{ _libtorch_variant } -latest.zip"
251
242
252
243
if os == MACOS_ARM64 :
253
244
arch = "arm64"
@@ -256,16 +247,9 @@ def get_libtorch_install_command(
256
247
build_name = f"libtorch-macos-{ arch } -{ CURRENT_VERSION } .zip"
257
248
258
249
elif os == LINUX and (channel in (RELEASE , TEST )):
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
- )
250
+ build_name = (
251
+ f"{ prefix } -{ _libtorch_variant } -{ CURRENT_VERSION } %2B{ desired_cuda } .zip"
252
+ )
269
253
270
254
elif os == WINDOWS and (channel in (RELEASE , TEST )):
271
255
build_name = (
@@ -412,7 +396,6 @@ def generate_libtorch_matrix(
412
396
channel ,
413
397
gpu_arch_type ,
414
398
libtorch_variant ,
415
- devtoolset ,
416
399
desired_cuda ,
417
400
libtorch_config ,
418
401
),
0 commit comments