python-tensorflow-addons-cuda-git

2 files changed, 61 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..c6253c493c0f
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,21 @@
+pkgbase = python-tensorflow-addons-cuda-git
+	pkgdesc = Useful extra functionality for TensorFlow 2.x
+	pkgver = r746.25959474
+	pkgrel = 1
+	url = https://github.com/tensorflow/addons
+	arch = any
+	license = APACHE
+	makedepends = python-setuptools
+	makedepends = bazel
+	makedepends = rsync
+	depends = tensorflow-cuda
+	depends = python-tensorflow
+	depends = cuda>=10.2
+	depends = cuda<10.3
+	depends = cudnn>=7
+	depends = cudnn<8
+	source = git+https://github.com/tensorflow/addons.git
+	sha256sums = SKIP
+
+pkgname = python-tensorflow-addons-cuda-git
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..ff73ca288d8a
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,40 @@
+# Maintainer: piernov <piernov@piernov.org>
+
+pkgname=python-tensorflow-addons-cuda-git
+pkgver=r746.25959474
+pkgrel=1
+pkgdesc="Useful extra functionality for TensorFlow 2.x"
+url="https://github.com/tensorflow/addons"
+arch=('any')
+license=('APACHE')
+depends=('tensorflow-cuda' 'python-tensorflow' 'cuda>=10.2' 'cuda<10.3' 'cudnn>=7' 'cudnn<8')
+makedepends=('python-setuptools' 'bazel' 'rsync')
+source=("git+https://github.com/tensorflow/addons.git")
+sha256sums=('SKIP')
+
+pkgver() {
+  cd addons
+  printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"
+}
+
+build() {
+  # Force OpenJDK-11
+  export JAVA_HOME=/usr/lib/jvm/java-11-openjdk
+
+  export TF_NEED_CUDA=1
+  export TF_CUDA_VERSION=10.2
+  export TF_CUDNN_VERSION=7
+  export CUDA_TOOLKIT_PATH=/opt/cuda
+  export CUDNN_INSTALL_PATH=/usr
+  export TF_CUDA_PATHS=/opt/cuda/targets/x86_64-linux
+
+  cd "${srcdir}"/addons
+  python3 ./configure.py --no-deps
+  bazel build --copt=-O3 --copt=-march=native build_pip_pkg
+  bazel-bin/build_pip_pkg artifacts
+}
+
+package() {
+  cd "${srcdir}"/addons
+  PIP_CONFIG_FILE=/dev/null pip install --isolated --root="$pkgdir" --ignore-installed --no-deps artifacts/tensorflow_addons-*.whl
+}