Build cuda wheels

Author: govindchari

.gitignore

@@ -3,4 +3,5 @@ build/
 .vscode/
 __pycache__/
 qoco_custom*/
-src/bindings.cpp
+src/bindings.cpp
+dist/

Dockerfile.cuda-manylinux

@@ -0,0 +1,19 @@
+# Base manylinux image with Python interpreters
+FROM quay.io/pypa/manylinux_2_28_x86_64:2025.11.09-2
+
+# Install prerequisites
+RUN yum install -y wget tar bzip2 xz gzip make gcc gcc-c++ git
+
+# Install CUDA Toolkit 13.1 and cuDSS 0.7.1
+RUN wget https://developer.download.nvidia.com/compute/cuda/13.0.0/local_installers/cuda_13.0.0_580.65.06_linux.run && \
+    sh cuda_13.0.0_580.65.06_linux.run --silent --toolkit && \
+    rm cuda_13.0.0_580.65.06_linux.run && \
+    curl -O https://developer.download.nvidia.com/compute/cudss/0.7.1/local_installers/cudss-local-repo-rhel10-0.7.1-0.7.1-1.x86_64.rpm && \
+    rpm -i cudss-local-repo-rhel10-0.7.1-0.7.1-1.x86_64.rpm && \
+    dnf clean all && \
+    dnf -y install cudss
+
+# Set CUDA environment variables
+ENV PATH=/usr/local/cuda/bin:$PATH
+ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
+ENV CUDAToolkit_ROOT=/usr/local/cuda/bin

backend/cuda/cibuildwheel.toml

@@ -1,14 +1,4 @@
-[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-13-0"
-]
-environment = { CMAKE_CUDA_COMPILER = "/usr/local/cuda-13.0/bin/nvcc" }
+skip = ["*-musllinux*"]
+manylinux-x86_64-image = "mycuda-manylinux:latest"
+environment = { CUDAToolkit_ROOT = "/usr/local/cuda/bin" }

backend/cuda/pyproject.toml

@@ -12,15 +12,11 @@ dependencies = ["numpy>=1.7", "scipy>=0.13.2", "setuptools", "qoco>=0.2.0"]
 
 [tool.scikit-build]
 install.components = ["python"]
-cmake.source-dir = "../.."
 
 [tool.scikit-build.cmake.define]
 QOCO_ALGEBRA_BACKEND = "cuda"
 QOCO_EXT_MODULE_NAME = "qoco_cuda"
 
 [project.urls]
 Homepage = "https://github.com/qoco-org/qoco"
-Issues = "https://github.com/qoco-org/qoco/issues"
-
-[tool.setuptools_scm]
-root = "../.."
+Issues = "https://github.com/qoco-org/qoco/issues"

build_cuda_wheels.sh

@@ -0,0 +1,3 @@
+mv backend/cuda/pyproject.toml .
+docker build -f Dockerfile.cuda-manylinux -t mycuda-manylinux:latest .
+cibuildwheel --platform linux --output-dir dist --config-file backend/cuda/cibuildwheel.toml