@@ -379,31 +379,7 @@ def initialize_options(self):
379
379
def build_extension (self , ext ):
380
380
if building_wheel and sys .platform == "linux" :
381
381
# 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" )
407
383
super ().build_extension (ext )
408
384
409
385
0 commit comments