zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Require a newer version of cudnn for cupy (#46247)

Browse Source 
cupy 12 needs a newer version of cudnn as documented here.  Supercedes #46128

Co-authored-by: Robert Underwood <runderwood@anl.gov>
This commit is contained in:
Robert Underwood 2024-09-06 01:12:46 -04:00 committed by GitHub
parent 9a8bff01ad
commit 9d754c127a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
var/spack/repos/builtin/packages/py-cupy/package.py
View File

@ -53,7 +53,8 @@ class PyCupy(PythonPackage, CudaPackage, ROCmPackage):
for a in CudaPackage.cuda_arch_values:
depends_on("nccl +cuda cuda_arch={0}".format(a), when="+cuda cuda_arch={0}".format(a))
depends_on("cudnn", when="+cuda")
depends_on("cudnn@8.8", when="@12.0.0: +cuda")
depends_on("cudnn@8.5", when="@11.2.0:11.6.0 +cuda")
depends_on("cutensor", when="@:12.1.0 +cuda")
depends_on("cutensor@2.0.1.2", when="@13.1: +cuda")