Add CUDA_HOME variable to cuda module file. (#10088)

2018-12-13 00:05:13 -07:00 · 2018-12-13 00:05:13 -07:00 · 5dd9a321a4
commit 5dd9a321a4
parent 0b42cf4d96
1 changed files with 3 additions and 0 deletions
--- 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')