add CUDA_HOME arg to nccl makefile (#3666)
This commit is contained in:
parent
b7eee1bd7b
commit
9b19968bac
@ -36,6 +36,10 @@ class Nccl(MakefilePackage):
|
|||||||
|
|
||||||
depends_on('cuda')
|
depends_on('cuda')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def build_targets(self):
|
||||||
|
return ['CUDA_HOME={0}'.format(self.spec['cuda'].prefix)]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def install_targets(self):
|
def install_targets(self):
|
||||||
return ['PREFIX={0}'.format(self.prefix), 'install']
|
return ['PREFIX={0}'.format(self.prefix), 'install']
|
||||||
|
Loading…
Reference in New Issue
Block a user