diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2020-10-05 09:22:33 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-10-05 16:22:33 +0200 |
commit | 0d640200073c660a25ea9791ed601824d3ec172f (patch) | |
tree | 3be7d98068f9dc0d74844cfe36fcb8d6e7c63153 /var | |
parent | cb7a1ac2bfed721e1f3273d4541f0f4df77fa1b6 (diff) | |
download | spack-0d640200073c660a25ea9791ed601824d3ec172f.tar.gz spack-0d640200073c660a25ea9791ed601824d3ec172f.tar.bz2 spack-0d640200073c660a25ea9791ed601824d3ec172f.tar.xz spack-0d640200073c660a25ea9791ed601824d3ec172f.zip |
py-horovod: added versions up to v0.20.3 (#18977)
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/py-horovod/package.py | 65 |
1 files changed, 52 insertions, 13 deletions
diff --git a/var/spack/repos/builtin/packages/py-horovod/package.py b/var/spack/repos/builtin/packages/py-horovod/package.py index a4cb5ed1b8..099d0cbf00 100644 --- a/var/spack/repos/builtin/packages/py-horovod/package.py +++ b/var/spack/repos/builtin/packages/py-horovod/package.py @@ -4,16 +4,20 @@ # SPDX-License-Identifier: (Apache-2.0 OR MIT) -class PyHorovod(PythonPackage): +class PyHorovod(PythonPackage, CudaPackage): """Horovod is a distributed deep learning training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.""" homepage = "https://github.com/horovod" git = "https://github.com/horovod/horovod.git" - maintainers = ['adamjstewart'] + maintainers = ['adamjstewart', 'aweits', 'tgaddair'] version('master', branch='master', submodules=True) + version('0.20.3', tag='v0.20.3', submodules=True) + version('0.20.2', tag='v0.20.2', submodules=True) + version('0.20.1', tag='v0.20.1', submodules=True) + version('0.20.0', tag='v0.20.0', submodules=True) version('0.19.5', tag='v0.19.5', submodules=True) version('0.19.4', tag='v0.19.4', submodules=True) version('0.19.3', tag='v0.19.3', submodules=True) @@ -32,7 +36,7 @@ class PyHorovod(PythonPackage): # https://github.com/horovod/horovod/blob/master/docs/install.rst variant('frameworks', default='pytorch', description='Deep learning frameworks to build support for', - values=('tensorflow', 'pytorch', 'mxnet', 'keras', 'spark'), + values=('tensorflow', 'pytorch', 'mxnet', 'keras', 'spark', 'ray'), multi=True) variant('controllers', default='mpi', description='Controllers to coordinate work between processes', @@ -40,17 +44,23 @@ class PyHorovod(PythonPackage): variant('tensor_ops', default='nccl', description='Framework to use for GPU/CPU operations', values=('nccl', 'mpi', 'gloo', 'ccl'), multi=False) + variant('cuda', default=True, description='Build with CUDA') + variant('rocm', default=False, description='Build with ROCm') # Required dependencies + depends_on('python@3.6:', type=('build', 'run'), when='@0.20:') depends_on('py-setuptools', type='build') depends_on('py-cloudpickle', type=('build', 'run')) depends_on('py-psutil', type=('build', 'run')) depends_on('py-pyyaml', type=('build', 'run')) - depends_on('py-six', type=('build', 'run')) + depends_on('py-six', type=('build', 'run'), when='@:0.19') + depends_on('py-dataclasses', type=('build', 'run'), when='@0.20: ^python@:3.6') # Framework dependencies depends_on('py-tensorflow@1.1.0:', type=('build', 'link', 'run'), when='frameworks=tensorflow') + depends_on('py-tensorflow@1.15:', type=('build', 'link', 'run'), when='frameworks=tensorflow @0.20:') depends_on('py-torch@0.4.0:', type=('build', 'link', 'run'), when='frameworks=pytorch') + depends_on('py-torch@1.2:', type=('build', 'link', 'run'), when='frameworks=pytorch @0.20:') depends_on('py-torchvision', type=('build', 'run'), when='frameworks=pytorch @:0.19.1') depends_on('py-cffi@1.4.0:', type=('build', 'run'), when='frameworks=pytorch') depends_on('mxnet@1.4.1:+python', type=('build', 'link', 'run'), when='frameworks=mxnet') @@ -61,11 +71,17 @@ class PyHorovod(PythonPackage): depends_on('py-petastorm@0.9.0:', type=('build', 'run'), when='frameworks=spark @0.19.2:') depends_on('py-pyarrow@0.15.0:', type=('build', 'run'), when='frameworks=spark') depends_on('py-pyspark@2.3.2:', type=('build', 'run'), when='frameworks=spark') + depends_on('py-ray', type=('build', 'run'), when='frameworks=ray') + + # Build dependencies + depends_on('cmake@2.8.12:', type='build', when='@0.20:') + depends_on('pkgconfig', type='build') # Controller dependencies depends_on('mpi', when='controllers=mpi') # There does not appear to be a way to use an external Gloo installation depends_on('cmake', type='build', when='controllers=gloo') + depends_on('libuv@1.26:', when='controllers=gloo platform=darwin') # Tensor Operations dependencies depends_on('nccl@2:', when='tensor_ops=nccl') @@ -77,13 +93,28 @@ class PyHorovod(PythonPackage): depends_on('py-mock', type='test') depends_on('py-pytest', type='test') depends_on('py-pytest-forked', type='test') + depends_on('py-parameterized', type='test', when='@0.20:') - conflicts('controllers=gloo', when='platform=darwin', msg='Gloo cannot be compiled on MacOS') + conflicts('cuda_arch=none', when='+cuda', + msg='Must specify CUDA compute capabilities of your GPU, see ' + 'https://developer.nvidia.com/cuda-gpus') + conflicts('tensor_ops=nccl', when='~cuda~rocm', msg='NCCL requires either CUDA or ROCm support') + conflicts('framework=ray', when='@:0.19', msg='Ray integration was added in 0.20.X') + conflicts('controllers=gloo', when='@:0.20.0 platform=darwin', msg='Gloo cannot be compiled on MacOS') # https://github.com/horovod/horovod/pull/1835 patch('fma.patch', when='@0.19.0:0.19.1') def setup_build_environment(self, env): + # https://github.com/horovod/horovod/blob/master/docs/install.rst#environment-variables + + # Build system + env.set('PKG_CONFIG_EXECUTABLE', + self.spec['pkgconfig'].prefix.bin.join('pkg-config')) + if '^cmake' in self.spec: + env.set('HOROVOD_CMAKE', self.spec['cmake'].command.path) + env.set('MAKEFLAGS', '-j{0}'.format(make_jobs)) + # Frameworks if 'frameworks=tensorflow' in self.spec: env.set('HOROVOD_WITH_TENSORFLOW', 1) @@ -110,20 +141,28 @@ class PyHorovod(PythonPackage): # Tensor Operations if 'tensor_ops=nccl' in self.spec: - env.set('HOROVOD_GPU', 'CUDA') - - env.set('HOROVOD_CUDA_HOME', self.spec['cuda'].prefix) - env.set('HOROVOD_CUDA_INCLUDE', - self.spec['cuda'].headers.directories[0]) - env.set('HOROVOD_CUDA_LIB', self.spec['cuda'].libs.directories[0]) + env.set('HOROVOD_GPU_ALLREDUCE', 'NCCL') + env.set('HOROVOD_GPU_ALLGATHER', 'NCCL') + env.set('HOROVOD_GPU_BROADCAST', 'NCCL') env.set('HOROVOD_NCCL_HOME', self.spec['nccl'].prefix) env.set('HOROVOD_NCCL_INCLUDE', self.spec['nccl'].headers.directories[0]) env.set('HOROVOD_NCCL_LIB', self.spec['nccl'].libs.directories[0]) - env.set('HOROVOD_GPU_ALLREDUCE', 'NCCL') - env.set('HOROVOD_GPU_BROADCAST', 'NCCL') + if '+cuda' in self.spec: + env.set('HOROVOD_GPU', 'CUDA') + + env.set('HOROVOD_CUDA_HOME', self.spec['cuda'].prefix) + cuda_cc_list = ','.join(self.spec.variants['cuda_arch'].value) + env.set('HOROVOD_BUILD_CUDA_CC_LIST', cuda_cc_list) + env.set('HOROVOD_CUDA_INCLUDE', + self.spec['cuda'].headers.directories[0]) + env.set('HOROVOD_CUDA_LIB', + self.spec['cuda'].libs.directories[0]) + elif '+rocm' in self.spec: + env.set('HOROVOD_GPU', 'ROCM') + # env.set('HOROVOD_ROCM_HOME', self.spec['rocm'].prefix) else: env.set('HOROVOD_CPU_OPERATIONS', self.spec.variants['tensor_ops'].value.upper()) |