Skip to content

Commit 71550ac

Browse files
test 2
1 parent 30f2cf9 commit 71550ac

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pyproject.toml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,14 @@ before-all = [
2828
"chmod +x scripts/build-env/manylinux.sh",
2929
"scripts/build-env/manylinux.sh"
3030
]
31+
repair-wheel-command = [
32+
"auditwheel repair -w {dest_dir} --exclude libcudart.so.* --exclude libcublas.so.* --exclude libcublasLt.so.* {wheel}"
33+
]
3134

3235
[tool.cibuildwheel.linux.environment]
3336
cuda = "12.4"
3437
temp = "/tmp/cuda-temp"
3538
TENSOR_ARRAY_INSTALL_PATH = "/tmp/tensor-array-temp"
3639
CUDA_PATH = "/usr/local/cuda-${cuda}"
3740
PATH = "${PATH}:${CUDA_PATH}/bin"
38-
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

Comments
 (0)