Enable mps backend on arm64 runner

pfxuan · Apr 15, 2024 · 5edd0dd · 5edd0dd
1 parent 8bbba79
commit 5edd0dd
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.github/workflows/macos.yml b/.github/workflows/macos.yml
@@ -3,7 +3,7 @@ name: OpenSplat (MacOS X86 and ARM)
 on:
   push:
     branches:
-      - main
+      - mps-build
   pull_request:
     types: [ assigned, opened, synchronize, reopened ]
   release:
@@ -95,13 +95,21 @@ jobs:
           set -x
           mkdir build
           cd build
+          
+          if [[ "${{ matrix.arch }}" == "arm64" ]]; then
+            export GPU_RUNTIME=MPS
+          else
+            export GPU_RUNTIME=CPU
+          fi
+          
           cmake .. \
             -G${CMAKE_GENERATOR} \
             -DCMAKE_BUILD_TYPE=${{ matrix.cmake-build-type }} \
             -DCMAKE_C_COMPILER_LAUNCHER=$(which ccache) \
             -DCMAKE_CXX_COMPILER_LAUNCHER=$(which ccache) \
             -DCMAKE_PREFIX_PATH=${{ github.workspace }}/libtorch \
-            -DCMAKE_INSTALL_PREFIX=${{github.workspace}}/install \
+            -DCMAKE_INSTALL_PREFIX=${{ github.workspace }}/install \
+            -DGPU_RUNTIME=${GPU_RUNTIME} \
             -DOPENSPLAT_BUILD_SIMPLE_TRAINER=ON
           ninja
           ls -l .