hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add rccl and nccl variants to cp2k and cosma (#40451)

Browse source
This commit is contained in:
Taillefumier Mathieu 2023-10-23 21:37:42 +02:00 committed by GitHub
parent 8def7f5583
commit 1e4a5791b2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
var/spack/repos/builtin/packages/cosma/package.py
View file

@ -48,6 +48,9 @@ class Cosma(CMakePackage):
with when("+cuda"): with when("+cuda"):
variant("nccl", default=False, description="Use cuda nccl") 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("cmake@3.22:", type="build")
depends_on("mpi@3:") depends_on("mpi@3:")
depends_on("blas", when="~cuda ~rocm") depends_on("blas", when="~cuda ~rocm")
@ -114,6 +117,7 @@ def cmake_args(self):
self.define_from_variant("COSMA_WITH_TESTS", "tests"), self.define_from_variant("COSMA_WITH_TESTS", "tests"),
self.define_from_variant("COSMA_WITH_APPS", "apps"), self.define_from_variant("COSMA_WITH_APPS", "apps"),
self.define_from_variant("COSMA_WITH_NCCL", "nccl"), 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_GPU_AWARE_MPI", "gpu_direct"),
self.define_from_variant("COSMA_WITH_PROFILING", "profiling"), self.define_from_variant("COSMA_WITH_PROFILING", "profiling"),
self.define("COSMA_WITH_BENCHMARKS", False), self.define("COSMA_WITH_BENCHMARKS", False),