nWEIdia
diff --git a/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 9 additions & 0 deletions b/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 9 additions & 0 deletions
@@ -220,6 +220,9 @@ 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.4 builds on Windows
+        if os == "windows" and "12.4" in arches:
+            arches.remove("12.4")
     for python_version in python_versions:
         # We don't currently build conda packages for rocm
         for arch_version in arches:
@@ -256,6 +259,9 @@ def generate_libtorch_matrix(
             arches += ROCM_ARCHES
         elif os == "windows":
             arches += CUDA_ARCHES
+            # skip CUDA 12.4 builds on windows
+            if "12.4" in arches:
+                arches.remove("12.4")
 
     if libtorch_variants is None:
         libtorch_variants = [
@@ -320,6 +326,9 @@ def generate_wheels_matrix(
             arches += CPU_CXX11_ABI_ARCH + CUDA_ARCHES + ROCM_ARCHES
         elif os == "windows":
             arches += CUDA_ARCHES
+            # skip CUDA 12.4 builds on Windows
+            if "12.4" in arches:
+                arches.remove("12.4")
         elif os == "linux-aarch64":
             # Only want the one arch as the CPU type is different and
             # uses different build/test scripts