diff options
| author | Mahendra Paipuri <44365948+mahendrapaipuri@users.noreply.github.com> | 2021-11-01 15:05:56 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-01 15:05:56 +0100 |
| commit | be0df5c47a60e50c93992b67c1d64920b31cd80b (patch) | |
| tree | 55036c5b65c8bd89247f365a9a6da60fab566464 | |
| parent | 80ba7040f8bce05af3f9cb561f8192c120f3e874 (diff) | |
| download | spack-be0df5c47a60e50c93992b67c1d64920b31cd80b.tar.gz spack-be0df5c47a60e50c93992b67c1d64920b31cd80b.tar.bz2 spack-be0df5c47a60e50c93992b67c1d64920b31cd80b.tar.xz spack-be0df5c47a60e50c93992b67c1d64920b31cd80b.zip | |
ucx:add rocm variant (#26992)
Co-authored-by: mahendrapaipuri <mahendra.paipuri@inria.fr>
| -rw-r--r-- | var/spack/repos/builtin/packages/ucx/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/ucx/package.py b/var/spack/repos/builtin/packages/ucx/package.py index 1ec325745d..498384f915 100644 --- a/var/spack/repos/builtin/packages/ucx/package.py +++ b/var/spack/repos/builtin/packages/ucx/package.py @@ -65,6 +65,8 @@ class Ucx(AutotoolsPackage, CudaPackage): description='Enable XPMEM support') variant('cma', default=False, description="Enable Cross Memory Attach") + variant('rocm', default=False, + description="Enable ROCm support") variant('rc', default=False, description="Compile with IB Reliable Connection support") variant('dc', default=False, @@ -89,6 +91,8 @@ class Ucx(AutotoolsPackage, CudaPackage): depends_on('gdrcopy@1.3', when='@:1.6+gdrcopy') conflicts('+gdrcopy', when='~cuda', 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('knem', when='+knem') depends_on('binutils+ld', when='%aocc', type='build') @@ -136,6 +140,7 @@ class Ucx(AutotoolsPackage, CudaPackage): 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('rocm')) config_args.extend(self.with_or_without('java', activation_value='prefix')) config_args.extend(self.with_or_without('cuda', |