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

[Mark Wright]

commit:     38fe3e1488bd82e32c16cc8d73de52cdb43149cc
Author:     Mark Wright <gienah <AT> gentoo <DOT> org>
AuthorDate: Tue May  1 10:19:44 2018 +0000
Commit:     Mark Wright <gienah <AT> gentoo <DOT> org>
CommitDate: Tue May  1 10:19:44 2018 +0000
URL:        https://gitweb.gentoo.org/proj/sci.git/commit/?id=38fe3e14

sci-libs/tensorflow: Bump to 1.8.0, build C and C++ APIs

Package-Manager: Portage-2.3.31, Repoman-2.3.9

 sci-libs/tensorflow/tensorflow-1.8.0.ebuild | 171 ++++++++++++++++++++++++++++
 1 file changed, 171 insertions(+)

diff --git a/sci-libs/tensorflow/tensorflow-1.8.0.ebuild b/sci-libs/tensorflow/tensorflow-1.8.0.ebuild
new file mode 100644
index 000000000..34d309cf4
--- /dev/null
+++ b/sci-libs/tensorflow/tensorflow-1.8.0.ebuild
@@ -0,0 +1,171 @@
+# Copyright 1999-2018 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=6
+
+PYTHON_COMPAT=( python2_7 python3_{3,4,5,6} )
+
+inherit python-r1 distutils-r1 eutils versionator
+
+DESCRIPTION="Library for numerical computation using data flow graphs"
+HOMEPAGE="https://www.tensorflow.org
+	https://github.com/tensorflow/tensorflow"
+SRC_URI="https://github.com/${PN}/${PN}/archive/v${PV}.tar.gz -> ${P}.tar.gz
+	https://dev.gentoo.org/~gienah/snapshots/${P}-bazel-cache-repos.tar.xz"
+
+LICENSE="Apache-2.0"
+SLOT="0"
+KEYWORDS="~amd64"
+IUSE="cuda cxx mpi"
+
+# To create the cache repo tar file, temporarilly remove the
+# ${P}-bazel-cache-repos.tar.xz from SRC_URI and src_upack.  Then build
+# it so that bazel will download the files:
+# FEATURES="noclean -network-sandbox" emerge -av sci-libs/tensorflow
+# cd /var/tmp/portage/sci-libs/${P}
+# tar --owner=portage --group=portage -cJvf \
+# /usr/portage/distfiles/${P}-bazel-cache-repos.tar.xz \
+# homedir/.cache/bazel/_bazel_portage/cache/repos/v1
+
+# TensorFlow 1.7 may be the last time we support Cuda versions below 8.0.
+# Starting with TensorFlow 1.8 release, 8.0 will be the minimum supported
+# version.
+# TensorFlow 1.7 may be the last time we support cuDNN versions below 6.0.
+# Starting with TensorFlow 1.8 release, 6.0 will be the minimum supported
+# version.
+# Possibly missing deps:
+# 	dev-python/gast
+DEPEND="
+	cxx? ( dev-libs/protobuf )
+	dev-python/absl-py
+	dev-python/astor
+	dev-python/numpy
+	dev-python/protobuf-python
+	dev-python/six
+	dev-python/termcolor
+	dev-python/wheel
+	dev-libs/jemalloc
+	dev-libs/protobuf-c
+	dev-util/bazel
+	media-libs/giflib
+	virtual/jpeg:0
+	cuda? ( >=dev-util/nvidia-cuda-toolkit-8.0[profiler] >=dev-libs/cudnn-6 )
+	mpi? ( virtual/mpi )"
+	#opencl? ( virtual/opencl )"
+RDEPEND="${DEPEND}"
+
+src_unpack() {
+	unpack ${P}.tar.gz
+	pushd .. || die
+	unpack distdir/${P}-bazel-cache-repos.tar.xz
+	popd || die
+}
+
+# TODO: seems it also supports some MPI implementations
+src_configure(){
+	# there is no setup.py but there is configure
+	# https://www.tensorflow.org/install/install_sources
+	# https://www.tensorflow.org/install/install_linux#InstallingNativePip
+	#
+	# usage: configure.py [-h] [--workspace WORKSPACE]
+	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
+		if use cuda; then
+			export TF_NEED_CUDA=1
+		else
+			export TF_NEED_CUDA=0
+		fi
+		if use mpi; then
+			export TF_NEED_MPI=1
+		else
+			export TF_NEED_MPI=0
+		fi
+		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() {
+	# F: fopen_wr
+	# S: deny
+	# P: /proc/self/setgroups
+	# A: /proc/self/setgroups
+	# R: /proc/7712/setgroups
+	# C: unable to read /proc/1/cmdline
+	addpredict /proc
+
+	local opt=$(usex cuda "--config=cuda" "")
+	einfo ">>> Compiling ${PN} C"$(usex cxx " and C++" "")
+	bazel build \
+		  --config=opt ${opt} \
+		  //tensorflow:libtensorflow.so \
+		  //tensorflow:libtensorflow_framework.so \
+		  $(usex cxx "//tensorflow:libtensorflow_cc.so" "") || die
+
+	python_compile() {
+		einfo ">>> Compiling ${PN} ${MULTIBUILD_VARIANT}"
+		bazel build \
+			  --config=opt ${opt} \
+			  //tensorflow/tools/pip_package:build_pip_package || die
+		bazel-bin/tensorflow/tools/pip_package/build_pip_package tensorflow_pkg || die
+		unzip -o -d ${PN}_pkg_${MULTIBUILD_VARIANT} ${PN}_pkg/${P}-*.whl || die
+		rm -f ${PN}_pkg_${MULTIBUILD_VARIANT}/lib${PN}_framework.so || die
+	}
+	python_foreach_impl python_compile
+	bazel shutdown || die
+}
+
+src_test() {
+	python_foreach_impl python_test
+}
+
+src_install() {
+	local SO1=$(get_major_version)
+	local SOVER=$(version_format_string '$1.$2')
+	local tl="${PN} ${PN}_framework"
+	dodir /usr/include/${PN}/${PN}/c
+	insinto /usr/include/${PN}/${PN}/c
+	doins ${PN}/c/c_api.h
+	if use cxx; then
+		for i in $(find ${PN}/cc ${PN}/core third_party/eigen3/unsupported \
+						third_party/eigen3/Eigen \( -path third_party \
+						-prune -o -type f -name \*.h -print \) -o \
+						\( -path ${PN} -prune -o -type f -print \) ); do
+			dodir $(dirname /usr/include/${PN}/${i})
+			insinto $(dirname /usr/include/${PN}/${i})
+			doins ${i}
+		done
+		tl+=" ${PN}_cc"
+	fi
+	for i in ${tl}; do
+		dolib.so bazel-bin/${PN}/lib${i}.so
+		dosym "lib${i}.so" \
+			  "/usr/$(get_libdir)/lib${i}.so.${SO1}" \
+			|| die "Could not create /usr/$(get_libdir)/lib${i}.so.${SO1} symlink"
+		dosym "lib${i}.so" \
+			  "/usr/$(get_libdir)/lib${i}.so.${SOVER}" \
+			|| die "Could not create /usr/$(get_libdir)/lib${i}.so.${SOVER} symlink"
+	done
+	python_install() {
+		python_domodule ${PN}_pkg_${MULTIBUILD_VARIANT}/${P}.data/purelib/${PN}
+		dosym "../../../lib${PN}_framework.so" \
+			  "$(python_get_sitedir)/${PN}/lib${PN}_framework.so" \
+			|| die "Could not create $(python_get_sitedir)/lib${PN}_framework.so symlink for python module"
+	}
+	python_foreach_impl python_install
+	einstalldocs
+}