test

Author: BigNoobWasTaken

pyproject.toml

@@ -32,6 +32,7 @@ before-all = [
 [tool.cibuildwheel.linux.environment]
 cuda = "12.4"
 temp = "/tmp/cuda-temp"
+TENSOR_ARRAY_INSTALL_PATH = "/tmp/tensor-array-temp"
 CUDA_PATH = "/usr/local/cuda-${cuda}"
 PATH = "${PATH}:${CUDA_PATH}/bin"
 LD_LIBRARY_PATH = "${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"

scripts/build-env/manylinux.sh

@@ -62,7 +62,7 @@ fi
 cd build
 # Configure the build with CMake
 echo "Configuring the build with CMake..."
-cmake .. -DCMAKE_CUDA_HOST_COMPILER="/usr/bin/gcc"
+cmake .. -DCMAKE_CUDA_HOST_COMPILER="/usr/bin/gcc" -DCMAKE_INSTALL_PREFIX=${TENSOR_ARRAY_INSTALL_PATH}
 cmake --build .
 cmake --install .
 

setup.py

@@ -7,20 +7,22 @@
 from setuptools import setup, find_packages
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 
-__version__ = "0.0.7"
+__version__ = "0.0.8"
 
 def main():
     cwd = os.path.dirname(os.path.abspath(__file__))
     with open(os.path.join(cwd, "README.md"), encoding="utf-8") as f:
         long_description = f.read()
 
+    tensor_array_lib_path = os.environ['TENSOR_ARRAY_INSTALL_PATH']
+
     ext_modules = [
         Pybind11Extension(
             "tensor_array._ext",
             sources = glob.glob(os.path.join("cpp", "*.cc")),
-            include_dirs=["/usr/local/include"],
-            library_dirs=["/usr/local/lib", "/usr/local/lib64"],
-            # libraries=["tensorarray_core", "tensorarray_layers"],
+            include_dirs=[tensor_array_lib_path + "/include"],
+            library_dirs=[tensor_array_lib_path + "/lib/tensor-array", tensor_array_lib_path + "/lib64/tensor-array"],
+            libraries=["tensorarray_core", "tensorarray_layers"],
             define_macros=[("VERSION_INFO", __version__)],
             ),
     ]