Initial upload: python-apex-git 0.1.r217-1

3 files changed, 60 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..b4d02cba5980
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,18 @@
+pkgbase = python-apex-git
+	pkgdesc = A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
+	pkgver = 0.1.r217
+	pkgrel = 1
+	url = https://github.com/NVIDIA/apex
+	arch = any
+	license = BSD
+	makedepends = python-setuptools
+	depends = python
+	depends = cuda
+	depends = python-pytorch
+	provides = python-apex
+	conflicts = python-apex
+	source = git+https://github.com/NVIDIA/apex
+	sha256sums = SKIP
+
+pkgname = python-apex-git
+
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 000000000000..d628f818f530
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+apex
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..3ea877b060f4
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,41 @@
+# Maintainer: Leo Mao
+pkgname=python-apex-git
+_pkgname=apex
+_pkgver=0.1
+pkgver=0.1.r217
+pkgrel=1
+pkgdesc="A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch"
+arch=('any')
+url="https://github.com/NVIDIA/apex"
+_github='NVIDIA/apex'
+license=('BSD')
+depends=('python' 'cuda' 'python-pytorch')
+makedepends=('python-setuptools')
+provides=('python-apex')
+conflicts=('python-apex')
+source=("git+$url")
+sha256sums=('SKIP')
+
+pkgver () {
+  cd "${_pkgname}"
+  (
+    set -o pipefail
+    rev=$(git rev-list --count HEAD 2>/dev/null)
+    echo $_pkgver.r$rev
+  )
+}
+
+build() {
+  msg "Building Python 3"
+  cd "$srcdir/${_pkgname}"
+  CC=/opt/cuda/bin/gcc CXX=/opt/cuda/bin/g++ \
+  python setup.py build --cuda_ext --cpp_ext
+}
+
+package() {
+  cd "$srcdir/${_pkgname}"
+  python setup.py install --cuda_ext --cpp_ext --root="$pkgdir"/ --optimize=1
+  install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
+}
+
+# vim:set ts=2 sw=2 et: