From f7299cede7a2b44db588c4544c85cd067378af61 Mon Sep 17 00:00:00 2001
From: Ewan Miller <ewan.miller@kcl.ac.uk>
Date: Fri, 20 Sep 2024 17:12:36 -0500
Subject: [PATCH] Have separate forks for cpu and gpu distribution

---
 .github/workflows/pypi-distribution.yaml | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/.github/workflows/pypi-distribution.yaml b/.github/workflows/pypi-distribution.yaml
index 106861e..2ea6533 100644
--- 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: