Skip to content

Commit fe057e4

Browse files
committed
Remove rpath $ORIGIN code from setup.py
1 parent 11aa151 commit fe057e4

File tree

1 file changed

+1
-25
lines changed

1 file changed

+1
-25
lines changed

cuda_bindings/setup.py

Lines changed: 1 addition & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -379,31 +379,7 @@ def initialize_options(self):
379379
def build_extension(self, ext):
380380
if building_wheel and sys.platform == "linux":
381381
# Strip binaries to remove debug symbols
382-
extra_linker_flags = ["-Wl,--strip-all"]
383-
384-
# Allow extensions to discover libraries at runtime
385-
# relative their wheels installation.
386-
if ext.name == "cuda.bindings._bindings.cynvrtc":
387-
ldflag = "-Wl,--disable-new-dtags,-rpath,$ORIGIN/../../../nvidia/cuda_nvrtc/lib"
388-
elif ext.name == "cuda.bindings._internal.nvjitlink":
389-
ldflag = "-Wl,--disable-new-dtags,-rpath,$ORIGIN/../../../nvidia/nvjitlink/lib"
390-
elif ext.name == "cuda.bindings._internal.nvvm":
391-
# from <loc>/site-packages/cuda/bindings/_internal/
392-
# to <loc>/site-packages/nvidia/cuda_nvcc/nvvm/lib64/
393-
rel1 = "$ORIGIN/../../../nvidia/cuda_nvcc/nvvm/lib64"
394-
# from <loc>/lib/python3.*/site-packages/cuda/bindings/_internal/
395-
# to <loc>/nvvm/lib64/
396-
rel2 = "$ORIGIN/../../../../../../nvvm/lib64"
397-
ldflag = f"-Wl,--disable-new-dtags,-rpath,{rel1},-rpath,{rel2}"
398-
else:
399-
ldflag = None
400-
401-
if ldflag:
402-
extra_linker_flags.append(ldflag)
403-
else:
404-
extra_linker_flags = []
405-
406-
ext.extra_link_args += extra_linker_flags
382+
ext.extra_link_args.append("-Wl,--strip-all")
407383
super().build_extension(ext)
408384

409385

0 commit comments

Comments
 (0)