test

Author: BigNoobWasTaken

pyproject.toml

@@ -7,13 +7,13 @@ requires = [
 build-backend = "setuptools.build_meta"
 
 [tool.cibuildwheel]
-environment = { cuda = "12.9", temp = "/tmp/cuda-temp" }
+environment-pass = ["CUDA_PATH"]
+environment.PATH = "${PATH}:${CUDA_PATH}/bin"
+environment.LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_PATH/lib:$CUDA_PATH/lib64"
+environment.cuda = "12.9"
+environment.temp = "/tmp/cuda-temp"
 xbuild-tools = ["cmake", "ninja"]
 
 [tool.cibuildwheel.linux]
-before-all = [
-  "chmod +x {package}/tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh",
-  "dnf -y install redhat-lsb-core wget",
-  "{package}/tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh"
-]
-xbuild-tools = ["cmake", "ninja"]
+before-all = "scripts/build-env/manylinux.sh"
+xbuild-tools = ["cmake", "ninja", "nvcc"]

scripts/build-env/manylinux.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+cd ../..
+
+# Install dependencies for building Tensor-Array on manylinux
+chmod +x tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh
+dnf -y install redhat-lsb-core wget
+tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh
+
+# Check if nvcc is available
+if ! command -v nvcc &> /dev/null; then
+    echo "nvcc could not be found. Please ensure CUDA is installed correctly."
+    exit 1
+fi