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

Re-enable CUDA 12.1 builds for Windows (#100268)

Browse Source 
Related: https://github.com/pytorch/pytorch/pull/98492
This PR enables Windows builds after the needed AMIs are ready.

CC @atalman

Pull Request resolved: https://github.com/pytorch/pytorch/pull/100268
Approved by: https://github.com/atalman, https://github.com/malfet
This commit is contained in:
pbialecki
2023-04-28 21:11:27 +00:00
committed by PyTorch MergeBot
parent 5b98910139
commit 0a5c930499
5 changed files with 3888 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,9 +99,6 @@ def generate_conda_matrix(os: str) -> List[Dict[str, str]]:
python_versions = FULL_PYTHON_VERSIONS
if os == "linux" or os == "windows":
arches += CUDA_ARCHES
# skip CUDA 12.1 builds on Windows
if os == "windows" and "12.1" in arches:
arches.remove("12.1")
for python_version in python_versions:
# We don't currently build conda packages for rocm
for arch_version in arches:
@ -138,9 +135,6 @@ def generate_libtorch_matrix(
arches += ROCM_ARCHES
elif os == "windows":
arches += CUDA_ARCHES
# skip CUDA 12.1 builds on Windows
if "12.1" in arches:
arches.remove("12.1")
if libtorch_variants is None:
libtorch_variants = [
@ -206,9 +200,6 @@ def generate_wheels_matrix(
arches += CPU_CXX11_ABI_ARCH + CUDA_ARCHES + ROCM_ARCHES
elif os == "windows":
arches += CUDA_ARCHES
# skip CUDA 12.1 builds on Windows
if "12.1" in arches:
arches.remove("12.1")
ret: List[Dict[str, str]] = []
for python_version in python_versions: