Skip to content

Commit

Permalink
Stop cuda-10.2 binary builds (pytorch#85873)
Browse files Browse the repository at this point in the history
Deprecate cuda 10.2 nightly

Pull Request resolved: pytorch#85873
Approved by: https://github.com/malfet
  • Loading branch information
atalman authored and pytorchmergebot committed Sep 29, 2022
1 parent 3cdf621 commit a807f19
Show file tree
Hide file tree
Showing 7 changed files with 16 additions and 1,065 deletions.
13 changes: 4 additions & 9 deletions .github/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from typing import Dict, List, Tuple, Optional


CUDA_ARCHES = ["10.2", "11.6", "11.7"]
CUDA_ARCHES = ["11.6", "11.7"]


ROCM_ARCHES = ["5.1.1", "5.2"]
Expand Down Expand Up @@ -90,11 +90,8 @@ def generate_conda_matrix(os: str) -> List[Dict[str, str]]:
ret: List[Dict[str, str]] = []
arches = ["cpu"]
python_versions = FULL_PYTHON_VERSIONS
if os == "linux":
if os == "linux" or os == "windows":
arches += CUDA_ARCHES
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"])
elif os == "macos-arm64":
python_versions = list_without(python_versions, ["3.7"])
for python_version in python_versions:
Expand Down Expand Up @@ -129,8 +126,7 @@ def generate_libtorch_matrix(os: str, abi_version: str,
arches += CUDA_ARCHES
arches += ROCM_ARCHES
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"])
arches += CUDA_ARCHES

if libtorch_variants is None:
libtorch_variants = [
Expand Down Expand Up @@ -198,8 +194,7 @@ def generate_wheels_matrix(os: str,
if os == "linux":
arches += CUDA_ARCHES + ROCM_ARCHES
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"])
arches += CUDA_ARCHES

ret: List[Dict[str, str]] = []
for python_version in python_versions:
Expand Down
2 changes: 1 addition & 1 deletion .github/scripts/generate_ci_workflows.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ class OperatingSystem:
package_type="manywheel",
build_configs=generate_binary_build_matrix.generate_wheels_matrix(
OperatingSystem.LINUX,
arches=["10.2"],
arches=["11.6"],
python_versions=["3.7"]),
branches="master",
),
Expand Down
240 changes: 0 additions & 240 deletions .github/workflows/generated-linux-binary-conda-nightly.yml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading

0 comments on commit a807f19

Please sign in to comment.