hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

ucx: added variants to support IB, mlx etc (#21876)

Browse source 
Co-authored-by: mohan babu <mohbabul@amd.com>
This commit is contained in:
AMD Toolchain Support 2021-03-01 20:41:37 +05:30 committed by GitHub
parent c529ccaab0
commit afb7e57d20
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 34 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
var/spack/repos/builtin/packages/ucx/package.py
View file

@ -60,6 +60,22 @@ class Ucx(AutotoolsPackage, CudaPackage):
description='Enable KNEM support') description='Enable KNEM support')
variant('xpmem', default=False, variant('xpmem', default=False,
description='Enable XPMEM support') description='Enable XPMEM support')
variant('cma', default=False,
description="nable Cross Memory Attach")
variant('rc', default=False,
description="Compile with IB Reliable Connection support")
variant('dc', default=False,
description="Compile with IB Dynamic Connection support")
variant('ud', default=False,
description="Compile with IB Unreliable Datagram support")
variant('mlx5-dv', default=False,
description="Compile with mlx5 Direct Verbs support")
variant('ib-hw-tm', default=False,
description="Compile with IB Tag Matching support")
variant('dm', default=False,
description="Compile with Device Memory support")
variant('cm', default=False,
description="Compile with IB Connection Manager support")
depends_on('numactl') depends_on('numactl')
depends_on('rdma-core') depends_on('rdma-core')
@ -88,16 +104,34 @@ def configure_args(self):
else: else:
config_args.append('--disable-mt') config_args.append('--disable-mt')
if '+cma' in spec:
config_args.append('--enable-cma')
else:
config_args.append('--disable-cma')
if '+paramter_checking' in spec: if '+paramter_checking' in spec:
config_args.append('--enable-params-check') config_args.append('--enable-params-check')
else: else:
config_args.append('--disable-params-check') config_args.append('--disable-params-check')
# Activate SIMD based on properties of the target
if 'avx' in self.spec.target:
config_args.append('--with-avx')
else:
config_args.append('--without-avx')
config_args.extend(self.enable_or_disable('optimizations')) config_args.extend(self.enable_or_disable('optimizations'))
config_args.extend(self.enable_or_disable('assertions')) config_args.extend(self.enable_or_disable('assertions'))
config_args.extend(self.enable_or_disable('logging')) config_args.extend(self.enable_or_disable('logging'))
config_args.extend(self.with_or_without('pic')) config_args.extend(self.with_or_without('pic'))
config_args.extend(self.with_or_without('rc'))
config_args.extend(self.with_or_without('ud'))
config_args.extend(self.with_or_without('dc'))
config_args.extend(self.with_or_without('mlx5-dv'))
config_args.extend(self.with_or_without('ib-hw-tm'))
config_args.extend(self.with_or_without('dm'))
config_args.extend(self.with_or_without('cm'))
config_args.extend(self.with_or_without('java', config_args.extend(self.with_or_without('java',
activation_value='prefix')) activation_value='prefix'))
config_args.extend(self.with_or_without('cuda', config_args.extend(self.with_or_without('cuda',