summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorTaillefumier Mathieu <29380261+mtaillefumier@users.noreply.github.com>2023-10-23 21:37:42 +0200
committerGitHub <noreply@github.com>2023-10-23 12:37:42 -0700
commit1e4a5791b28a506cb3843e95c9cec5860a104600 (patch)
treeee76d1ae4322572fc19245b90515b1c58fe3332e /var
parent8def7f5583e0bb52aff735cd3737e7080ab4993d (diff)
downloadspack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.gz
spack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.bz2
spack-1e4a5791b28a506cb3843e95c9cec5860a104600.tar.xz
spack-1e4a5791b28a506cb3843e95c9cec5860a104600.zip
Add rccl and nccl variants to cp2k and cosma (#40451)
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/cosma/package.py4
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),