2929CUDA_ARCHES_DICT = {
3030 "nightly" : ["12.6" , "12.8" , "12.9" ],
3131 "test" : ["12.6" , "12.8" , "12.9" ],
32- "release" : ["11.8 " , "12.6 " , "12.8 " ],
32+ "release" : ["12.6 " , "12.8 " , "12.9 " ],
3333}
34+
3435ROCM_ARCHES_DICT = {
3536 "nightly" : ["6.3" , "6.4" ],
3637 "test" : ["6.3" , "6.4" ],
37- "release" : ["6.2.4 " , "6.3 " ],
38+ "release" : ["6.3 " , "6.4 " ],
3839}
3940
4041CUDA_CUDNN_VERSIONS = {
4748STABLE_CUDA_VERSIONS = {
4849 "nightly" : "12.8" ,
4950 "test" : "12.8" ,
50- "release" : "12.6 " ,
51+ "release" : "12.8 " ,
5152}
5253
5354CUDA_AARCH64_ARCHES = ["12.9-aarch64" ]
7778
7879CURRENT_NIGHTLY_VERSION = "2.9.0"
7980CURRENT_CANDIDATE_VERSION = "2.8.0"
80- CURRENT_STABLE_VERSION = "2.7.1 "
81+ CURRENT_STABLE_VERSION = "2.8.0 "
8182CURRENT_VERSION = CURRENT_STABLE_VERSION
8283
8384# By default use Nightly for CUDA arches
@@ -228,7 +229,6 @@ def get_libtorch_install_command(
228229 channel : str ,
229230 gpu_arch_type : str ,
230231 libtorch_variant : str ,
231- devtoolset : str ,
232232 desired_cuda : str ,
233233 libtorch_config : str ,
234234) -> str :
@@ -238,16 +238,7 @@ def get_libtorch_install_command(
238238 if libtorch_config == "debug"
239239 else libtorch_variant
240240 )
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"
251242
252243 if os == MACOS_ARM64 :
253244 arch = "arm64"
@@ -256,16 +247,9 @@ def get_libtorch_install_command(
256247 build_name = f"libtorch-macos-{ arch } -{ CURRENT_VERSION } .zip"
257248
258249 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+ )
269253
270254 elif os == WINDOWS and (channel in (RELEASE , TEST )):
271255 build_name = (
@@ -412,7 +396,6 @@ def generate_libtorch_matrix(
412396 channel ,
413397 gpu_arch_type ,
414398 libtorch_variant ,
415- devtoolset ,
416399 desired_cuda ,
417400 libtorch_config ,
418401 ),
0 commit comments