cudaPackages.cudnn: use libcublas split outputs

This commit is contained in:
Connor Baker 2023-09-08 03:58:23 +00:00
parent 3cba01e8ab
commit 10bb3be659

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