Have separate forks for cpu and gpu distribution

ewanwm · Sep 20, 2024 · f7299ce · f7299ce
1 parent 36bf503
commit f7299ce
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/.github/workflows/pypi-distribution.yaml b/.github/workflows/pypi-distribution.yaml
@@ -6,11 +6,23 @@ on:
 
 jobs:
   build:
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - build_type: cpu
+          - build_type: cuda
+
     name: Build distribution 📦
     runs-on: ubuntu-latest
 
     steps:
     - uses: actions/checkout@v4
+
+    - name: cuda-toolkit              
+      if: ${{ startsWith(matrix.build_type, 'cuda') }}
+      uses: Jimver/cuda-toolkit@v0.2.17
+
     - name: Set up Python
       uses: actions/setup-python@v5
       with: