diff options
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 53 |
1 files changed, 44 insertions, 9 deletions
@@ -3,23 +3,36 @@ pkgbase=tensorflow-computecpp pkgname=(tensorflow-computecpp python-tensorflow-computecpp) pkgver=1.9 -pkgrel=3 +pkgrel=5 pkgdesc="Library for computation using data flow graphs for scalable machine learning (backend with ComputeCpp)" url="https://github.com/codeplaysoftware/tensorflow" epoch=1 license=('APACHE') arch=('x86_64') depends=(opencl-icd-loader computecpp) -makedepends=(git opencl-icd-loader computecpp bazel python-numpy python-pip python-wheel python-setuptools) +makedepends=(git opencl-icd-loader computecpp cmake opencl-headers + python-numpy python-pip python-wheel python-setuptools java-environment=8) options=(!ccache) -source=("git+${url}#tag=computecpp_1.0.1") -sha512sums=('SKIP') +source=("git+${url}" + #"https://github.com/bazelbuild/bazel/releases/download/0.17.2/bazel-0.17.2-dist.zip" + "https://archive.org/download/archlinux_pkg_bazel/bazel-0.17.2-1-x86_64.pkg.tar.xz" + python37.patch + py37.diff + gcc1.diff + gcc2.diff + gcc3.diff) +sha256sums=('SKIP' + '758e10caff4c1cb496d1cf49d6f4da2969b610b174276fb734b8502686d07ddd' + 'ef54b3783a05b5604cd8f448136567686806ad3a5759978f48549256807a8394' + 'b3997091bc7a32f9e8c062a88e9148273090ebf66aeebb5dc055baa41b7aae7e' + '7d9f32a46cac83ec1a7308ac380226cdf40f98830c869bcdf5feb7bf110abf9a' + '10de738141852cfebae9847b746ae9b58f3b3985561cccede929d8fbdba93551' + '742abe5d8bfd3f7ce33778a08cbb233337db56238d11ac2ad07171b0d6097bfb') prepare() { # These environment variables influence the behavior of the configure call below. export PYTHON_BIN_PATH=/usr/bin/python export USE_DEFAULT_PYTHON_LIB_PATH=1 - export CC_OPT_FLAGS="-march=native -mfpmath=sse -O2 -pipe" export TF_DOWNLOAD_CLANG=0 export TF_CUDA_CLANG=0 export TF_NEED_CUDA=0 @@ -39,18 +52,40 @@ prepare() { export TF_NEED_COMPUTECPP=1 export COMPUTECPP_TOOLKIT_PATH=/opt/ComputeCpp-CE export COMPUTE=:0 +# Device-sensitive magic, these should work on as much hardware as possible + export TF_USE_HALF_SYCL=0 + export TF_USE_DOUBLE_SYCL=0 + export TF_SYCL_BITCODE_TARGET=spir64 + export CC_OPT_FLAGS="-march=native -mfpmath=sse -O2 -pipe" # -sycl-compress-name for some amd cards # make sure the proxy variables are in all caps, otherwise bazel ignores them export HTTP_PROXY=`echo $http_proxy | sed -e 's/\/$//'` export HTTPS_PROXY=`echo $https_proxy | sed -e 's/\/$//'` + + cd ${srcdir}/tensorflow + git apply --index ../python37.patch + git apply --index --whitespace=nowarn ../py37.diff + git apply --index --whitespace=nowarn ../gcc1.diff + git apply --index --whitespace=nowarn ../gcc2.diff + git apply --index --whitespace=nowarn ../gcc3.diff } build() { - # _bazel_09_fix="--incompatible_load_argument_is_label=false" + # Build bazel + echo "Please note: currently, bazel version <0.18 is required to build this package." + echo "Fixing that for you" # "Building it temporarily..." + cd "$srcdir" + # ./compile.sh + export PATH=`pwd`/usr/bin:$PATH + cd ${srcdir}/tensorflow + if [ ! -f .bazelrc ]; then # configure should be in prepare, but bazel has to be built first atm + ./configure + fi - ./configure - bazel build --config=opt --config=sycl //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package # ${_bazel_09_fix} +# Please take notice this requires at least 7GB of swap/disk space and 0.8+(3.2*threads)GB of RAM to build + bazel build -c opt --config=sycl //tensorflow:libtensorflow.so \ + //tensorflow/tools/pip_package:build_pip_package # --jobs 1 --verbose_failures bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmp } @@ -77,7 +112,7 @@ package_python-tensorflow-computecpp() { cd ${srcdir}/tensorflow WHEEL_PACKAGE=$(find ${srcdir}/tmp -name "tensor*.whl") - pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies + pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies --no-warn-script-location # tensorboard has been separated from upstream but they still install it with # tensorflow. I don't know what kind of sense that makes but we have to clean |