summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
authorAdriĆ  Arrufat2018-01-20 12:34:37 +0100
committerAdriĆ  Arrufat2018-01-20 12:34:37 +0100
commit1eeec69c574c42e60aca53b7951130803e914355 (patch)
treec3258ddf1286d31ecacfdfbc9c4ca36b8d02a911 /PKGBUILD
parentf6b6c92937b68ad8b880c814088b7392e3ca70da (diff)
downloadaur-1eeec69c574c42e60aca53b7951130803e914355.tar.gz
update to work with bazel 0.9.0
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD8
1 files changed, 5 insertions, 3 deletions
diff --git a/PKGBUILD b/PKGBUILD
index bff5b99541f..db4e98be89b 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,7 +3,7 @@
pkgbase=tensorflow-git
pkgname=(tensorflow-git tensorflow-cuda-git python-tensorflow-git python-tensorflow-cuda-git)
-pkgver=1.3.0+rc1+3547+gbedfe8ac14
+pkgver=1.5.0+rc1+1320+g6e6c7e9545
pkgrel=1
pkgdesc="Library for computation using data flow graphs for scalable machine learning"
url="https://tensorflow.org/"
@@ -36,6 +36,7 @@ prepare() {
export TF_NEED_OPENCL=0
export TF_NEED_MPI=0
export TF_NEED_S3=0
+ export TF_NEED_OPENCL_SYCL=0
# make sure the proxy variables are in all caps, otherwise bazel ignores them
export HTTP_PROXY=`echo $http_proxy | sed -e 's/\/$//'`
@@ -43,12 +44,13 @@ prepare() {
}
build() {
+ _bazel_09_fix="--incompatible_load_argument_is_label=false"
cd ${srcdir}/tensorflow
export TF_NEED_CUDA=0
./configure
- bazel build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package
+ bazel build --config=opt --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix}
bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmp
cd ${srcdir}/tensorflow-cuda
@@ -62,7 +64,7 @@ build() {
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,6.2
./configure
- bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package
+ bazel build --config=opt --config=cuda --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix}
bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcuda
}