Merge pull request #253951 from ConnorBaker/fix/cudaPackages-cudnn-use-libcublas-split-output

cudaPackages.cudnn: use libcublas split outputs
This commit is contained in:
Pierre Bourdon 2023-09-08 21:02:29 +02:00 committed by GitHub
commit 51c4092004
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -31,12 +31,6 @@ assert useCudatoolkitRunfile || (libcublas != null); let
# versionTriple :: String
# Version with three components: major.minor.patch
versionTriple = majorMinorPatch version;
# cudatoolkit_root :: Derivation
cudatoolkit_root =
if useCudatoolkitRunfile
then cudatoolkit
else libcublas;
in
backendStdenv.mkDerivation {
pname = "cudatoolkit-${cudaMajorVersion}-cudnn";
@ -65,7 +59,10 @@ in
stdenv.cc.cc.lib
zlib
cudatoolkit_root
] ++ lists.optionals useCudatoolkitRunfile [
cudatoolkit
] ++ lists.optionals (!useCudatoolkitRunfile) [
libcublas.lib
];
# We used to patch Runpath here, but now we use autoPatchelfHook