python-apex-git

PKGBUILD

# Maintainer: Butui Hu <hot123tea123@gmail.com>

pkgname=python-apex-git
pkgver=r376.bd0db55
pkgrel=1
pkgdesc="A PyTorch Extension for easy mixed precision and distributed training in PyTorch"
arch=('x86_64')
url="https://github.com/NVIDIA/apex"
license=('BSD')
depends=(cuda python-pytorch-cuda)
makedepends=(git python-setuptools)
checkdepends=(python-pytest)
provides=(python-apex)
conflicts=(python-apex)
source=("${pkgname}::git+https://github.com/NVIDIA/apex.git")
md5sums=('SKIP')

pkgver() {
	cd "${srcdir}/${pkgname}"

# Git, no tags available
	printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"
}

build() {
	cd "${srcdir}/${pkgname}"
	python setup.py --cuda_ext --cpp_ext build
}

package() {
	cd "${srcdir}/${pkgname}"
	python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
 	install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
}
# vim:set ts=2 sw=2 et:
# Maintainer: Butui Hu <hot123tea123@gmail.com>

pkgname=python-apex-git
pkgver=r376.bd0db55
pkgrel=1
pkgdesc="A PyTorch Extension for easy mixed precision and distributed training in PyTorch"
arch=('x86_64')
url="https://github.com/NVIDIA/apex"
license=('BSD')
depends=(cuda python-pytorch-cuda)
makedepends=(git python-setuptools)
checkdepends=(python-pytest)
provides=(python-apex)
conflicts=(python-apex)
source=("${pkgname}::git+https://github.com/NVIDIA/apex.git")
md5sums=('SKIP')

pkgver() {
	cd "${srcdir}/${pkgname}"

# Git, no tags available
	printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"
}

build() {
	cd "${srcdir}/${pkgname}"
	python setup.py --cuda_ext --cpp_ext build
}

package() {
	cd "${srcdir}/${pkgname}"
	python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
 	install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
}
# vim:set ts=2 sw=2 et: