From a4b17b95036102cba5e662449217ec33d64c6df7 Mon Sep 17 00:00:00 2001 From: Brian Van Essen Date: Mon, 27 Apr 2020 13:15:46 -0700 Subject: cuDNN: Help dependents find libs/includes on Power arch (#16128) On Power architectures cuDNN will install in a target directory. This sets cuDNN_ROOT to point to the subdirectory to help dependents use this install. --- var/spack/repos/builtin/packages/cudnn/package.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index 4491af18a8..74f5133d90 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -190,6 +190,11 @@ class Cudnn(Package): return url.format(directory, cuda, ver) + def setup_run_environment(self, env): + if 'target=ppc64le: platform=linux' in self.spec: + env.set('cuDNN_ROOT', os.path.join( + self.prefix, 'targets', 'ppc64le-linux')) + def install(self, spec, prefix): install_tree('.', prefix) -- cgit v1.2.3-70-g09d2