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>
This commit is contained in:
parent
2a245fdd21
commit
d9edc92119
1 changed files with 3 additions and 0 deletions
|
@ -582,6 +582,8 @@ def setup_build_environment(self, env):
|
|||
|
||||
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 @@ def cmake_prefix_paths(self):
|
|||
|
||||
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"):
|
||||
|
|
Loading…
Reference in a new issue