Test building cuda wheels

Author: govindchari

.github/workflows/build.yml

@@ -1,36 +1,60 @@
-name: Build CUDA Linux
+# Taken from osqp-python
+
+name: Build Wheels
 
 on:
+  # Triggers the workflow on push or pull request events
   push:
     branches:
-      - master
-    tags:
-      - '*'
+      - "*"
+      - "*/*"
+      - "**"
   pull_request:
-    branches:
-      - master
+    branches: [main]
 
 jobs:
+  build_sdist:
+    name: Build source
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+    steps:
+      - uses: actions/checkout@master
+        with:
+          submodules: "recursive"
+
+      - name: Build source and wheel
+        run: |
+          python -m pip install build
+          python -m build --outdir=wheelhouse
+
+      - name: Upload sdist and wheel to github
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-sdist
+          path: wheelhouse/*
+          if-no-files-found: error
+
   build_wheels:
-    name: Build wheel on ${{ matrix.os }}
+    name: Building wheels on ${{ matrix.os }}
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-latest]
+        os: [ubuntu-latest, macos-13, macos-latest, windows-latest]
 
     steps:
-    - uses: actions/checkout@master
-
-    - name: Build wheels
-      uses: pypa/cibuildwheel@v2.23
-      with:
-        package-dir: backend/cuda
-        config-file: backend/cuda/cibuildwheel.toml
-        output-dir: wheelhouse
-
-    - name: Upload artifacts to github
-      uses: actions/upload-artifact@v4
-      with:
-        name: wheels-cuda-${{ matrix.os }}
-        path: ./wheelhouse
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+          submodules: true
+
+      - uses: pypa/cibuildwheel@v2.21
+        env:
+          CIBW_SKIP: "pp38-*"
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: cibw-wheels-${{ matrix.os }}
+          path: wheelhouse/*.whl

.github/workflows/built_cuda.yml

@@ -0,0 +1,38 @@
+# Taken from osqp-python
+
+name: Build CUDA Linux
+
+on:
+  # Triggers the workflow on push or pull request events
+  push:
+    branches:
+      - "*"
+      - "*/*"
+      - "**"
+  pull_request:
+    branches: [main]
+
+jobs:
+  build_wheels:
+    name: Build wheel on ${{ matrix.os }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest]
+
+    steps:
+    - uses: actions/checkout@master
+
+    - name: Build wheels
+      uses: pypa/cibuildwheel@v2.23
+      with:
+        package-dir: backend/cuda
+        config-file: backend/cuda/cibuildwheel.toml
+        output-dir: wheelhouse
+
+    - name: Upload artifacts to github
+      uses: actions/upload-artifact@v4
+      with:
+        name: wheels-cuda-${{ matrix.os }}
+        path: ./wheelhouse

backends/cuda/cibuildwheel.toml

@@ -0,0 +1,17 @@
+[tool.cibuildwheel]
+build = "cp3*"
+skip = ["cp36-*", "cp37-*", "*-win32", "*-manylinux_i686", "*-musllinux_*"]
+build-verbosity = 1
+before-build = "rm -rf {package}/osqp_sources/build"
+repair-wheel-command = ""
+
+[tool.cibuildwheel.linux]
+before-all = [
+  "yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo",
+  "yum search cuda-toolkit*",
+  "yum install -y cuda-toolkit-12-6"
+]
+environment = { CMAKE_CUDA_COMPILER = "/usr/local/cuda-12.6/bin/nvcc" }
+
+[tool.cibuildwheel.windows]
+environment = { CMAKE_CUDA_COMPILER = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.6/bin/nvcc.exe", CUDA_TOOLKIT_ROOT_DIR = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.6", CMAKE_GENERATOR_TOOLSET = "cuda=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.6" }