diff options
author | Andrew W Elble <aweits@rit.edu> | 2020-05-14 21:04:06 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-05-14 20:04:06 -0500 |
commit | 8c025324af19afa11be31d9eee3788707c149693 (patch) | |
tree | b28a328be235cb93c5fe7aa592b00ffbb224db9a | |
parent | 0ed2fa3b9b4beac50bdfc82658a3bdfb395ae104 (diff) | |
download | spack-8c025324af19afa11be31d9eee3788707c149693.tar.gz spack-8c025324af19afa11be31d9eee3788707c149693.tar.bz2 spack-8c025324af19afa11be31d9eee3788707c149693.tar.xz spack-8c025324af19afa11be31d9eee3788707c149693.zip |
ucx: cuda / gdrcopy variants (#16605)
picking up where #15724 left off.
-rw-r--r-- | var/spack/repos/builtin/packages/ucx/package.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/ucx/package.py b/var/spack/repos/builtin/packages/ucx/package.py index b345107c29..2b3cfa4ab8 100644 --- a/var/spack/repos/builtin/packages/ucx/package.py +++ b/var/spack/repos/builtin/packages/ucx/package.py @@ -6,7 +6,7 @@ from spack import * -class Ucx(AutotoolsPackage): +class Ucx(AutotoolsPackage, CudaPackage): """a communication library implementing high-performance messaging for MPI/PGAS frameworks""" @@ -48,11 +48,16 @@ class Ucx(AutotoolsPackage): description='Builds with PIC support') variant('java', default=False, description='Builds with Java bindings') + variant('gdrcopy', default=False, + description='Enable gdrcopy support') depends_on('numactl') depends_on('rdma-core') depends_on('java@8', 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): spec = self.spec @@ -92,4 +97,16 @@ class Ucx(AutotoolsPackage): else: 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 |