Merge pull request #78 from pfxuan/mps-build

Enable MPS build
pierotofy · Apr 16, 2024 · 9275c62 · 9275c62
2 parents 8bbba79 + f6d4bff
commit 9275c62
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/.github/workflows/macos.yml b/.github/workflows/macos.yml
@@ -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 .
@@ -111,9 +119,11 @@ jobs:
         uses: actions/upload-artifact@v4
         with:
           path: |
+            build/libgsplat.a
             build/libgsplat_cpu.a
             build/opensplat
             build/simple_trainer
+            build/default.metallib
 
       - name: Clean Compiler Cache
         run: |