File tree Expand file tree Collapse file tree 2 files changed +26
-1
lines changed Expand file tree Collapse file tree 2 files changed +26
-1
lines changed Original file line number Diff line number Diff line change 87
87
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:$CUQUANTUM_SDK \
88
88
PL_BACKEND="${{ matrix.pl_backend }}"
89
89
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}"
91
91
92
92
CIBW_MANYLINUX_X86_64_IMAGE : manylinux2014
93
93
Original file line number Diff line number Diff line change
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 ())
You can’t perform that action at this time.
0 commit comments