# IT4Innovations 2018

easyblock = "PythonPackage"
name = 'horovod'
version = '0.11.3'

homepage = 'https://pypi.python.org/pypi/horovod'
description = 'Distributed training framework for TensorFlow.'

toolchain = {'name': 'Py', 'version': '3.6'}

source_urls = [PYPI_SOURCE]
sources = [SOURCE_TAR_GZ]

dependencies = [
    ('Keras', '2.1.4', '-Tensorflow-1.6.0rc0-CUDA-9.0.176'),
    ('OpenMPI', '1.10.7', '', ('GCC', '6.3.0-2.27')),
    ('NCCL', '2.1.4', '', True)
]

prebuildopts = 'HOROVOD_NCCL_HOME=/apps/all/NCCL/2.1.4 HOROVOD_GPU_ALLREDUCE=NCCL '

modextravars = {
    'HOROVOD_NCCL_HOME': '/apps/all/NCCL/2.1.4',
    'HOROVOD_GPU_ALLREDUCE': 'NCCL',
}

sanity_check_paths = {
    'files': [],
    'dirs': ['lib/python3.6/site-packages'],
}

moduleclass = 'python'
