diff options
author | Darjan Krijan | 2020-10-15 23:15:23 +0200 |
---|---|---|
committer | GitHub | 2020-10-15 17:15:23 -0400 |
commit | fa97314d266ed527c4ca43ab2f4b9a2c7b7d317a (patch) | |
tree | 63db5407f79fdc8b96f2b111a5d7b798949004fb /PKGBUILD | |
parent | ad6bd8ba4d83cf71c450fa15b9c68559749f4739 (diff) | |
download | aur-fa97314d266ed527c4ca43ab2f4b9a2c7b7d317a.tar.gz |
Fix no_opt/opt checks (#16)
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 62 |
1 files changed, 33 insertions, 29 deletions
@@ -11,12 +11,12 @@ _build_no_opt=1 _build_opt=1 pkgname=() -[ -n "$_build_no_opt" ] && pkgname+=(tensorflow-rocm python-tensorflow-rocm) -[ -n "$_build_opt" ] && pkgname+=(tensorflow-opt-rocm python-tensorflow-opt-rocm) +[ "$_build_no_opt" -eq 1 ] && pkgname+=(tensorflow-rocm python-tensorflow-rocm) +[ "$_build_opt" -eq 1 ] && pkgname+=(tensorflow-opt-rocm python-tensorflow-opt-rocm) pkgver=2.3.1 _pkgver=2.3.1 -pkgrel=2 +pkgrel=3 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/" license=('APACHE') @@ -139,34 +139,38 @@ build() { export CC=gcc-9 export CXX=g++-9 - echo "Building with rocm and without non-x86-64 optimizations" - cd "${srcdir}"/tensorflow-${_pkgver}-rocm - export CC_OPT_FLAGS="-march=x86-64" - export TF_NEED_CUDA=0 - export TF_NEED_ROCM=1 - ./configure - [ -n "$_build_no_opt" ] && bazel \ - build --config=mkl -c opt \ - //tensorflow:libtensorflow.so \ - //tensorflow:libtensorflow_cc.so \ - //tensorflow:install_headers \ - //tensorflow/tools/pip_package:build_pip_package - bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu "${srcdir}"/tmprocm + if [ "$_build_no_opt" -eq 1 ]; then + echo "Building with rocm and without non-x86-64 optimizations" + cd "${srcdir}"/tensorflow-${_pkgver}-rocm + export CC_OPT_FLAGS="-march=x86-64" + export TF_NEED_CUDA=0 + export TF_NEED_ROCM=1 + ./configure + bazel \ + build --config=mkl -c opt \ + //tensorflow:libtensorflow.so \ + //tensorflow:libtensorflow_cc.so \ + //tensorflow:install_headers \ + //tensorflow/tools/pip_package:build_pip_package + bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu "${srcdir}"/tmprocm + fi - echo "Building with rocm and with non-x86-64 optimizations" - cd "${srcdir}"/tensorflow-${_pkgver}-opt-rocm - export CC_OPT_FLAGS="-march=haswell -O3" - export TF_NEED_CUDA=0 - export TF_NEED_ROCM=1 - ./configure - [ -n "$_build_opt" ] && bazel \ - build --config=mkl --config=avx2_linux -c opt \ - //tensorflow:libtensorflow.so \ - //tensorflow:libtensorflow_cc.so \ - //tensorflow:install_headers \ - //tensorflow/tools/pip_package:build_pip_package - bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu "${srcdir}"/tmpoptrocm + if [ "$_build_opt" -eq 1 ]; then + echo "Building with rocm and with non-x86-64 optimizations" + cd "${srcdir}"/tensorflow-${_pkgver}-opt-rocm + export CC_OPT_FLAGS="-march=haswell -O3" + export TF_NEED_CUDA=0 + export TF_NEED_ROCM=1 + ./configure + bazel \ + build --config=mkl --config=avx2_linux -c opt \ + //tensorflow:libtensorflow.so \ + //tensorflow:libtensorflow_cc.so \ + //tensorflow:install_headers \ + //tensorflow/tools/pip_package:build_pip_package + bazel-bin/tensorflow/tools/pip_package/build_pip_package --gpu "${srcdir}"/tmpoptrocm + fi } _package() { |