diff options
| author | Taillefumier Mathieu <29380261+mtaillefumier@users.noreply.github.com> | 2023-10-23 21:37:42 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-23 12:37:42 -0700 |
| commit | 1e4a5791b28a506cb3843e95c9cec5860a104600 (patch) | |
| tree | ee76d1ae4322572fc19245b90515b1c58fe3332e | |
| parent | 8def7f5583e0bb52aff735cd3737e7080ab4993d (diff) | |
| download | spack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.gz spack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.bz2 spack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.xz spack-1e4a5791b28a506cb3843e95c9cec5860a104600.zip | |
Add rccl and nccl variants to cp2k and cosma (#40451)
| -rw-r--r-- | var/spack/repos/builtin/packages/cosma/package.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/cosma/package.py b/var/spack/repos/builtin/packages/cosma/package.py index 2fccafe187..19db9a0531 100644 --- a/var/spack/repos/builtin/packages/cosma/package.py +++ b/var/spack/repos/builtin/packages/cosma/package.py @@ -48,6 +48,9 @@ class Cosma(CMakePackage): with when("+cuda"): variant("nccl", default=False, description="Use cuda nccl") + with when("+rocm"): + variant("rccl", default=False, description="Use rocm rccl") + depends_on("cmake@3.22:", type="build") depends_on("mpi@3:") depends_on("blas", when="~cuda ~rocm") @@ -114,6 +117,7 @@ class Cosma(CMakePackage): self.define_from_variant("COSMA_WITH_TESTS", "tests"), self.define_from_variant("COSMA_WITH_APPS", "apps"), self.define_from_variant("COSMA_WITH_NCCL", "nccl"), + self.define_from_variant("COSMA_WITH_RCCL", "rccl"), self.define_from_variant("COSMA_WITH_GPU_AWARE_MPI", "gpu_direct"), self.define_from_variant("COSMA_WITH_PROFILING", "profiling"), self.define("COSMA_WITH_BENCHMARKS", False), |