frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade nightly wheels to rocm6.0 (#116983)

Browse Source 
Follow-up to https://github.com/pytorch/builder/pull/1647

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116983
Approved by: https://github.com/jeffdaily, https://github.com/atalman
This commit is contained in:
Jithun Nair
2024-01-11 20:36:00 +00:00
committed by PyTorch MergeBot
parent e55a778cbb
commit 24c39bb5e5
4 changed files with 721 additions and 721 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/scripts/generate_binary_build_matrix.py vendored
View File

@ -22,7 +22,7 @@ CUDA_ARCHES_FULL_VERSION = {"11.8": "11.8.0", "12.1": "12.1.1"}
CUDA_ARCHES_CUDNN_VERSION = {"11.8": "8", "12.1": "8"}
ROCM_ARCHES = ["5.6", "5.7"]
ROCM_ARCHES = ["5.7", "6.0"]
CPU_CXX11_ABI_ARCH = ["cpu-cxx11-abi"]