From 8c025324af19afa11be31d9eee3788707c149693 Mon Sep 17 00:00:00 2001 From: Andrew W Elble Date: Thu, 14 May 2020 21:04:06 -0400 Subject: ucx: cuda / gdrcopy variants (#16605) picking up where #15724 left off. --- var/spack/repos/builtin/packages/ucx/package.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) (limited to 'var') 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 -- cgit v1.2.3-60-g2f50