[gentoo-commits] proj/sci:master commit in: sci-libs/tensorflow/

[Martin Mokrejs]

commit:     d4716dbcfa408f14649ecf58d015c167cb57ded0
Author:     Martin Mokrejs <mmokrejs <AT> fold <DOT> natur <DOT> cuni <DOT> cz>
AuthorDate: Mon Apr  9 16:55:42 2018 +0000
Commit:     Martin Mokrejs <mmokrejs <AT> fold <DOT> natur <DOT> cuni <DOT> cz>
CommitDate: Mon Apr  9 16:55:42 2018 +0000
URL:        https://gitweb.gentoo.org/proj/sci.git/commit/?id=d4716dbc

sci-libs/tensorflow: more tweaks to use bazel build system

It does not build for me and later runs in sandbox violation error.
Possibly the build system could respect USE=mkl too.
The autogenerated? BUILD file speaks of static libs being output
instead of shared libs. It also mentions how to call linker when
linking 3rd-party apps against these libs.

Package-Manager: Portage-2.3.27, Repoman-2.3.9

 sci-libs/tensorflow/tensorflow-1.7.0.ebuild | 40 +++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/sci-libs/tensorflow/tensorflow-1.7.0.ebuild 
b/sci-libs/tensorflow/tensorflow-1.7.0.ebuild
index 0dae0c1d0..d411ba0ac 100644
--- a/sci-libs/tensorflow/tensorflow-1.7.0.ebuild
+++ b/sci-libs/tensorflow/tensorflow-1.7.0.ebuild
@@ -5,7 +5,7 @@ EAPI=6
 
 PYTHON_COMPAT=( python2_7 python3_{3,4,5,6} )
 
-inherit distutils-r1 eutils
+inherit python-r1 distutils-r1 eutils
 
 DESCRIPTION="Library for numerical computation using data flow graphs"
 HOMEPAGE="https://www.tensorflow.org
@@ -27,6 +27,7 @@ DEPEND="dev-util/bazel
        dev-python/wheel
        dev-python/numpy
        dev-libs/protobuf-c
+       dev-python/absl-py
        cuda? ( >=dev-util/nvidia-cuda-toolkit-7.0[profiler] >=dev-libs/cudnn-3 
)
        mpi? ( virtual/mpi )"
 RDEPEND="${DEPEND}"
@@ -39,5 +40,40 @@ src_configure(){
        # https://www.tensorflow.org/install/install_linux#InstallingNativePip
        #
        # usage: configure.py [-h] [--workspace WORKSPACE]
-       ./configure || die
+               python_configure() {
+               export PYTHON_BIN_PATH=${PYTHON}
+               export PYTHON_LIB_PATH=${PYTHON_SITEDIR}
+               export TF_NEED_JEMALLOC=1
+               export TF_NEED_GCP=0
+               export TF_NEED_HDFS=0
+               export TF_NEED_S3=0
+               export TF_NEED_KAFKA=0
+               export TF_ENABLE_XLA=0
+               export TF_NEED_GDR=0
+               export TF_NEED_VERBS=0
+               export TF_NEED_OPENCL=0
+               export TF_NEED_CUDA=0
+               export TF_NEED_MPI=0
+               export TF_NEED_OPENCL_SYCL=0
+               export CC_OPT_FLAGS=${CFLAGS}
+               export JAVA_HOME=$(java-config -O)
+               # TODO: protect by a USE flag test --config=mkl
+               ./configure || die
+       }
+       python_foreach_impl python_configure
+}
+
+
+src_compile() {
+       python_compile() {
+               # huh, by default tensorflow links static libs? See BUILD file
+               # set framework_shared_object=true somehow
+               bazel build --config=opt 
/tensorflow/tools/pip_package:build_pip_package || die
+               bazel-bin/tensorflow/tools/pip_package/build_pip_package 
tensorflow_pkg || die
+               unzip -o -d tensorflow_pkg 
tensorflow_pkg/${P}-cp35-cp35m-linux_x86_64.whl || die
+               python_domodule tensorflow_pkg/${P}.data/purelib/tensorflow
+               bazel test || die
+               bazel shutdown || die
+       }
+       python_foreach_impl python_compile
 }