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
|
# Copyright 2013-2021 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)
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.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)
|