diff options
author | rvinaybharadwaj <vinayr@lanl.gov> | 2020-03-11 12:00:02 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-03-11 12:00:02 -0600 |
commit | 87e69eed1c4864730aa221aa014c234e4ac3f83f (patch) | |
tree | df530a0ad8d5eab957e035c5e477051284928f26 /var | |
parent | 20c77cade532844c758290c242536436cecfe922 (diff) | |
download | spack-87e69eed1c4864730aa221aa014c234e4ac3f83f.tar.gz spack-87e69eed1c4864730aa221aa014c234e4ac3f83f.tar.bz2 spack-87e69eed1c4864730aa221aa014c234e4ac3f83f.tar.xz spack-87e69eed1c4864730aa221aa014c234e4ac3f83f.zip |
Horovod (#15380)
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/minigan/package.py | 28 | ||||
-rw-r--r-- | var/spack/repos/builtin/packages/py-horovod/package.py | 60 |
2 files changed, 88 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/minigan/package.py b/var/spack/repos/builtin/packages/minigan/package.py new file mode 100644 index 0000000000..5d3b553bcf --- /dev/null +++ b/var/spack/repos/builtin/packages/minigan/package.py @@ -0,0 +1,28 @@ +# Copyright 2013-2019 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 import * + + +class Minigan(Package): + """miniGAN is a generative adversarial network code developed as part of the + Exascale Computing Project's (ECP) ExaLearn project at + Sandia National Laboratories.""" + + homepage = "https://github.com/SandiaMLMiniApps/miniGAN" + url = "https://github.com/SandiaMLMiniApps/miniGAN/archive/1.0.0.tar.gz" + + version('1.0.0', sha256='ef6d5def9c7040af520acc64b7a8b6c8ec4b7901721b11b0cb25a583ea0c8ae3') + + depends_on('python', type=('build', 'run')) + depends_on('py-setuptools', type='build') + depends_on('py-torch', type=('build', 'run')) + depends_on('py-numpy', type=('build', 'run')) + depends_on('py-horovod@master', type=('build', 'run')) + depends_on('py-torchvision', type=('build', 'run')) + depends_on('py-matplotlib@3.0.0', type=('build', 'run')) + + def install(self, spec, prefix): + install_tree('.', prefix) diff --git a/var/spack/repos/builtin/packages/py-horovod/package.py b/var/spack/repos/builtin/packages/py-horovod/package.py new file mode 100644 index 0000000000..884721b286 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-horovod/package.py @@ -0,0 +1,60 @@ +# Copyright 2013-2019 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 import * +import fnmatch +import os + + +class PyHorovod(PythonPackage): + """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" + url = "https://github.com/horovod/horovod/archive/v0.19.0.tar.gz" + + version('master', branch='master', submodules='True') + version('0.19.0', sha256='0e9fec11cd7f5f39a09f0785d1097cb51c44537ae14c9b4b2578b5cdd21efb9b') + version('0.18.2', sha256='a073e08cec65474afdb2d011486b4cb6c7ac8fcb1eca3e02b169e1e7b4a66da6') + version('0.18.1', sha256='26e236d1f60955e9dd12b9f0a836f0691296a010fcd1ac72295970a780f4e4fb') + version('0.18.0', sha256='94f13e7110c5f3fd1aa194b9d886b5bb91c9bc02ade31bcb84fc6e7f9c043455') + version('0.17.1', sha256='14eea5744eda9c62988ffa278a9a5472cebbc6a287eca9ed48cacfcd177e8978') + version('0.17.0.post1', sha256='220b230611e22dc69777f1be4d9788a07e73a0722e511091fa156cdf68ca798b') + version('0.17.0', sha256='4bb121dda6cdaa1677535470adc1836493a9c4930ab19f6b491254ea47a12a4f') + version('0.16.4', sha256='c0168dfeb31a56ede52eae115f43fa2d06a5db55a37201064ef901c8000d708d') + version('0.16.3', sha256='1857cf1b335723366cc71e4bcd0583f2dde0c821212cda0e1b6bddfe4ba1ea0d') + version('0.16.2', sha256='baa9754e59ab0ee72d3b5769cf77e06a2c7b0a2d9626e0e14ca2ab131934ce74') + + variant('pytorch', default=True, description='Enables PyTorch') + variant('mxnet', default=False, description='Enables mxnet') + variant('mpi', default=True, description='Enables MPI build') + variant('gloo', default=False, description='Enables features related to distributed support') + variant('cuda', default=True, description='Enables CUDA build') + + depends_on('python', type=('build', 'run')) + 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-torch', type=('build', 'run')) + depends_on('py-pip', type=('build')) + + # Optional dependencies + depends_on('cuda', when='+cuda') + depends_on('nccl', when='+nccl') + depends_on('gloo', when='+gloo') + depends_on('mpi', when='+mpi') + depends_on('py-torch', type=('build', 'run'), when='+pytorch') + depends_on('mxnet', when='+mxnet') + + phases = ['clean', 'sdist', 'install'] + + def install(self, spec, prefix): + pip = which('pip') + for file in os.listdir(prefix): + if fnmatch.fnmatch(file, 'horovod-*.tar.gz'): + pip('install', file, '--prefix={0}'.format(prefix)) |