We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 30f2cf9 commit 71550acCopy full SHA for 71550ac
1 file changed
pyproject.toml
@@ -28,11 +28,14 @@ before-all = [
28
"chmod +x scripts/build-env/manylinux.sh",
29
"scripts/build-env/manylinux.sh"
30
]
31
+repair-wheel-command = [
32
+ "auditwheel repair -w {dest_dir} --exclude libcudart.so.* --exclude libcublas.so.* --exclude libcublasLt.so.* {wheel}"
33
+]
34
35
[tool.cibuildwheel.linux.environment]
36
cuda = "12.4"
37
temp = "/tmp/cuda-temp"
38
TENSOR_ARRAY_INSTALL_PATH = "/tmp/tensor-array-temp"
39
CUDA_PATH = "/usr/local/cuda-${cuda}"
40
PATH = "${PATH}:${CUDA_PATH}/bin"
-LD_LIBRARY_PATH = "${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"
41
+LD_LIBRARY_PATH = "${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64:${TENSOR_ARRAY_INSTALL_PATH}/lib/tensor-array:${TENSOR_ARRAY_INSTALL_PATH}/lib64/tensor-array"
0 commit comments