hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

CUDNN: Add versions 8.8.x and 8.9.x. (#41790)

Browse source 
NVIDIA changed the download URLs again so account for that.
This commit is contained in:
Rémi Lacroix 2023-12-23 21:20:08 +01:00 committed by GitHub
parent 13806213d9
commit b50f131a01
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 44 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
var/spack/repos/builtin/packages/cudnn/package.py
View file

@ -9,6 +9,44 @@
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 @@ def url_for_version(self, version):
.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 @@ def url_for_version(self, version):
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.