Skip to content

Commit

Permalink
Re-enable CUDA 12.1 builds for Windows (pytorch#100268)
Browse files Browse the repository at this point in the history
Related: pytorch#98492
This PR enables Windows builds after the needed AMIs are ready.

CC @atalman

Pull Request resolved: pytorch#100268
Approved by: https://github.com/atalman, https://github.com/malfet
  • Loading branch information
ptrblck authored and pytorchmergebot committed Apr 28, 2023
1 parent 5b98910 commit 0a5c930
Show file tree
Hide file tree
Showing 5 changed files with 4,178 additions and 299 deletions.
9 changes: 0 additions & 9 deletions .github/scripts/generate_binary_build_matrix.py
Expand Up @@ -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:
Expand Down Expand Up @@ -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 = [
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 0a5c930

Please sign in to comment.