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

.github: Ensure we're using correct build matrix (#72010)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72010

We were adding additional CUDA arches to our libtorch builds when we
shouldn't have been

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Test Plan: Imported from OSS

Reviewed By: atalman

Differential Revision: D33851196

Pulled By: seemethere

fbshipit-source-id: 52055d0cf5b528f45ef0aa33da297cd4175e8dcf
(cherry picked from commit f33b27ecab856a69c52625abf292f51dd2602229)
This commit is contained in:
Eli Uriegas
2022-01-28 11:30:48 -08:00
committed by PyTorch MergeBot
parent b2b63209e1
commit 09e54ffec3
3 changed files with 1 additions and 2297 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -121,7 +121,7 @@ def generate_libtorch_matrix(os: str, abi_version: str) -> List[Dict[str, str]]:
elif os == "windows":
# We don't build CUDA 10.2 for window see https://github.com/pytorch/pytorch/issues/65648
arches += list_without(CUDA_ARCHES, ["10.2"])
for arch_version in ["cpu"] + CUDA_ARCHES:
for arch_version in arches:
for libtorch_variant in libtorch_variants:
# We don't currently build libtorch for rocm
# one of the values in the following list must be exactly