Skip to content

Commit cd31e76

Browse files
authored
Add block-list for auditwheel builds with L-GPU (#534)
* Add block-list for auditwheel builds with L-GPU * Update mode permission for auditwheel
1 parent 37d5c18 commit cd31e76

File tree

2 files changed

+26
-1
lines changed

2 files changed

+26
-1
lines changed

.github/workflows/wheel_linux_x86_64_cu11.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ jobs:
8787
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:$CUQUANTUM_SDK \
8888
PL_BACKEND="${{ matrix.pl_backend }}"
8989
90-
CIBW_REPAIR_WHEEL_COMMAND_LINUX: "auditwheel repair -w {dest_dir} {wheel}"
90+
CIBW_REPAIR_WHEEL_COMMAND_LINUX: "./bin/auditwheel repair -w {dest_dir} {wheel}"
9191

9292
CIBW_MANYLINUX_X86_64_IMAGE: manylinux2014
9393

bin/auditwheel

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#!/usr/bin/env python3
2+
3+
# Patch to not ship CUDA system libraries
4+
# Follows https://github.com/DIPlib/diplib/tree/master/tools/travis
5+
import sys
6+
7+
from auditwheel.main import main
8+
from auditwheel.policy import _POLICIES as POLICIES
9+
10+
# Do not include licensed dynamic libraries
11+
libs = [
12+
"libcudart.so.11.0",
13+
"libcublasLt.so.11",
14+
"libcublas.so.11",
15+
"libcusparse.so.11",
16+
"libcustatevec.so.1",
17+
]
18+
19+
print(f"Excluding {libs}")
20+
21+
for p in POLICIES:
22+
p["lib_whitelist"].extend(libs)
23+
24+
if __name__ == "__main__":
25+
sys.exit(main())

0 commit comments

Comments
 (0)