summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorlilac2023-05-13 04:48:59 +0800
committerlilac2023-05-13 04:48:59 +0800
commit15e4eddcaecba90ab9160a89cdd6ff6a256d131b (patch)
tree6455736e90e3ccd9c147218f126371b558b03c15
parentb2323568bdfc2194c22202063167213ca737166d (diff)
downloadaur-15e4eddcaecba90ab9160a89cdd6ff6a256d131b.tar.gz
[lilac] updated to 0.28.0-1
-rw-r--r--.SRCINFO8
-rw-r--r--PKGBUILD23
2 files changed, 19 insertions, 12 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 6c905032c1c1..881ab05a04ae 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,16 +1,18 @@
pkgbase = python-horovod
pkgdesc = Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
- pkgver = 0.27.0
+ pkgver = 0.28.0
pkgrel = 1
url = https://github.com/horovod/horovod
arch = x86_64
license = Apache
makedepends = cmake
makedepends = git
- makedepends = python-pip
+ makedepends = python-build
+ makedepends = python-installer
makedepends = python-pytorch-cuda
makedepends = python-setuptools
makedepends = python-tensorflow-cuda
+ makedepends = python-wheel
depends = cuda
depends = cudnn
depends = nccl
@@ -27,7 +29,7 @@ pkgbase = python-horovod
depends = python-tqdm
optdepends = python-pytorch-cuda: pytorch framework
optdepends = python-tensorflow-cuda: tensorflow framework
- source = horovod-0.27.0::git+https://github.com/horovod/horovod.git#tag=v0.27.0
+ source = horovod-0.28.0::git+https://github.com/horovod/horovod.git#tag=v0.28.0
sha512sums = SKIP
pkgname = python-horovod
diff --git a/PKGBUILD b/PKGBUILD
index 8d6640aa84f6..b429f9cc097d 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,7 +2,7 @@
_pkgname=horovod
pkgname=python-horovod
-pkgver=0.27.0
+pkgver=0.28.0
pkgrel=1
pkgdesc='Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet'
arch=('x86_64')
@@ -27,10 +27,12 @@ depends=(
makedepends=(
cmake
git
- python-pip
+ python-build
+ python-installer
python-pytorch-cuda
python-setuptools
python-tensorflow-cuda
+ python-wheel
)
optdepends=(
'python-pytorch-cuda: pytorch framework'
@@ -43,29 +45,32 @@ prepare() {
cd "${_pkgname}-${pkgver}"
git submodule update --init --recursive
# modify these environment variable as you need, see also https://github.com/horovod/horovod/blob/master/docs/install.rst
- export HOROVOD_CUDA_HOME=/opt/cuda
+ export HOROVOD_BUILD_CUDA_CC_LIST="60,61,62,70,72,75,80,86,89,90"
export HOROVOD_CPU_OPERATIONS=GLOO
+ export HOROVOD_CUDA_HOME=/opt/cuda
export HOROVOD_GPU=CUDA
- export HOROVOD_GPU_ALLREDUCE=NCCL
- export HOROVOD_GPU_BROADCAST=NCCL
+ export HOROVOD_GPU_OPERATIONS=NCCL
+ export HOROVOD_NCCL_LINK=SHARED
+ export HOROVOD_WITHOUT_MXNET=1
export HOROVOD_WITH_GLOO=1
export HOROVOD_WITH_MPI=1
- export HOROVOD_WITH_MXNET=0
export HOROVOD_WITH_PYTORCH=1
export HOROVOD_WITH_TENSORFLOW=1
- export HOROVOD_BUILD_CUDA_CC_LIST="52,53,60,61,62,70,72,75,80,86"
export CC=/opt/cuda/bin/gcc
export CXX=/opt/cuda/bin/g++
+ # fix https://github.com/horovod/horovod/issues/3923
+ cd "third_party/gloo"
+ git pull https://github.com/facebookincubator/gloo.git
}
build() {
cd "${_pkgname}-${pkgver}"
- python setup.py build
+ python -m build --wheel --no-isolation -x
}
package() {
cd "${_pkgname}-${pkgver}"
- python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
+ python -m installer --destdir="${pkgdir}" dist/*.whl
install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
}
# vim:set ts=2 sw=2 et: