test

Author: BigNoobWasTaken

pyproject.toml

@@ -28,11 +28,13 @@ before-all = [
   "chmod +x scripts/build-env/manylinux.sh",
   "scripts/build-env/manylinux.sh"
 ]
-repair-wheel-command = "auditwheel repair --exclude libcu*.so.* -w {dest_dir} {wheel}"
+repair-wheel-command = [
+  "auditwheel repair -L tensor-array-repo/Tensor-Array/lib --exclude libcu*.so.* -w {dest_dir} {wheel}"
+]
 
 [tool.cibuildwheel.linux.environment]
 cuda = "12.9"
 temp = "/tmp/cuda-temp"
 CUDA_PATH = "/usr/local/cuda-${cuda}"
 PATH = "${PATH}:${CUDA_PATH}/bin"
-LD_LIBRARY_PATH = "${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64:/project/tensor-array-repo/Tensor-Array/lib"
+LD_LIBRARY_PATH = "${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"

setup.py

@@ -20,7 +20,7 @@ def main():
             sources = glob.glob(os.path.join("cpp", "*.cc")),
             include_dirs=["tensor-array-repo/Tensor-Array/include"],
             library_dirs=["tensor-array-repo/Tensor-Array/lib"],
-            libraries=["tensorarray_core"],
+            libraries=["tensorarray_core", "tensorarray_layers"],
             define_macros=[("VERSION_INFO", __version__)],
             ),
     ]
@@ -52,9 +52,6 @@ def main():
         package_dir = {
             "": "src",
         },
-        package_data = {
-            "": ["*.so"]
-        },
         ext_modules = ext_modules,
         cmdclass = {
             "build_ext": build_ext,