[arch-commits] Commit in tensorflow/trunk (PKGBUILD)

[Sven-Hendrik Haase]

    Date: Tuesday, October 3, 2017 @ 10:21:30
  Author: svenstaro
Revision: 261370

Prepare for cuda9

Modified:
  tensorflow/trunk/PKGBUILD

----------+
 PKGBUILD |   30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

Modified: PKGBUILD
===================================================================
--- PKGBUILD	2017-10-03 10:20:46 UTC (rev 261369)
+++ PKGBUILD	2017-10-03 10:21:30 UTC (rev 261370)
@@ -4,23 +4,22 @@
 
 pkgbase=tensorflow
 pkgname=(tensorflow tensorflow-cuda python-tensorflow python-tensorflow-cuda)
-pkgver=1.3.0
-pkgrel=5
+# I went with git for just this release because I couldn't be bothered to find all
+# patches needed for bazel 0.6. It's more than the issue tracker would have you think!
+pkgver=1.4.0.git1
+pkgrel=1
 pkgdesc="Library for computation using data flow graphs for scalable machine learning"
 url="https://www.tensorflow.org/"
 license=('Apache2')
 arch=('x86_64')
-makedepends=('git' 'bazel' 'python-numpy' 'gcc5' 'cuda'
-             'cudnn6' 'python-pip' 'python-wheel' 'python-setuptools')
+makedepends=('git' 'bazel' 'python-numpy' 'gcc6' 'cuda'
+             'cudnn' 'python-pip' 'python-wheel' 'python-setuptools')
 optdepends=('tensorboard: Tensorflow visualization toolkit')
-source=("https://github.com/tensorflow/tensorflow/archive/v${pkgver}.tar.gz")
-sha512sums=('b74228a13537e4cbcf8781167480a0bccb2c1ecf7bf21a02e35a182e073395f919645205036fd2193d5a186cfa8a869ba32c04fa5bd8f82cbe0ad8e56926398b')
+source=("git://github.com/tensorflow/tensorflow.git#commit=64f0ebd")
+sha512sums=('SKIP')
 
 prepare() {
-  # Necessary because of https://github.com/tensorflow/tensorflow/issues/12979#event-1250326815
-  cd tensorflow-${pkgver}
-  sed -ri "/^\W+sha256 = \"[^\"]+\"\W+$/d" tensorflow/workspace.bzl
-  cd ..
+  cp -r tensorflow tensorflow-${pkgver}
 
   cp -r tensorflow-${pkgver} tensorflow-${pkgver}-cuda
   # These environment variables influence the behavior of the configure call below.
@@ -32,6 +31,7 @@
   export TF_NEED_GCP=0
   export TF_NEED_HDFS=0
   export TF_ENABLE_XLA=1
+  export TF_NEED_GDR=0
   export TF_NEED_VERBS=0
   export TF_NEED_OPENCL=0
   export TF_NEED_MPI=0
@@ -49,13 +49,13 @@
   cd ${srcdir}/tensorflow-${pkgver}-cuda
 
   export TF_NEED_CUDA=1
-  export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-5
-  # For next version instead of the gcc-5 stuff:
+  # For next version instead of the gcc-6 stuff:
+  export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-6
   export TF_CUDA_CLANG=0
   # export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang
   export CUDA_TOOLKIT_PATH=/opt/cuda
   export TF_CUDA_VERSION=$($CUDA_TOOLKIT_PATH/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p')
-  export CUDNN_INSTALL_PATH=/opt/cudnn6
+  export CUDNN_INSTALL_PATH=/opt/cuda
   export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' $CUDNN_INSTALL_PATH/include/cudnn.h)
   export TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.1
 
@@ -75,7 +75,7 @@
 }
 
 package_tensorflow-cuda() {
-  depends=(cuda cudnn6)
+  depends=(cuda cudnn)
   conflicts=(tensorflow)
   provides=(tensorflow)
 
@@ -105,7 +105,7 @@
 }
 
 package_python-tensorflow-cuda() {
-  depends=(python cuda cudnn6 python-pycuda python-protobuf)
+  depends=(python cuda cudnn python-pycuda python-protobuf)
   conflicts=(python-tensorflow)
   provides=(python-tensorflow)