Initial import

2 files changed, 86 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..29548d6f28f2
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,25 @@
+pkgbase = tensorflow-git
+	pkgdesc = Open source software library for numerical computation using data flow graphs.
+	pkgver = 0.5.0.r10.gb4629c0
+	pkgrel = 1
+	url = https://tensorflow.org/
+	arch = i686
+	arch = x86_64
+	license = Apache2
+	makedepends = git
+	makedepends = python2-pip
+	depends = python2-numpy
+	depends = swig
+	depends = python2-wheel
+	provides = tensorflow
+	conflicts = tensorflow
+	conflicts = python2-tensorflow
+	source = git+https://github.com/tensorflow/tensorflow
+	source = git+https://github.com/google/protobuf
+	source = git+https://github.com/bazelbuild/bazel
+	md5sums = SKIP
+	md5sums = SKIP
+	md5sums = SKIP
+
+pkgname = tensorflow-git
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..9d2d3e64945d
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,61 @@
+# Maintainer: Pieter Robyns <pieter.robyns@uhasselt.be>
+# Upstream URL: https://github.com/tensorflow/tensorflow
+
+pkgname=tensorflow-git
+pkgver=0.5.0.r10.gb4629c0
+pkgrel=1
+pkgdesc="Open source software library for numerical computation using data flow graphs."
+arch=('i686' 'x86_64')
+url="https://tensorflow.org/"
+license=('Apache2')
+provides=('tensorflow')
+conflicts=('tensorflow' 'python2-tensorflow')
+depends=('python2-numpy' 'swig' 'python2-wheel')
+makedepends=('git' 'python2-pip')
+source=("git+https://github.com/tensorflow/tensorflow"
+        "git+https://github.com/google/protobuf"
+        "git+https://github.com/bazelbuild/bazel")
+md5sums=('SKIP' 'SKIP' 'SKIP')
+
+pkgver() {
+  cd "$srcdir/tensorflow"
+  git describe --long --tags | sed 's/\([^-]*-g\)/r\1/;s/-/./g;s/^v//'
+}
+
+prepare() {
+  # Config submodules
+  cd "$srcdir/tensorflow"
+  git submodule init
+  git config submodule.google/protobuf.url $srcdir/protobuf
+  git submodule update
+
+  # Set up some things for building
+  export PYTHON=python2
+  mkdir -p "$srcdir/tmp"
+}
+
+build() {
+  # Build bazel
+  echo "Please note: currently, bazel version 0.1.0 is required to build this package. Building it temporarily..."
+  cd "$srcdir/bazel"
+  git checkout tags/0.1.0
+  ./compile.sh
+  export TMP_BAZEL_EXECUTABLE=`pwd`/output/bazel
+
+  # Build tensorflow
+  cd "$srcdir/tensorflow"
+
+  ./configure
+  ${TMP_BAZEL_EXECUTABLE} build --jobs 2 -c opt //tensorflow/tools/pip_package:build_pip_package
+
+  sed -i 's/python/python2/g' bazel-bin/tensorflow/tools/pip_package/build_pip_package
+  bazel-bin/tensorflow/tools/pip_package/build_pip_package $srcdir/tmp
+}
+
+package() {
+  cd "$srcdir/tensorflow"
+
+  TMP_PKG=`find $srcdir/tmp -name "tensor*.whl"`
+  pip2 install --root $pkgdir/ $TMP_PKG --no-dependencies
+  install -Dm644 LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"
+}