37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
# Copyright 2013-2020 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 install(self, spec, prefix):
|
|
mkdirp(prefix.bin)
|
|
install_tree('./build', prefix.bin)
|