summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJon Rood <jon.rood@nrel.gov>2018-12-13 00:05:13 -0700
committerMassimiliano Culpo <massimiliano.culpo@gmail.com>2018-12-13 08:05:13 +0100
commit5dd9a321a402a69415709bd0a320e06e892a4b42 (patch)
tree82fac1b750e762dccd0593c41bde822cdd1e4b28
parent0b42cf4d96b306e9356b102626c294a85a2b6d1b (diff)
downloadspack-5dd9a321a402a69415709bd0a320e06e892a4b42.tar.gz
spack-5dd9a321a402a69415709bd0a320e06e892a4b42.tar.bz2
spack-5dd9a321a402a69415709bd0a320e06e892a4b42.tar.xz
spack-5dd9a321a402a69415709bd0a320e06e892a4b42.zip
Add CUDA_HOME variable to cuda module file. (#10088)
-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 b807ddd966..4bfcffccde 100644
--- a/var/spack/repos/builtin/packages/cuda/package.py
+++ b/var/spack/repos/builtin/packages/cuda/package.py
@@ -35,6 +35,9 @@ class Cuda(Package):
version('6.5.14', '90b1b8f77313600cc294d9271741f4da', expand=False,
url="http://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda_6.5.14_linux_64.run")
+ def setup_environment(self, spack_env, run_env):
+ run_env.set('CUDA_HOME', self.prefix)
+
def install(self, spec, prefix):
runfile = glob(join_path(self.stage.path, 'cuda*_linux*'))[0]
chmod = which('chmod')