cudnn: add 8.4.0

NixOS · Aug 2, 2022 · a43d296 · a43d296
1 parent 79bc66c
commit a43d296
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pkgs/development/libraries/science/math/cudnn/extension.nix b/pkgs/development/libraries/science/math/cudnn/extension.nix
@@ -97,6 +97,22 @@ final: prev: let
         supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" "11.7" ];
       }
     ];
+    "8.4.0" = [
+      rec {
+        fileVersion = "10.2";
+        fullVersion = "8.4.0.27";
+        hash = "sha256-FMXjykJYJxmW0f2VnELRfFgs5Nmv9FH4RSRGnnhP0VQ=";
+        url = "${urlPrefix}/v${majorMinorPatch fullVersion}/local_installers/${fileVersion}/cudnn-linux-x86_64-${fullVersion}_cuda${fileVersion}-archive.tar.xz";
+        supportedCudaVersions = [ "10.2" ];
+      }
+      rec {
+        fileVersion = "11.6";
+        fullVersion = "8.4.0.27";
+        hash = "sha256-0Zva/ZgAx50p5vb/+p+eLBDREy1sL/ELFZPgV+dN0FA=";
+        url = "${urlPrefix}/v${majorMinorPatch fullVersion}/local_installers/${fileVersion}/cudnn-linux-x86_64-${fullVersion}_cuda${fileVersion}-archive.tar.xz";
+        supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" "11.7" ];
+      }
+    ];
   };
 
   # Default attributes
@@ -111,7 +127,7 @@ final: prev: let
     "11.4" = "8.3.2";
     "11.5" = "8.3.2";
     "11.6" = "8.3.2";
-    "11.7" = "8.3.2";
+    "11.7" = "8.4.0";
   }.${cudaVersion};
 
 in cuDnnPackages