add CUDA_HOME arg to nccl makefile (#3666)

2017-04-03 14:08:28 +01:00 · 2017-04-03 14:08:28 +01:00 · 9b19968bac
commit 9b19968bac
parent b7eee1bd7b
1 changed files with 4 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/nccl/package.py
+++ b/var/spack/repos/builtin/packages/nccl/package.py
@ -36,6 +36,10 @@ class Nccl(MakefilePackage):

    depends_on('cuda')

+    @property
+    def build_targets(self):
+        return ['CUDA_HOME={0}'.format(self.spec['cuda'].prefix)]
+
    @property
    def install_targets(self):
        return ['PREFIX={0}'.format(self.prefix), 'install']