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

ucx:add rocm variant (#26992)

Browse Source 
Co-authored-by: mahendrapaipuri <mahendra.paipuri@inria.fr>
This commit is contained in:
Mahendra Paipuri 2021-11-01 15:05:56 +01:00 committed by GitHub
parent 80ba7040f8
commit be0df5c47a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,6 +65,8 @@ class Ucx(AutotoolsPackage, CudaPackage):
description='Enable XPMEM support') description='Enable XPMEM support')
variant('cma', default=False, variant('cma', default=False,
description="Enable Cross Memory Attach") description="Enable Cross Memory Attach")
variant('rocm', default=False,
description="Enable ROCm support")
variant('rc', default=False, variant('rc', default=False,
description="Compile with IB Reliable Connection support") description="Compile with IB Reliable Connection support")
variant('dc', default=False, variant('dc', default=False,
@ -89,6 +91,8 @@ class Ucx(AutotoolsPackage, CudaPackage):
depends_on('gdrcopy@1.3', when='@:1.6+gdrcopy') depends_on('gdrcopy@1.3', when='@:1.6+gdrcopy')
conflicts('+gdrcopy', when='~cuda', conflicts('+gdrcopy', when='~cuda',
msg='gdrcopy currently requires cuda support') msg='gdrcopy currently requires cuda support')
conflicts('+rocm', when='+gdrcopy',
msg='gdrcopy > 2.0 does not support rocm')
depends_on('xpmem', when='+xpmem') depends_on('xpmem', when='+xpmem')
depends_on('knem', when='+knem') depends_on('knem', when='+knem')
depends_on('binutils+ld', when='%aocc', type='build') depends_on('binutils+ld', when='%aocc', type='build')
@ -136,6 +140,7 @@ def configure_args(self):
config_args.extend(self.with_or_without('ib-hw-tm')) 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('dm'))
config_args.extend(self.with_or_without('cm')) config_args.extend(self.with_or_without('cm'))
config_args.extend(self.with_or_without('rocm'))
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',