pytorch
diff --git a/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 0 additions & 9 deletions b/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 0 additions & 9 deletions
@@ -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: