From 9d754c127a2ea70d6882ec2ce730375ea9272959 Mon Sep 17 00:00:00 2001 From: Robert Underwood Date: Fri, 6 Sep 2024 01:12:46 -0400 Subject: 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 --- var/spack/repos/builtin/packages/py-cupy/package.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'var') 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") -- cgit v1.2.3-70-g09d2