1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack.package import *
class NcclTests(MakefilePackage, CudaPackage):
"""These tests check both the performance and
the correctness of NCCL operations."""
homepage = "https://github.com/NVIDIA/nccl-tests"
url = "https://github.com/NVIDIA/nccl-tests/archive/v2.0.0.tar.gz"
version("2.13.6", sha256="52b472a58a4918d3221a9b8c4bd9335382643d7e241983918b64692d685cc3d1")
version("2.0.0", sha256="731fc3b7c37de59cfe880bf198349ac185639ef23570749ea6aef334c850c49c")
variant("mpi", default=True, description="with MPI support")
variant("cuda", default=True, description="with CUDA support, must be true")
conflicts("~cuda", msg="nccl-tests require cuda")
depends_on("nccl")
depends_on("cuda")
depends_on("mpi", when="+mpi")
@property
def build_targets(self):
targets = []
targets.append("CUDA_HOME={0}".format(self.spec["cuda"].prefix))
targets.append("NCCL_HOME={0}".format(self.spec["nccl"].prefix))
if "+mpi" in self.spec:
targets.append("MPI_HOME={0}".format(self.spec["mpi"].prefix))
targets.append("MPI=1")
return targets
def setup_build_environment(self, env):
cuda_arch = self.spec.variants["cuda_arch"].value
cuda_gencode = " ".join(self.cuda_flags(cuda_arch))
env.set("NVCC_GENCODE", cuda_gencode)
def install(self, spec, prefix):
mkdirp(prefix.bin)
install_tree("./build", prefix.bin)
|