pkgbase = python-horovod
	pkgdesc = Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
	pkgver = 0.20.3
	pkgrel = 2
	url = https://github.com/horovod/horovod
	arch = x86_64
	license = Apache
	makedepends = cmake
	makedepends = git
	makedepends = mxnet-cuda
	makedepends = python-pytorch-cuda
	makedepends = python-setuptools
	makedepends = python-tensorflow-cuda
	depends = cuda
	depends = cudnn
	depends = nccl
	depends = openmpi
	depends = python-cloudpickle
	depends = python-h5py
	depends = python-mock
	depends = python-mpi4py
	depends = python-psutil
	depends = python-pycparser
	depends = python-pyaml
	depends = python-scipy
	depends = python-six
	depends = python-tqdm
	optdepends = mxnet-cuda: mxnet framework
	optdepends = python-pytorch-cuda: pytorch framework
	optdepends = python-tensorflow-cuda: tensorflow framework
	source = horovod-0.20.3::git+https://github.com/horovod/horovod.git#tag=v0.20.3
	sha512sums = SKIP

pkgname = python-horovod