python3Packages.jaxlib: --set-rpath -> --add-rpath

This commit is contained in:
Samuel Ainsworth 2024-02-21 19:34:07 +00:00
parent 6e0570c14d
commit 1e239b3ae2

View file

@ -416,7 +416,7 @@ buildPythonPackage {
ln -s ${cudaPackagesGoogle.cuda_nvcc.bin}/bin/ptxas $out/bin/ptxas
find $out -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do
patchelf --set-rpath "${cuda_libs_joined}/lib:${cudnn}/lib:${nccl}/lib:$(patchelf --print-rpath "$lib")" "$lib"
patchelf --add-rpath "${lib.makeLibraryPath [cuda_libs_joined cudnn nccl]}" "$lib"
done
'';