Commit 856e04dc authored by David Guibert's avatar David Guibert
Browse files

cudnn: throw error if the default version is not supported

parent ad6aeedf
Loading
Loading
Loading
Loading
+3 −1
Original line number Diff line number Diff line
@@ -28,7 +28,9 @@ final: prev: let
    # Add all supported builds as attributes
    allBuilds = mapAttrs' (version: file: nameValuePair (computeName version) (buildCuDnnPackage (removeAttrs file ["fileVersion"]))) supportedVersions;
    # Set the default attributes, e.g. cudnn = cudnn_8_3_1;
    defaultBuild = { "cudnn" = allBuilds.${computeName cuDnnDefaultVersion}; };
    defaultBuild = { "cudnn" = if allBuilds ? ${computeName cuDnnDefaultVersion}
      then allBuilds.${computeName cuDnnDefaultVersion}
      else throw "cudnn-${cuDnnDefaultVersion} does not support your cuda version ${cudaVersion}"; };
  in allBuilds // defaultBuild;

  cuDnnVersions = let