only skip Win build for CUDA 12.8 for now (#6252)

cc @atalman
pytorch · Feb 4, 2025 · 3712049 · 3712049
1 parent 261da25
commit 3712049
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -457,7 +457,7 @@ def generate_wheels_matrix(
             if os in (LINUX, WINDOWS):
                 arches += CUDA_ARCHES
             # todo: remove once windows cuda 12.8 binaries are available
-            if channel == NIGHTLY and os != LINUX:
+            if channel == NIGHTLY and os == WINDOWS:
                 arches.remove("12.8")
 
         if with_rocm == ENABLE and os == LINUX: