From 5dd9a321a402a69415709bd0a320e06e892a4b42 Mon Sep 17 00:00:00 2001 From: Jon Rood Date: Thu, 13 Dec 2018 00:05:13 -0700 Subject: Add CUDA_HOME variable to cuda module file. (#10088) --- var/spack/repos/builtin/packages/cuda/package.py | 3 +++ 1 file changed, 3 insertions(+) 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') -- cgit v1.2.3-60-g2f50