public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #96563 from obsidiansystems/skip-cudnn_cnn_infer-instead-of-removing

Browse Source 
cudnn: skip libcudnn_cnn_infer.so, instead of removing
This commit is contained in:
Matthew Bauer 2020-09-02 16:29:28 -05:00 committed by GitHub
commit 8e6f466967
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkgs/development/libraries/science/math/cudnn/generic.nix
View File

@ -32,19 +32,20 @@ stdenv.mkDerivation {
mkdir -p $out mkdir -p $out
cp -a include $out/include cp -a include $out/include
cp -a lib64 $out/lib64 cp -a lib64 $out/lib64
${lib.optionalString (lib.versionAtLeast version "8") ''
# patchelf fails on libcudnn_cnn_infer due to it being too big.
# I'm hoping it's not needed for most programs.
# (https://github.com/NixOS/patchelf/issues/222)
rm -f $out/lib64/libcudnn_cnn_infer*
''}
''; '';
# Set RUNPATH so that libcuda in /run/opengl-driver(-32)/lib can be found. # Set RUNPATH so that libcuda in /run/opengl-driver(-32)/lib can be found.
# See the explanation in addOpenGLRunpath. # See the explanation in addOpenGLRunpath.
postFixup = '' postFixup = ''
addOpenGLRunpath $out/lib/lib*.so for lib in $out/lib/lib*.so; do
# patchelf fails on libcudnn_cnn_infer due to it being too big.
# Most programs will still get the RPATH since they link to
# other things.
# (https://github.com/NixOS/patchelf/issues/222)
if [ "$(basename $lib)" != libcudnn_cnn_infer.so ]; then
addOpenGLRunpath $lib
fi
done
''; '';
propagatedBuildInputs = [ propagatedBuildInputs = [