Try to exclude Python 3.12 from torch 2.0 and 2.1

lab-cosmo · Mar 26, 2024 · 276ecf6 · 276ecf6
1 parent f81b3db
commit 276ecf6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/build-cuda-wheels.yml b/.github/workflows/build-cuda-wheels.yml
@@ -71,7 +71,12 @@ jobs:
         env:
           CIBW_BEFORE_ALL: bash /host/home/runner/work/sphericart/sphericart/scripts/cibw-cuda-setup.sh ${{ matrix.cuda }}
           CIBW_BUILD_VERBOSITY: 3
-          CIBW_BUILD: cp39-* cp310-* cp311-* cp312-*
+          CIBW_BUILD: >
+            bash -c 'if [[ "${{ matrix.torch }}" == "2.2.0" ]]; then
+              echo "cp39-* cp310-* cp311-* cp312-*"
+            else
+              echo "cp39-* cp310-* cp311-*"
+            fi'
           CIBW_SKIP: "*-musllinux* *-win32 *-manylinux_i686"
           CIBW_MANYLINUX_X86_64_IMAGE: manylinux2014
           # set environment variables for sphericart-torch build