File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -28,11 +28,14 @@ before-all = [
28
28
" chmod +x scripts/build-env/manylinux.sh" ,
29
29
" scripts/build-env/manylinux.sh"
30
30
]
31
+ repair-wheel-command = [
32
+ " auditwheel repair -w {dest_dir} --exclude libcudart.so.* --exclude libcublas.so.* --exclude libcublasLt.so.* {wheel}"
33
+ ]
31
34
32
35
[tool .cibuildwheel .linux .environment ]
33
36
cuda = " 12.4"
34
37
temp = " /tmp/cuda-temp"
35
38
TENSOR_ARRAY_INSTALL_PATH = " /tmp/tensor-array-temp"
36
39
CUDA_PATH = " /usr/local/cuda-${cuda}"
37
40
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 "
You can’t perform that action at this time.
0 commit comments