zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ucx: cuda / gdrcopy variants (#16605)

Browse Source 
picking up where #15724 left off.
This commit is contained in:
Andrew W Elble 2020-05-14 21:04:06 -04:00 committed by GitHub
parent 0ed2fa3b9b
commit 8c025324af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
var/spack/repos/builtin/packages/ucx/package.py
View File

@ -6,7 +6,7 @@
from spack import * from spack import *
class Ucx(AutotoolsPackage): class Ucx(AutotoolsPackage, CudaPackage):
"""a communication library implementing high-performance messaging for """a communication library implementing high-performance messaging for
MPI/PGAS frameworks""" MPI/PGAS frameworks"""
@ -48,11 +48,16 @@ class Ucx(AutotoolsPackage):
description='Builds with PIC support') description='Builds with PIC support')
variant('java', default=False, variant('java', default=False,
description='Builds with Java bindings') description='Builds with Java bindings')
variant('gdrcopy', default=False,
description='Enable gdrcopy support')
depends_on('numactl') depends_on('numactl')
depends_on('rdma-core') depends_on('rdma-core')
depends_on('java@8', when='+java') depends_on('java@8', when='+java')
depends_on('maven', when='+java') depends_on('maven', when='+java')
depends_on('gdrcopy@1.3', when='+gdrcopy')
conflicts('+gdrcopy', when='~cuda',
msg='gdrcopy currently requires cuda support')
def configure_args(self): def configure_args(self):
spec = self.spec spec = self.spec
@ -92,4 +97,16 @@ def configure_args(self):
else: else:
config_args.append('--without-java') config_args.append('--without-java')
if '+cuda' in spec:
config_args.append('--with-cuda={0}'.format(
self.spec['cuda'].prefix))
else:
config_args.append('--without-cuda')
if '+gdrcopy' in spec:
config_args.append('--with-gdrcopy={0}'.format(
self.spec['gdrcopy'].prefix))
else:
config_args.append('--without-gdrcopy')
return config_args return config_args