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 = [
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 ]
3336cuda = " 12.4"
3437temp = " /tmp/cuda-temp"
3538TENSOR_ARRAY_INSTALL_PATH = " /tmp/tensor-array-temp"
3639CUDA_PATH = " /usr/local/cuda-${cuda}"
3740PATH = " ${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