diff options
| author | Rémi Lacroix <remi.lacroix@idris.fr> | 2023-12-23 21:20:08 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-23 14:20:08 -0600 |
| commit | b50f131a0170c73b2389036307fe66b6207e1342 (patch) | |
| tree | 65c6045e2d65a2963fb5bb7708a2e68e98b97efb | |
| parent | 13806213d99649f545caec24de4b5dc50f3fe454 (diff) | |
| download | spack-b50f131a0170c73b2389036307fe66b6207e1342.tar.gz spack-b50f131a0170c73b2389036307fe66b6207e1342.tar.bz2 spack-b50f131a0170c73b2389036307fe66b6207e1342.tar.xz spack-b50f131a0170c73b2389036307fe66b6207e1342.zip | |
CUDNN: Add versions 8.8.x and 8.9.x. (#41790)
NVIDIA changed the download URLs again so account for that.
| -rw-r--r-- | var/spack/repos/builtin/packages/cudnn/package.py | 46 |
1 files changed, 44 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index 54d6708e40..715d19ba90 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -9,6 +9,44 @@ import platform from spack.package import * _versions = { + # cuDNN 8.9.7 + "8.9.7.29-12": { + "Linux-x86_64": "475333625c7e42a7af3ca0b2f7506a106e30c93b1aa0081cd9c13efb6e21e3bb", + "Linux-ppc64le": "8574d291b299f9cc0134304473c9933bd098cc717e8d0876f4aba9f9eebe1b76", + }, + "8.9.7.29-11": { + "Linux-x86_64": "a3e2509028cecda0117ce5a0f42106346e82e86d390f4bb9475afc976c77402e", + "Linux-ppc64le": "f23fd7d59f9d4f743fa926f317dab0d37f6ea21edb2726ceb607bea45b0f9f36", + }, + # cuDNN 8.9.5 + "8.9.5.30-12": { + "Linux-x86_64": "2a2eb89a2ab51071151c6082f1e816c702167a711a9372f9f73a7b5c4b06e01a", + "Linux-ppc64le": "38388ec3c99c6646aaf5c707985cd35e25c67f653d780c4081c2df5557ab665f", + "Linux-aarch64": "0491f7b02f55c22077eb678bf314c1f917524bd507cf5b658239bf98a47233a1", + }, + "8.9.5.30-11": { + "Linux-x86_64": "bbe10e3c08cd7e4aea1012213781e4fe270e1c908263444f567cafefb2cc6525", + "Linux-ppc64le": "d678f8b2903b95de7eeaef38890c5674705864ea049b2b63e90565f2c0ea682f", + }, + # cuDNN 8.9.0 + "8.9.0.131-12": { + "Linux-x86_64": "477631002be61022b60961cba0a501271507a93f81d6b08384bc320cb8706c98", + "Linux-ppc64le": "ff239e4cbbf21fa18104b62a887686e2197f820ad58817d62e509c735a331829", + "Linux-aarch64": "fab70f4fb3b933ff502200a1d954d2c6fc205ff9c9b1d271ea4c41e980a66596", + }, + "8.9.0.131-11": { + "Linux-x86_64": "3cb82c50723f14b41d43523f222cd52cc9d50b3ad67c380f4be51bd1133daa2d", + "Linux-ppc64le": "18778de490550c5b584e96560208e5e37678397037946e10a1c2824174c69725", + }, + # cuDNN 8.8.1 + "8.8.1.3-12": { + "Linux-x86_64": "79d77a769c7e7175abc7b5c2ed5c494148c0618a864138722c887f95c623777c", + "Linux-ppc64le": "b0e89021a846952cad8cfc674edce2883f6e344ebd47a2394f706b1136715bc7", + }, + "8.8.1.3-11": { + "Linux-x86_64": "af7584cae0cc5524b5913ef08c29ba6154113c60eb0a37a0590a91b515a8a8f9", + "Linux-ppc64le": "d086003d09d5388aa42142f07483a773aa74b602478b0933e24fc63f56f1658f", + }, # cuDNN 8.7.0 "8.7.0.84-11.8": { "Linux-x86_64": "976c4cba7233c97ae74006afab5172976300ba40f5b250a21f8cf71f59c9f76d", @@ -287,7 +325,7 @@ class Cudnn(Package): .replace("darwin", "osx") .replace("aarch64", "aarch64sbsa") ) - else: + elif version < Version("8.8.0"): sys_key = sys_key.replace("aarch64", "sbsa") if version >= Version("8.3.1"): @@ -318,8 +356,12 @@ class Cudnn(Package): ver = version[:2] cuda = version[2:] + # 8.8.0 changed the base url again + if version >= Version("8.8.0"): + url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/{0}/cudnn-{0}-{1}_cuda{2}-archive.tar.xz" + return url.format(sys_key, ver, cuda.up_to(1)) # 8.5.0 removed minor from cuda version - if version >= Version("8.5.0"): + elif version >= Version("8.5.0"): url = "https://developer.download.nvidia.com/compute/redist/cudnn/v{0}/cudnn-{1}-{2}_cuda{3}-archive.tar.xz" return url.format(directory, sys_key, ver, cuda.up_to(1)) # 8.3.1 switched to xzip tarballs and reordered url parts. |