diff options
| author | Dominic Hofer <6570912+dominichofer@users.noreply.github.com> | 2023-10-26 01:22:22 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-25 16:22:22 -0700 |
| commit | d9edc92119446f0fadec634b5c2f437fb1f83482 (patch) | |
| tree | 13774d64b2b0d5febf650b012bda0f697530b4ba | |
| parent | 2a245fdd21a6a5657e59527df6ef1f3914a74222 (diff) | |
| download | spack-d9edc92119446f0fadec634b5c2f437fb1f83482.tar.gz spack-d9edc92119446f0fadec634b5c2f437fb1f83482.tar.bz2 spack-d9edc92119446f0fadec634b5c2f437fb1f83482.tar.xz spack-d9edc92119446f0fadec634b5c2f437fb1f83482.zip | |
cuda: add NVHPC_CUDA_HOME. (#40507)
* [cuda] Add NVHPC_CUDA_HOME.
* Add CUDA_HOME and NVHC_CUDA_HOME to cuda's dependent build env.
---------
Co-authored-by: Dominic Hofer <dominic.hofer@meteoswiss.ch>
| -rw-r--r-- | var/spack/repos/builtin/packages/cuda/package.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/cuda/package.py b/var/spack/repos/builtin/packages/cuda/package.py index 77bf03da37..e624acbaa9 100644 --- a/var/spack/repos/builtin/packages/cuda/package.py +++ b/var/spack/repos/builtin/packages/cuda/package.py @@ -582,6 +582,8 @@ class Cuda(Package): def setup_dependent_build_environment(self, env, dependent_spec): env.set("CUDAHOSTCXX", dependent_spec.package.compiler.cxx) + env.set("CUDA_HOME", self.prefix) + env.set("NVHPC_CUDA_HOME", self.prefix) @property def cmake_prefix_paths(self): @@ -593,6 +595,7 @@ class Cuda(Package): def setup_run_environment(self, env): env.set("CUDA_HOME", self.prefix) + env.set("NVHPC_CUDA_HOME", self.prefix) def install(self, spec, prefix): if os.path.exists("/tmp/cuda-installer.log"): |