From afb7e57d202bd427724ecdec5c9c4bffbed4b21d Mon Sep 17 00:00:00 2001 From: AMD Toolchain Support <73240730+amd-toolchain-support@users.noreply.github.com> Date: Mon, 1 Mar 2021 20:41:37 +0530 Subject: [PATCH] ucx: added variants to support IB, mlx etc (#21876) Co-authored-by: mohan babu --- .../repos/builtin/packages/ucx/package.py | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/var/spack/repos/builtin/packages/ucx/package.py b/var/spack/repos/builtin/packages/ucx/package.py index 3abc53cb04..cd058f9581 100644 --- a/var/spack/repos/builtin/packages/ucx/package.py +++ b/var/spack/repos/builtin/packages/ucx/package.py @@ -60,6 +60,22 @@ class Ucx(AutotoolsPackage, CudaPackage): description='Enable KNEM support') variant('xpmem', default=False, 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('rdma-core') @@ -88,16 +104,34 @@ def configure_args(self): else: 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: config_args.append('--enable-params-check') else: 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('assertions')) config_args.extend(self.enable_or_disable('logging')) 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', activation_value='prefix')) config_args.extend(self.with_or_without('cuda',