Build sphericart-torch wheels on macOS

.github/workflows/build-torch-wheels.yml

@@ -16,16 +16,22 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [ubuntu-22.04]
-        arch: ["x86_64"]
         pytorch-version: ["2.4", "2.5"]
+        arch: ["arm64", "x86_64"]
+        os: ["ubuntu-22.04", "macos-14"]
+        exclude:
+          # remove mismatched arch/os pairs
+          - {os: macos-14, arch: x86_64}
+          - {os: ubuntu-22.04, arch: arm64}
+
     name: "Torch ${{matrix.pytorch-version}} ${{matrix.os}} ${{matrix.arch}}"
 
     steps:
     - name: Checkout code
       uses: actions/checkout@v4
 
     - name: Build custom manylinux Docker Image with CUDA
+      if: matrix.os == 'ubuntu-22.04'
       run: |
         docker build --no-cache \
             -t sphericart_manylinux2014_"${{ matrix.arch }}" \
@@ -55,6 +61,10 @@ jobs:
               --exclude libtorch_cuda.so \
               --exclude libc10_cuda.so  \
               -w {dest_dir} {wheel}
+        CIBW_REPAIR_WHEEL_COMMAND_MACOS: |
+          delocate-wheel --ignore-missing-dependencies \
+                         --require-archs {delocate_archs} \
+                         -w {dest_dir} -v {wheel}
         CIBW_ENVIRONMENT: >
           SPHERICART_ARCH_NATIVE=OFF
           SPHERICART_NO_LOCAL_DEPS=1
@@ -76,6 +86,9 @@ jobs:
           - name: x86_64 Linux
             os: ubuntu-22.04
             arch: x86_64
+          - name: arm64 macOS
+            os: macos-14
+            arch: arm64
     steps:
       - uses: actions/checkout@v4
 
@@ -96,6 +109,8 @@ jobs:
 
       - name: merge wheels
         run: |
+          set -ux
+
           # collect all torch versions used for the build
           REQUIRES_TORCH=$(find dist -name "*.whl" -exec unzip -p {} "sphericart_torch-*.dist-info/METADATA" \; | grep "Requires-Dist: torch")
           MERGED_TORCH_REQUIRE=$(python scripts/create-torch-versions-range.py "$REQUIRES_TORCH")