Gentoo Archives: gentoo-commits

From: Jason Zaman <perfinion@g.o>
To: gentoo-commits@l.g.o
Subject: [gentoo-commits] repo/gentoo:master commit in: sci-libs/tensorflow/
Date: Sat, 15 Sep 2018 16:02:39
Message-Id: 1537027065.0d7adf51d80763d2ca8c3e5ffac4dd9b3edbe388.perfinion@gentoo
1 commit: 0d7adf51d80763d2ca8c3e5ffac4dd9b3edbe388
2 Author: Jason Zaman <perfinion <AT> gentoo <DOT> org>
3 AuthorDate: Sat Sep 15 15:41:26 2018 +0000
4 Commit: Jason Zaman <perfinion <AT> gentoo <DOT> org>
5 CommitDate: Sat Sep 15 15:57:45 2018 +0000
6 URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=0d7adf51
7
8 sci-libs/tensorflow: bump to 1.11.0_rc0
9
10 Package-Manager: Portage-2.3.40, Repoman-2.3.9
11
12 sci-libs/tensorflow/Manifest | 3 +
13 sci-libs/tensorflow/tensorflow-1.11.0_rc0.ebuild | 431 +++++++++++++++++++++++
14 2 files changed, 434 insertions(+)
15
16 diff --git a/sci-libs/tensorflow/Manifest b/sci-libs/tensorflow/Manifest
17 index 9e53fefae2b..7e07d3ef146 100644
18 --- a/sci-libs/tensorflow/Manifest
19 +++ b/sci-libs/tensorflow/Manifest
20 @@ -1,6 +1,7 @@
21 DIST ARM_NEON_2_x86_SSE-0f77d9d182265259b135dad949230ecbf1a2633d.tar.gz 99753 BLAKE2B 79b0c9e2f3a8de91bb042e55cbc9f589970add4c9bdccd96c9a0f38887418c97aa67fc433b4feffb92a1384039b0d68a7cc6cda141371427a310172422bd41d6 SHA512 a21df040a5b3f67d9be006ae414113fa1ca86d38fa446acfab18afcbbb89ee7c48776fac5565d0694c99f1ff6ead3b10a9915eac7aae4d2a532ef5ac126a0072
22 DIST LMDB_0.9.19.tar.gz 143141 BLAKE2B 69f39e2194638989d8a17b79edd7918966d5608f4e611a7446b45e7321adf3b7d4bd69107fe00a1476d32cbd43b83f512c8498cefa3fca7eea8ba615821341fe SHA512 1d9825f09592ad92a540a1dec232cf6649f41cc67b0d59dc8958a71f4090f347c84995f32a166e2c2efecb862981f455814dd07af99bc3489e42fe3fd8bc6191
23 DIST abseil-cpp-9613678332c976568272c8f4a78631a29159271d.tar.gz 875263 BLAKE2B f11af5582ef3e00155361de2185b703fc9edde1b6cbefce62224da7b13e1beae219076f144b7b8d4bf63d1b21a73beb77fbb437e6d91e6d59a233846d91c1a99 SHA512 a3694d4c74f542f7a55872707548469e1bdc6dcd05d40ffd62de41d7440dc1fb98d260f690a6799ad39142e35226b04aacc6122aa1edf104fc9d396997995a8c
24 +DIST abseil-cpp-f0f15c2778b0e4959244dd25e63f445a455870f5.tar.gz 1025885 BLAKE2B aa8c79912d94b67a23f42e1f2de387e8493fdff32b79ea5e958fd5b44bc2a8c804170a251d8166c4f01fd883f20962a437822b2bdd5ec614161491487833bdfc SHA512 54b664528f754f6c76888599d3a7069a7284b5dc17a31442a52faa9d3fdca23cbe3e4c1628d79c95c483f4ce6058e4d874e27ea4f859b9cafb8bb192e5093670
25 DIST abseil-py-0.2.2.tar.gz 189357 BLAKE2B b53976b50f1738d671a32df2fabf197785cb4be1f68dbf7f0bdd5bbbb1a73439646f5b6a9ea42d5df40debdbb8e8c72c554f428a948f3117dba5d6a8e2b473f2 SHA512 d2a393e78acc9ac28fc9b1129c23de9c2ab6059a7527eaa29e4182356b16cbce1fede94af2a8b51af9d1c606b162d8ae84eb3036ced784a89d4f7547bfce73ed
26 DIST abseil-py-ea8c4d2ddbf3fba610c4d613260561699b776db8.tar.gz 189090 BLAKE2B ac4e7ffffb6dc2ad922d03cbfee1858335728fffe5721761fb0207b242ccf0f8d9739bed29987c8cc718f7cc3c297e05bf7506b36f31a06b9019be8215f842a7 SHA512 faf00229a062dcf71d9147bc76538dd4ee3adcb76b7e46943663eb43e3023587c68819d1ca157709c3f071e5796bbcf8e10277ccca1f5b83f33fb0025520d373
27 DIST abseil_cpp-720c017e30339fd1786ce4aac68bc8559736e53f.tar.gz 529637 BLAKE2B fea7eceda492764613ffe944331e6b08af33d600036e9b2453f4c09e892f22f79a95b00a29a9bdf9e705d11502027d88659e19aa484cf4a02158209b058cec13 SHA512 3ac70cb924518627dfc6ec2eb5f65bc0e1235898eefebb0d0410867270065aa745d21456f1212b916e732b3f5dada18b03e526e02d09a09f445531798219006e
28 @@ -60,10 +61,12 @@ DIST sqlite-amalgamation-3230100.zip 2183147 BLAKE2B af25542f81ac0592d143695466d
29 DIST swig-3.0.8.tar.gz 7937213 BLAKE2B 38f17631a36f178a478bf77f1a7ea77b50b91ed95ca0363f0f75d86bf7da6d855db314c1b7e67634b0e6e1438a0d141d0241595987142154ee356d60955248bc SHA512 85605bd98bf2b56f5bfca23ae23d76d764d76a174b05836c8686825e912d6326c370e9cf2134c0bf4f425560be103b16bf9c9d075077f52e713a69082616e906
30 DIST tensorflow-1.10.0.tar.gz 24490786 BLAKE2B 2ecec19b2b2ebb2e292ec07fee458017aa3fcc4c1445e6c9419ea9bd37990a58607c8f3134c785be0e3ae098c332a734fb85717b2ecd781551a0c8f20ffca8f2 SHA512 a25257ad7e561c2040ba9dda699acb24b757e6d79c60324cab20d8d8a428fef79a06a3627a95a59e2c8408a094666b9aa6d8af1c5419fa8c238e1f5e10e9ec33
31 DIST tensorflow-1.10.0_rc1.tar.gz 24486453 BLAKE2B 3b41653695cfa03ae5ca3a4e16d82a0f3248ce674acb76210747614e98fabb9131a298ad947185c462d2e3573dfb7d1580de934d1a0b40710bba16208df7e333 SHA512 889430262f686613f39982b9b2ed328db16b651848cbf1dd06fbf9a5ef2304ddf0b43ca21eef2bbcc86800051497f18ba166427788c74a65aac7d3c10ae60f52
32 +DIST tensorflow-1.11.0_rc0.tar.gz 25032294 BLAKE2B 7a99630445eb51db39c9a0b948dc04b63d48a55cf56e96f431a597695bfc9f36701b42ba9de3bcbb4134661f583fec2f2068a396a559b422806b973b149e62f0 SHA512 1f537188338d09fb63705d8d1eb9e6f6e89274ded5d16f165f63ff97d6e1f534d7e9dc412fb38e00d834f8ff5ec5f00a6fa5ab464ba0ab47b501916cb8828d7f
33 DIST tensorflow-1.8.0.tar.gz 22649439 BLAKE2B 7384c2cf742fb5a8b4e266e95080ae2513c1b112ab97f74839fa8e81bd91bd24645be8afb02e4447ad5fba9f47c4d146f59aa12085937cd3a364ec34c99590f3 SHA512 7280e65d26fb3f15d95f7217ee3bc08d1424cd144cf25bf638fa114fa835b2505dfaf457c55700067d24f485b77120973d094ec568e6d1b1054857402f9c352d
34 DIST tensorflow-1.9.0.tar.gz 23571524 BLAKE2B 2961ef5a35d3271512999478f6e260c90c48537e251b41d60ae9e033386c0dd9e1337af85c5441afca1b8cf5b3726349ba0e56cccdfa95fdf1c5e9e4d4db8af5 SHA512 eb25dafccf2ec600c4ca28612362c8317b8b9160610ca9fb60f265815651c155b19ae5be3f213a0d9e716d9e25637d6502164433955247cf6fa5d75408b7ac84
35 DIST tensorflow-patches-1.10.0.tar.bz2 4032 BLAKE2B 8588deec74179d8044af631de8e651491cc713553ee15ece9dff5052bd95b10924bdbe4301453f00d0f2fccd5d595d496b48fe45b0259837653b2de9e1898ab9 SHA512 1f57995ca9b3278735636cc42c88dc45558fa6a1d656108293f8df3fd9961d75f5e979bf0a9909136134e1cdaab41a34f83e605b48b23dddf17a6b1be684117b
36 DIST tensorflow-patches-1.10.0_rc1.tar.bz2 4032 BLAKE2B 8588deec74179d8044af631de8e651491cc713553ee15ece9dff5052bd95b10924bdbe4301453f00d0f2fccd5d595d496b48fe45b0259837653b2de9e1898ab9 SHA512 1f57995ca9b3278735636cc42c88dc45558fa6a1d656108293f8df3fd9961d75f5e979bf0a9909136134e1cdaab41a34f83e605b48b23dddf17a6b1be684117b
37 +DIST tensorflow-patches-1.11.0_rc0.tar.bz2 10103 BLAKE2B 66c9e8838f25be13851be3265af880a22355d0e8eb2cc297fff2118df1d6070bbbb216631e522f704a0d41d4029742f073a76acb01bbf150d827e69c9f6f5fc5 SHA512 05c3e7c1dfe50716b8043d7b1febf7a4f3e641c65b9c778a2fe0486ac0e9c091d50dba1c8e3c2b9417758ce8181688bb47ec0eb7804f8aa50726d52cfd202a8b
38 DIST tensorflow-patches-1.9.0.tar.bz2 8588 BLAKE2B 7fbdaa680b3bc372c760861709776cf333d587398297fb596332bb7394a7747c86cdc87615ae6f74e81d4e6b39a34aa3a9b6ee7cab95237316f20a315773b12b SHA512 35d56a1e71e500f41e423a2b58d58d210737afd2d9cd955aed227036a066506be1b39e7809f71d82c80f055ed43dfe1813213975acfb8d86be38baf24510e0f3
39 DIST tensorflow-python-license.txt 45132 BLAKE2B 1f572a06eeb4a58a5563b87a2db381a6e9eb0195cf1d006bda0d3da158e62a1e67e889a7d3d0da83d8609c0d048887bcbbd0d7056fd8e4f56b654047f36936fa SHA512 1b8c2f9733fcf27d560879418c366b5c3e44420d42adb3b857ee741793ddc75ad18324b016909457e5311f2143593392ce4404d12962f076e62f6036afbb521e
40 DIST termcolor-1.1.0.tar.gz 3912 BLAKE2B 8d661a708eff02894b6e390befddaef1c9d14f78f0c7d10ac332fa149f7892109f77bd279c9a488a3aa4070569c92f0ce847e3911d419ff549a10555b34534fb SHA512 4bd06bf4405a9ef6c005cd4d159ef602f7fc7fccb3e57586da1187c402f4d0b9051ef930cae423065c51ff4be8a22ceae556a61a6b3c8c519d623c066c340b53
41
42 diff --git a/sci-libs/tensorflow/tensorflow-1.11.0_rc0.ebuild b/sci-libs/tensorflow/tensorflow-1.11.0_rc0.ebuild
43 new file mode 100644
44 index 00000000000..16e8c2defc3
45 --- /dev/null
46 +++ b/sci-libs/tensorflow/tensorflow-1.11.0_rc0.ebuild
47 @@ -0,0 +1,431 @@
48 +# Copyright 1999-2018 Gentoo Foundation
49 +# Distributed under the terms of the GNU General Public License v2
50 +
51 +EAPI=6
52 +
53 +DISTUTILS_OPTIONAL=1
54 +PYTHON_COMPAT=( python2_7 python3_{5,6} )
55 +MY_PV=${PV/_rc/-rc}
56 +MY_P=${PN}-${MY_PV}
57 +
58 +inherit check-reqs cuda distutils-r1 eapi7-ver multiprocessing toolchain-funcs
59 +
60 +DESCRIPTION="Computation framework using data flow graphs for scalable machine learning"
61 +HOMEPAGE="https://www.tensorflow.org/"
62 +
63 +LICENSE="Apache-2.0"
64 +SLOT="0"
65 +KEYWORDS="~amd64"
66 +IUSE="cuda jemalloc mpi +python +system-libs"
67 +CPU_USE_FLAGS_X86="sse sse2 sse3 sse4_1 sse4_2 avx avx2 fma3 fma4"
68 +for i in $CPU_USE_FLAGS_X86; do
69 + IUSE+=" cpu_flags_x86_$i"
70 +done
71 +
72 +# distfiles that bazel uses for the workspace, will be copied to basel-distdir
73 +bazel_external_uris="
74 + http://www.kurims.kyoto-u.ac.jp/~ooura/fft.tgz -> oourafft-20061228.tgz
75 + https://bitbucket.org/eigen/eigen/get/fd6845384b86.tar.gz -> eigen-fd6845384b86.tar.gz
76 + https://github.com/abseil/abseil-cpp/archive/f0f15c2778b0e4959244dd25e63f445a455870f5.tar.gz -> abseil-cpp-f0f15c2778b0e4959244dd25e63f445a455870f5.tar.gz
77 + https://github.com/bazelbuild/rules_closure/archive/dbb96841cc0a5fb2664c37822803b06dab20c7d1.tar.gz -> bazelbuild-rules_closure-dbb96841cc0a5fb2664c37822803b06dab20c7d1.tar.gz
78 + https://github.com/google/double-conversion/archive/3992066a95b823efc8ccc1baf82a1cfc73f6e9b8.zip -> double-conversion-3992066a95b823efc8ccc1baf82a1cfc73f6e9b8.zip
79 + https://github.com/google/farmhash/archive/816a4ae622e964763ca0862d9dbd19324a1eaf45.tar.gz -> farmhash-816a4ae622e964763ca0862d9dbd19324a1eaf45.tar.gz
80 + https://github.com/google/gemmlowp/archive/38ebac7b059e84692f53e5938f97a9943c120d98.zip -> gemmlowp-38ebac7b059e84692f53e5938f97a9943c120d98.zip
81 + https://github.com/google/highwayhash/archive/fd3d9af80465e4383162e4a7c5e2f406e82dd968.tar.gz -> highwayhash-fd3d9af80465e4383162e4a7c5e2f406e82dd968.tar.gz
82 + cuda? (
83 + https://github.com/nvidia/nccl/archive/03d856977ecbaac87e598c0c4bafca96761b9ac7.tar.gz -> nvidia-nccl-03d856977ecbaac87e598c0c4bafca96761b9ac7.tar.gz
84 + https://github.com/NVlabs/cub/archive/1.8.0.zip -> cub-1.8.0.zip
85 + )
86 + python? (
87 + https://github.com/intel/ARM_NEON_2_x86_SSE/archive/0f77d9d182265259b135dad949230ecbf1a2633d.tar.gz -> ARM_NEON_2_x86_SSE-0f77d9d182265259b135dad949230ecbf1a2633d.tar.gz
88 + https://mirror.bazel.build/docs.python.org/2.7/_sources/license.txt -> tensorflow-python-license.txt
89 + https://pypi.python.org/packages/5c/78/ff794fcae2ce8aa6323e789d1f8b3b7765f601e7702726f430e814822b96/gast-0.2.0.tar.gz
90 + https://pypi.python.org/packages/bc/cc/3cdb0a02e7e96f6c70bd971bc8a90b8463fda83e264fa9c5c1c98ceabd81/backports.weakref-1.0rc1.tar.gz
91 + !system-libs? (
92 + https://github.com/abseil/abseil-py/archive/pypi-v0.2.2.tar.gz -> abseil-py-0.2.2.tar.gz
93 + https://github.com/googleapis/googleapis/archive/f81082ea1e2f85c43649bee26e0d9871d4b41cdb.zip -> googleapis-f81082ea1e2f85c43649bee26e0d9871d4b41cdb.zip
94 + https://github.com/GoogleCloudPlatform/google-cloud-cpp/archive/f875700a023bdd706333cde45aee8758b272c357.tar.gz -> google-cloud-cpp-f875700a023bdd706333cde45aee8758b272c357.tar.gz
95 + https://github.com/google/boringssl/archive/a0fb951d2a26a8ee746b52f3ba81ab011a0af778.tar.gz -> boringssl-a0fb951d2a26a8ee746b52f3ba81ab011a0af778.tar.gz
96 + )
97 + )
98 + !system-libs? (
99 + https://github.com/google/protobuf/archive/v3.6.0.tar.gz -> protobuf-3.6.0.tar.gz
100 + )"
101 +
102 +SRC_URI="https://github.com/${PN}/${PN}/archive/v${MY_PV}.tar.gz -> ${P}.tar.gz
103 + https://dev.gentoo.org/~perfinion/patches/tensorflow-patches-${PVR}.tar.bz2
104 + ${bazel_external_uris}"
105 +
106 +RDEPEND="
107 + app-arch/snappy
108 + dev-db/lmdb
109 + dev-db/sqlite
110 + >=dev-libs/jsoncpp-1.8.4
111 + dev-libs/libpcre
112 + dev-libs/nsync
113 + >=dev-libs/protobuf-3.6.0
114 + >=dev-libs/re2-0.2018.04.01
115 + media-libs/giflib
116 + media-libs/libjpeg-turbo
117 + media-libs/libpng:0
118 + >=net-libs/grpc-1.13.0
119 + net-misc/curl
120 + sys-libs/zlib
121 + cuda? (
122 + >=dev-util/nvidia-cuda-toolkit-8.0[profiler]
123 + >=dev-libs/cudnn-6.0
124 + )
125 + jemalloc? ( >=dev-libs/jemalloc-4.4.0 )
126 + mpi? ( virtual/mpi )
127 + python? (
128 + ${PYTHON_DEPS}
129 + >=dev-libs/flatbuffers-1.8.0
130 + dev-python/absl-py[${PYTHON_USEDEP}]
131 + dev-python/astor[${PYTHON_USEDEP}]
132 + dev-python/numpy[${PYTHON_USEDEP}]
133 + >=dev-python/protobuf-python-3.6.0[${PYTHON_USEDEP}]
134 + dev-python/six[${PYTHON_USEDEP}]
135 + dev-python/termcolor[${PYTHON_USEDEP}]
136 + >=net-libs/grpc-1.13.0[python,${PYTHON_USEDEP}]
137 + virtual/python-enum34[${PYTHON_USEDEP}]
138 + system-libs? (
139 + net-libs/google-cloud-cpp
140 + )
141 + )
142 + system-libs? (
143 + dev-libs/openssl:0
144 + )"
145 +DEPEND="${RDEPEND}
146 + !python? ( dev-lang/python )
147 + app-arch/unzip
148 + >=dev-util/bazel-0.16.0
149 + dev-java/java-config
150 + dev-python/mock
151 + dev-lang/nasm
152 + dev-lang/swig
153 + dev-python/cython"
154 +REQUIRED_USE="python? ( ${PYTHON_REQUIRED_USE} )"
155 +
156 +S="${WORKDIR}/${MY_P}"
157 +
158 +DOCS=( AUTHORS CONTRIBUTING.md ISSUE_TEMPLATE.md README.md RELEASE.md )
159 +CHECKREQS_MEMORY="5G"
160 +CHECKREQS_DISK_BUILD="5G"
161 +
162 +bazel-get-cpu-flags() {
163 + local i f=()
164 + # Keep this list in sync with tensorflow/core/platform/cpu_feature_guard.cc.
165 + for i in sse sse2 sse3 sse4_1 sse4_2 avx avx2 fma4; do
166 + use cpu_flags_x86_${i} && f+=( -m${i/_/.} )
167 + done
168 + use cpu_flags_x86_fma3 && f+=( -mfma )
169 + echo "${f[*]}"
170 +}
171 +
172 +bazel-get-flags() {
173 + local i fs=()
174 + for i in ${CFLAGS} $(bazel-get-cpu-flags); do
175 + fs+=( "--copt=${i}" "--host_copt=${i}" )
176 + done
177 + for i in ${CXXFLAGS} $(bazel-get-cpu-flags); do
178 + fs+=( "--cxxopt=${i}" "--host_cxxopt=${i}" )
179 + done
180 + for i in ${CPPFLAGS}; do
181 + fs+=( "--copt=${i}" "--host_copt=${i}" )
182 + fs+=( "--cxxopt=${i}" "--host_cxxopt=${i}" )
183 + done
184 + for i in ${LDFLAGS}; do
185 + fs+=( "--linkopt=${i}" "--host_linkopt=${i}" )
186 + done
187 + echo "${fs[*]}"
188 +}
189 +
190 +setup_bazelrc() {
191 + if [[ -f "${T}/bazelrc" ]]; then
192 + return
193 + fi
194 +
195 + # F: fopen_wr
196 + # P: /proc/self/setgroups
197 + # Even with standalone enabled, the Bazel sandbox binary is run for feature test:
198 + # https://github.com/bazelbuild/bazel/blob/7b091c1397a82258e26ab5336df6c8dae1d97384/src/main/java/com/google/devtools/build/lib/sandbox/LinuxSandboxedSpawnRunner.java#L61
199 + # https://github.com/bazelbuild/bazel/blob/76555482873ffcf1d32fb40106f89231b37f850a/src/main/tools/linux-sandbox-pid1.cc#L113
200 + addpredict /proc
201 +
202 + mkdir -p "${T}/bazel-cache" || die
203 + mkdir -p "${T}/bazel-distdir" || die
204 +
205 + cat > "${T}/bazelrc" <<-EOF || die
206 + startup --batch
207 +
208 + # dont strip HOME, portage sets a temp per-package dir
209 + build --action_env HOME
210 +
211 + # make bazel respect MAKEOPTS
212 + build --jobs=$(makeopts_jobs) $(bazel-get-flags)
213 + build --compilation_mode=opt --host_compilation_mode=opt
214 +
215 + # Use standalone strategy to deactivate the bazel sandbox, since it
216 + # conflicts with FEATURES=sandbox.
217 + build --spawn_strategy=standalone --genrule_strategy=standalone
218 + test --spawn_strategy=standalone --genrule_strategy=standalone
219 +
220 + build --strip=never
221 + build --verbose_failures --noshow_loading_progress
222 + test --verbose_test_summary --verbose_failures --noshow_loading_progress
223 +
224 + # make bazel only fetch distfiles from the cache
225 + fetch --repository_cache=${T}/bazel-cache/ --distdir=${T}/bazel-distdir/
226 + build --repository_cache=${T}/bazel-cache/ --distdir=${T}/bazel-distdir/
227 +
228 + build --define=PREFIX=${EPREFIX%/}/usr
229 + build --define=LIBDIR=\$(PREFIX)/$(get_libdir)
230 +
231 + EOF
232 +}
233 +
234 +ebazel() {
235 + # Use different build folders for each multibuild variant.
236 + local base_suffix="${MULTIBUILD_VARIANT+-}${MULTIBUILD_VARIANT}"
237 + local output_base="${WORKDIR}/bazel-base${base_suffix}"
238 + mkdir -p "${output_base}" || die
239 +
240 + einfo Running: bazel --output_base="${output_base}" "$@"
241 + bazel --output_base="${output_base}" $@ || die
242 +}
243 +
244 +load_distfiles() {
245 + # Populate the bazel distdir to fetch from since it cannot use the network
246 + # Bazel looks in distdir but will only look for the original filename, not
247 + # the possibly renamed one that portage downloaded. If the line has -> we
248 + # need to rename it back, otherwise a simple copy is fine.
249 +
250 + local src dst uri rename
251 +
252 + while read uri rename dst; do
253 + src="${uri##*/}"
254 + [[ -z $src ]] && continue
255 + if [[ "$rename" != "->" ]]; then
256 + dst="${src}"
257 + fi
258 +
259 + [[ ${A} =~ ${dst} ]] || continue
260 +
261 + if [[ "$dst" == "$src" ]]; then
262 + einfo "Copying $dst to bazel distdir ..."
263 + else
264 + einfo "Copying $dst to bazel distdir $src ..."
265 + fi
266 + ln -s "${DISTDIR}/${dst}" "${T}/bazel-distdir/${src}" || die
267 + done <<< "$(sed -re 's/!?[A-Za-z]+\?\s+\(\s*//g; s/\s+\)//g' <<< "${bazel_external_uris}")"
268 +}
269 +
270 +pkg_setup() {
271 + check-reqs_pkg_setup
272 +}
273 +
274 +src_unpack() {
275 + # Only unpack the main distfile
276 + unpack "${P}.tar.gz"
277 + unpack tensorflow-patches-${PVR}.tar.bz2
278 +}
279 +
280 +src_prepare() {
281 + export JAVA_HOME=$(java-config --jre-home) # so keepwork works
282 +
283 + setup_bazelrc
284 + load_distfiles
285 +
286 + eapply "${WORKDIR}"/patches/*.patch
287 +
288 + default
289 + use python && python_copy_sources
290 +
291 + if use cuda; then
292 + local i
293 + for i in /dev/nvidia*; do
294 + addpredict $i
295 + done
296 + fi
297 +}
298 +
299 +src_configure() {
300 + export JAVA_HOME=$(java-config --jre-home) # so keepwork works
301 +
302 + do_configure() {
303 + export CC_OPT_FLAGS=" "
304 + export TF_NEED_JEMALLOC=$(usex jemalloc 1 0)
305 + export TF_NEED_GCP=0
306 + export TF_NEED_HDFS=0
307 + export TF_NEED_S3=0
308 + export TF_NEED_AWS=0
309 + export TF_NEED_KAFKA=0
310 + export TF_ENABLE_XLA=0
311 + export TF_NEED_GDR=0
312 + export TF_NEED_VERBS=0
313 + export TF_NEED_NGRAPH=0
314 + export TF_NEED_OPENCL_SYCL=0
315 + export TF_NEED_OPENCL=0
316 + export TF_NEED_COMPUTECPP=0
317 + export TF_NEED_MKL=0
318 + export TF_NEED_MPI=$(usex mpi 1 0)
319 + export TF_SET_ANDROID_WORKSPACE=0
320 +
321 + if use python; then
322 + python_export PYTHON_SITEDIR
323 + export PYTHON_BIN_PATH="${PYTHON}"
324 + export PYTHON_LIB_PATH="${PYTHON_SITEDIR}"
325 + else
326 + export PYTHON_BIN_PATH="$(which python)"
327 + export PYTHON_LIB_PATH="$(python -c 'from distutils.sysconfig import *; print(get_python_lib())')"
328 + fi
329 +
330 + export TF_NEED_CUDA=$(usex cuda 1 0)
331 + export TF_DOWNLOAD_CLANG=0
332 + export TF_CUDA_CLANG=0
333 + export TF_NEED_TENSORRT=0
334 + if use cuda; then
335 + export CUDA_TOOLKIT_PATH="${EPREFIX%/}/opt/cuda"
336 + export CUDNN_INSTALL_PATH="${EPREFIX%/}/opt/cuda"
337 + export GCC_HOST_COMPILER_PATH="$(cuda_gccdir)/$(tc-getCC)"
338 + export TF_NCCL_VERSION="1"
339 +
340 + TF_CUDA_VERSION="$(best_version dev-util/nvidia-cuda-toolkit)"
341 + TF_CUDA_VERSION="${TF_CUDA_VERSION##*cuda-toolkit-}"
342 + export TF_CUDA_VERSION="$(ver_cut 1-2 ${TF_CUDA_VERSION})"
343 + einfo "Setting CUDA version: $TF_CUDA_VERSION"
344 +
345 + TF_CUDNN_VERSION="$(best_version dev-libs/cudnn)"
346 + TF_CUDNN_VERSION="${TF_CUDNN_VERSION##*cudnn-}"
347 + export TF_CUDNN_VERSION="$(ver_cut 1-2 ${TF_CUDNN_VERSION})"
348 + einfo "Setting CUDNN version: $TF_CUDNN_VERSION"
349 + fi
350 +
351 + local SYSLIBS=(
352 + astor_archive
353 + com_googlesource_code_re2
354 + curl
355 + cython
356 + flatbuffers
357 + gif_archive
358 + grpc
359 + jemalloc
360 + jpeg
361 + jsoncpp_git
362 + lmdb
363 + nasm
364 + nsync
365 + org_sqlite
366 + pcre
367 + png_archive
368 + six_archive
369 + snappy
370 + swig
371 + termcolor_archive
372 + zlib_archive
373 + )
374 + if use system-libs; then
375 + SYSLIBS+=(
376 + absl_py
377 + astor_archive
378 + boringssl
379 + com_github_googleapis_googleapis
380 + com_github_googlecloudplatform_google_cloud_cpp
381 + com_google_protobuf
382 + com_google_protobuf_cc
383 + protobuf_archive
384 + )
385 + fi
386 +
387 + SYSLIBS="${SYSLIBS[@]}"
388 + export TF_SYSTEM_LIBS="${SYSLIBS// /,}"
389 +
390 + # Only one bazelrc is read, import our one before configure sets its options
391 + echo "import ${T}/bazelrc" >> ./.bazelrc
392 +
393 + # This is not autoconf
394 + ./configure || die
395 + }
396 + if use python; then
397 + python_foreach_impl run_in_build_dir do_configure
398 + else
399 + do_configure
400 + fi
401 +}
402 +
403 +src_compile() {
404 + export JAVA_HOME=$(java-config --jre-home) # so keepwork works
405 +
406 + if use python; then
407 + python_setup
408 + local MULTIBUILD_VARIANT="${EPYTHON/./_}"
409 + cd "${S}-${MULTIBUILD_VARIANT}" || die
410 + fi
411 +
412 + # fail early if anything is missing
413 + ebazel build --nobuild \
414 + //tensorflow:libtensorflow_framework.so \
415 + //tensorflow:libtensorflow.so \
416 + //tensorflow:libtensorflow_cc.so \
417 + $(usex python '//tensorflow/tools/pip_package:build_pip_package' '')
418 +
419 + ebazel build \
420 + //tensorflow:libtensorflow_framework.so \
421 + //tensorflow:libtensorflow.so
422 + ebazel build //tensorflow:libtensorflow_cc.so
423 +
424 + do_compile() {
425 + ebazel build //tensorflow/tools/pip_package:build_pip_package
426 + }
427 + use python && python_foreach_impl run_in_build_dir do_compile
428 +}
429 +
430 +src_install() {
431 + local i j
432 + export JAVA_HOME=$(java-config --jre-home) # so keepwork works
433 +
434 + do_install() {
435 + einfo "Installing ${EPYTHON} files"
436 + local srcdir="${T}/src-${MULTIBUILD_VARIANT}"
437 + mkdir -p "${srcdir}" || die
438 + bazel-bin/tensorflow/tools/pip_package/build_pip_package --src "${srcdir}" || die
439 + cd "${srcdir}" || die
440 + esetup.py install
441 +
442 + # libtensorflow_framework.so is in /usr/lib already
443 + python_export PYTHON_SITEDIR PYTHON_SCRIPTDIR
444 + rm -f "${D}/${PYTHON_SITEDIR}/${PN}/lib${PN}_framework.so" "${D}/${PYTHON_SCRIPTDIR}/tensorboard" || die
445 + python_optimize
446 + }
447 +
448 + if use python; then
449 + python_foreach_impl run_in_build_dir do_install
450 +
451 + # Symlink to python-exec scripts
452 + for i in "${ED}"/usr/lib/python-exec/*/*; do
453 + n="${i##*/}"
454 + [[ -e "${ED}/usr/bin/${n}" ]] || dosym ../lib/python-exec/python-exec2 "/usr/bin/${n}"
455 + done
456 +
457 + python_setup
458 + local MULTIBUILD_VARIANT="${EPYTHON/./_}"
459 + cd "${S}-${MULTIBUILD_VARIANT}" || die
460 + fi
461 +
462 + einfo "Installing headers"
463 + ebazel build //tensorflow:install_headers
464 + insinto /usr/include/${PN}/
465 + doins -r bazel-genfiles/tensorflow/include/*
466 +
467 + einfo "Installing libs"
468 + # Generate pkg-config file
469 + ${PN}/c/generate-pc.sh --prefix="${EPREFIX}"/usr --libdir=$(get_libdir) --version=${MY_PV} || die
470 + insinto /usr/$(get_libdir)/pkgconfig
471 + doins ${PN}.pc
472 +
473 + dolib.so bazel-bin/tensorflow/lib${PN}_framework.so
474 + dolib.so bazel-bin/tensorflow/lib${PN}.so
475 + dolib.so bazel-bin/tensorflow/lib${PN}_cc.so
476 +
477 + einstalldocs
478 +}