diff options
-rw-r--r-- | var/spack/repos/builtin/packages/cudnn/package.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index cfe5641489..29234bdfdc 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -307,9 +307,18 @@ class Cudnn(Package): return url.format(directory, cuda, sys_key, ver) def setup_run_environment(self, env): + # Package is not compiled, and does not work unless LD_LIBRARY_PATH is set + env.prepend_path("LD_LIBRARY_PATH", self.prefix.lib) + if "target=ppc64le: platform=linux" in self.spec: env.set("cuDNN_ROOT", os.path.join(self.prefix, "targets", "ppc64le-linux")) + def setup_dependent_build_environment(self, env, dependent_spec): + self.setup_run_environment(env) + + def setup_dependent_run_environment(self, env, dependent_spec): + self.setup_run_environment(env) + def install(self, spec, prefix): install_tree(".", prefix) |