diff options
| author | Robert Underwood <robertu94@users.noreply.github.com> | 2024-09-06 01:12:46 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-05 23:12:46 -0600 |
| commit | 9d754c127a2ea70d6882ec2ce730375ea9272959 (patch) | |
| tree | 9f16ebfbb882f8e958866354468cb80c41e7d8d8 /var | |
| parent | 9a8bff01adc1239c4bea338f96ea3222162b621e (diff) | |
| download | spack-9d754c127a2ea70d6882ec2ce730375ea9272959.tar.gz spack-9d754c127a2ea70d6882ec2ce730375ea9272959.tar.bz2 spack-9d754c127a2ea70d6882ec2ce730375ea9272959.tar.xz spack-9d754c127a2ea70d6882ec2ce730375ea9272959.zip | |
Require a newer version of cudnn for cupy (#46247)
cupy 12 needs a newer version of cudnn as documented here. Supercedes #46128
Co-authored-by: Robert Underwood <runderwood@anl.gov>
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-cupy/package.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-cupy/package.py b/var/spack/repos/builtin/packages/py-cupy/package.py index 8865cb6806..1e54a36779 100644 --- a/var/spack/repos/builtin/packages/py-cupy/package.py +++ b/var/spack/repos/builtin/packages/py-cupy/package.py @@ -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") |