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

[ROCm] update nightly builds to rocm5.2 (#82353)

Browse Source 
### Description
Updates ROCm arches for nightly binaries to 5.1.1 and 5.2, dropping 5.0.

Corresponding builder PR:
https://github.com/pytorch/builder/pull/1033

### Testing
PR CI with extra label to enable binary builds.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82353
Approved by: https://github.com/janeyx99, https://github.com/malfet
This commit is contained in:
Jeff Daily
2022-08-11 00:33:57 +00:00
committed by PyTorch MergeBot
parent 9590cf6d79
commit 2dae93b212
4 changed files with 1133 additions and 1133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,7 +16,7 @@ from typing import Dict, List, Tuple, Optional
CUDA_ARCHES = ["10.2", "11.3", "11.6", "11.7"]
ROCM_ARCHES = ["5.0", "5.1.1"]
ROCM_ARCHES = ["5.1.1", "5.2"]
def arch_type(arch_version: str) -> str: