diff options
author | Brian Van Essen <vanessen1@llnl.gov> | 2020-04-27 13:15:46 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-04-27 13:15:46 -0700 |
commit | a4b17b95036102cba5e662449217ec33d64c6df7 (patch) | |
tree | 8101fe9bb63da5b103295273c2a46a62fab51e20 | |
parent | 1c45153e00fe6c31a58736e781565fe094b5a433 (diff) | |
download | spack-a4b17b95036102cba5e662449217ec33d64c6df7.tar.gz spack-a4b17b95036102cba5e662449217ec33d64c6df7.tar.bz2 spack-a4b17b95036102cba5e662449217ec33d64c6df7.tar.xz spack-a4b17b95036102cba5e662449217ec33d64c6df7.zip |
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.
-rw-r--r-- | var/spack/repos/builtin/packages/cudnn/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
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) |