diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2023-04-08 12:05:34 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-08 12:05:34 -0500 |
commit | 6de3786c36053f413595986e6a91b94e23b3e6b1 (patch) | |
tree | 22c6eae250f955089d743c7170c88eeede51c6aa | |
parent | 3869761216ce065da190c58622cd1323c3024952 (diff) | |
download | spack-6de3786c36053f413595986e6a91b94e23b3e6b1.tar.gz spack-6de3786c36053f413595986e6a91b94e23b3e6b1.tar.bz2 spack-6de3786c36053f413595986e6a91b94e23b3e6b1.tar.xz spack-6de3786c36053f413595986e6a91b94e23b3e6b1.zip |
cuDNN: LD_LIBRARY_PATH required (#36466)
-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) |