add cuda

Author: alanray-tech

.github/workflows/linux-cuda.yml

@@ -38,22 +38,44 @@ jobs:
           CIBW_BUILD: "cp311-*linux*"
           CIBW_MANYLINUX_X86_64_IMAGE: "quay.io/pypa/manylinux_2_28_x86_64:latest"
           CIBW_BEFORE_BUILD: |
-            # Install CUDA toolkit
-            # For manylinux containers, you may need to install CUDA manually
-            # Option 1: Use system package manager (if available)
-            yum install -y cuda-toolkit-12-4 2>/dev/null || \
-            apt-get update && apt-get install -y nvidia-cuda-toolkit 2>/dev/null || \
-            echo "CUDA toolkit installation skipped - may need manual setup"
+            # Install CUDA toolkit for manylinux containers
+            # Note: CUDA installation in manylinux containers is complex
+            # Consider using a CUDA-enabled base image or pre-installing CUDA
+            
+            echo "Setting up CUDA environment..."
+            
+            # Try to find existing CUDA installation
+            if [ -d "/usr/local/cuda" ]; then
+              export CUDA_HOME=/usr/local/cuda
+            elif [ -d "/opt/cuda" ]; then
+              export CUDA_HOME=/opt/cuda
+            else
+              # Try to install CUDA via package manager
+              if command -v yum &> /dev/null; then
+                # For RHEL/CentOS/AlmaLinux - try to add CUDA repo and install
+                echo "Attempting to install CUDA via yum..."
+                yum install -y wget || true
+                # Note: This may require manual setup of CUDA repository
+                # For production, consider using a CUDA-enabled Docker image
+              fi
+              
+              # Default CUDA_HOME if installation fails
+              export CUDA_HOME=/usr/local/cuda
+              mkdir -p $CUDA_HOME/bin $CUDA_HOME/lib64 || true
+            fi
             
             # Set CUDA environment variables
-            export CUDA_HOME=/usr/local/cuda
-            export PATH=$PATH:/usr/local/cuda/bin
-            export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
+            export PATH=$PATH:$CUDA_HOME/bin
+            export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
             
             # Verify CUDA if available
             if command -v nvcc &> /dev/null; then
-              echo "CUDA compiler version:"
-              nvcc --version || true
+              echo "CUDA compiler found:"
+              nvcc --version
+            else
+              echo "Warning: CUDA compiler (nvcc) not found"
+              echo "CUDA_HOME: $CUDA_HOME"
+              echo "You may need to install CUDA manually or use a CUDA-enabled base image"
             fi
           CIBW_ENVIRONMENT: "CUDA_HOME=/usr/local/cuda PATH=$PATH:/usr/local/cuda/bin LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
           CIBW_CONFIG_SETTINGS: "wheel.build-tag=${{ github.run_number }}"

linux-cuda/pyproject.toml

@@ -1,24 +1,33 @@
-# CUDA-specific build configuration for Linux
-# This file is used when building with cibuildwheel for CUDA-enabled wheels
-# The actual CUDA installation and environment setup is handled in the GitHub Actions workflow
+[build-system]
+requires = ["scikit-build-core>=0.9.0", "pybind11>=2.11.0", "setuptools-scm"]
+build-backend = "scikit_build_core.build"
+
+[project]
+name = "hello-cuda"
+dynamic = ["version"]
+description = "CUDA-enabled pybind11 hello world module built with CMake."
+readme = "README.md"
+requires-python = ">=3.11,<3.12"
+license = { text = "MIT" }
+authors = [{ name = "Example Author" }]
+classifiers = [
+  "Programming Language :: Python :: 3",
+  "Programming Language :: C++",
+  "Programming Language :: CUDA",
+  "License :: OSI Approved :: MIT License",
+]
 
 [tool.scikit-build]
+metadata.version.provider = "scikit_build_core.metadata.setuptools_scm"
 minimum-version = "0.9"
 cmake.version = ">=3.26"
 build-dir = "build/{wheel_tag}"
 wheel.packages = []
 
 # CMake configuration for CUDA
 # These args will be passed to CMake during the build
-# Note: CUDA language support should be enabled in your CMakeLists.txt
 cmake.args = [
     "-DCMAKE_CUDA_ARCHITECTURES=75;80;86;89;90",  # Common GPU architectures (adjust as needed)
     "-DENABLE_CUDA=ON",
     "-DCMAKE_CUDA_COMPILER=nvcc",
 ]
-
-# Example CMakeLists.txt additions needed for CUDA:
-# cmake_minimum_required(VERSION 3.26)
-# enable_language(CUDA)
-# find_package(CUDA REQUIRED)
-# set(CMAKE_CUDA_STANDARD 17)