different approach for arch list

Author: stefdoerr

.github/workflows/publish.yml

@@ -62,6 +62,15 @@ jobs:
           CPU_TRAIN: ${{ runner.os == 'macOS' && 'true' || 'false' }}
           WITH_CUDA: ${{ matrix.cuda != 'cpu' && '1' || '0' }}
 
+      - name: Set CUDA arch list for Windows
+        if: matrix.os == 'windows-2019'
+        run: |
+          if [[ "${{ matrix.cuda }}" == "cu118" ]]; then
+            echo ("TORCH_CUDA_ARCH_LIST='3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9+PTX'") >> $env:GITHUB_ENV
+          elif [[ "${{ matrix.cuda }}" == "cu124" ]]; then
+            echo ("TORCH_CUDA_ARCH_LIST='5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX'") >> $env:GITHUB_ENV
+          fi
+
       - name: Build wheels
         if: matrix.os == 'windows-2019'
         shell: cmd # Use cmd on Windows to avoid bash environment taking priority over MSVC variables
@@ -71,7 +80,8 @@ jobs:
           WITH_CUDA: ${{ matrix.cuda != 'cpu' && '1' || '0' }}
           DISTUTILS_USE_SDK: "1" # Windows requires this to use vc for building
           SKIP_TORCH_COMPILE: "true"
-          CIBW_BUILD_VERBOSITY_WINDOWS: "2"
+          CIBW_BUILD_VERBOSITY_WINDOWS: "1"
+          TORCH_CUDA_ARCH_LIST: ${{ env.TORCH_CUDA_ARCH_LIST }}
 
       - uses: actions/upload-artifact@v4
         with:

cibuildwheel_support/prepare_build_environment_windows_cu118.sh

@@ -10,7 +10,6 @@ rm cuda.exe
 
 export WITH_CUDA=1
 export CUDA_HOME="$CUDA_ROOT"
-export TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9+PTX"
 
 # Create pip directory if it doesn't exist
 mkdir -p "C:\ProgramData\pip"

cibuildwheel_support/prepare_build_environment_windows_cu124.sh

@@ -10,4 +10,3 @@ rm cuda.exe
 
 export WITH_CUDA=1
 export CUDA_HOME="$CUDA_ROOT"
-export TORCH_CUDA_ARCH_LIST="5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX"