Commit 1e239b3a authored by Samuel Ainsworth's avatar Samuel Ainsworth
Browse files

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

parent 6e0570c1

pkgs/development/python-modules/jaxlib/default.nix

+1 −1
Original line number Diff line number Diff line
@@ -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

  '';