summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorDominic Hofer <6570912+dominichofer@users.noreply.github.com>2023-10-26 01:22:22 +0200
committerGitHub <noreply@github.com>2023-10-25 16:22:22 -0700
commitd9edc92119446f0fadec634b5c2f437fb1f83482 (patch)
tree13774d64b2b0d5febf650b012bda0f697530b4ba /var
parent2a245fdd21a6a5657e59527df6ef1f3914a74222 (diff)
downloadspack-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>
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/cuda/package.py3
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"):