1 |
commit: cf9a3c5ff72cb0d2998a739cdfe83cc5213fa8a0 |
2 |
Author: Jason Zaman <perfinion <AT> gentoo <DOT> org> |
3 |
AuthorDate: Mon Dec 23 02:32:20 2019 +0000 |
4 |
Commit: Jason Zaman <perfinion <AT> gentoo <DOT> org> |
5 |
CommitDate: Mon Dec 23 02:32:52 2019 +0000 |
6 |
URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=cf9a3c5f |
7 |
|
8 |
sci-libs/tensorflow: bump 2.1.0_rc1 |
9 |
|
10 |
Bug: https://bugs.gentoo.org/701342 |
11 |
Package-Manager: Portage-2.3.79, Repoman-2.3.16 |
12 |
Signed-off-by: Jason Zaman <perfinion <AT> gentoo.org> |
13 |
|
14 |
sci-libs/tensorflow/Manifest | 2 + |
15 |
.../tensorflow-2.1.0-python3.8-pywrap_tensor.patch | 11 + |
16 |
sci-libs/tensorflow/tensorflow-2.1.0_rc1.ebuild | 372 +++++++++++++++++++++ |
17 |
3 files changed, 385 insertions(+) |
18 |
|
19 |
diff --git a/sci-libs/tensorflow/Manifest b/sci-libs/tensorflow/Manifest |
20 |
index 1fccc3eaf93..945fe743e24 100644 |
21 |
--- a/sci-libs/tensorflow/Manifest |
22 |
+++ b/sci-libs/tensorflow/Manifest |
23 |
@@ -13,6 +13,7 @@ DIST bazelbuild-rules_swift.0.12.1.tar.gz 111674 BLAKE2B 7c116d6098aa0725034dce4 |
24 |
DIST cub-1.8.0.zip 602396 BLAKE2B a5e302a52e04f85ae8e1836c59e799532a8b961a8d10c08fe1241c9045f88e31f1ebda6d26124b85c6df40968e1c7b3985a39961bf7614535aafcab2da1a576a SHA512 6167c5be94989f88e05a07508cf69cf2c6b22a5ac733804043daa46bd86a44a15a758b1ffb811bab28b82039d027516ed85b3d9f7677a740a7e00ec9c2f81aed |
25 |
DIST eigen-049af2f56331.tar.gz 2485848 BLAKE2B 9d8a91782c9d40b7a6df101d2f43df25b5a3622bc02a7b9576929afb1f2f3d7c8b80671d24f1e31670883003b60c4cc0f3c9dadff0430d2ee047eb5122805edd SHA512 ab37106e9dfebbb42d7919c16b301887607080345b7ed63c8e3b8568489cfdc2c3ca0117a77905162cb3ba8324753040c841d2c96521fa4b9486f0778f1962df |
26 |
DIST eigen-49177915a14a.tar.gz 2496808 BLAKE2B 4e89b72a2ca121e9d1951a7179434c9eb70104ad45e63ae9811e10e28e32d51386acd2fe6938970404dcada8a377d1b3c07a68a98a68a036a6e85b2cf9ff4b48 SHA512 eb036225b41ffd8cc61ef99d2d22030c0d1e0f87eefcea4dd2b1fdb16a5316e0f4f17527983a00a8a684067654bf1f87c82cfaed5764829778660485ebbfa4b5 |
27 |
+DIST eigen-4e696901f873a2347f76d931cf2f701e31e15d05.tar.gz 2580537 BLAKE2B 927897c80f3529da40a76ddde26113403926d730fa4539aa9e4121406e8d3852abcfdbc55d178723300e9daa26176f6170db06bb49a59cb190e8c7e6dfdea4dd SHA512 d5461e615044aa94b6ac9ab2815536f4bdf2b9ccfeb578cc7bc62554ba928b4192450f72f1350f3231ed44d2bb90e95218de9a5d0bcee9ec02867499a6dc5db0 |
28 |
DIST eigen-afc120bc03bd.tar.gz 2546157 BLAKE2B 72e051d77cd22c26a4203cf244e87b1700b4dce664d7ee5f80d0dd86eb5557c93e57c50664f1d79b13796a77c5ccdf72203ec66e9376d383b5a3b056326c7b61 SHA512 61f57a22d2a7947c14cfde87946dc41528df696884502a13000ee5c9b964672a5a8d8c10f505f28ed7a2d39deb820c6818d191dc3ce13215b8b9775b33da2ed8 |
29 |
DIST farmhash-816a4ae622e964763ca0862d9dbd19324a1eaf45.tar.gz 467122 BLAKE2B 8b9dd426f4b9f732df6c8c09d868d1b1cc006c395b1525c46ea91c75318b723358c8e64bb0d86a73aace2032eded93f0d80cc7c2c77fddd6a39e3402ab4f2cb7 SHA512 7f5110514a8cdc7ce48d33fd94ad67d96a24e42e11b203e4d0986f4190545f051b2a63f8e09f2ac38c3aa4a5c62d9a0e02cd8ce02045037381fb62a99f0dcd00 |
30 |
DIST gemmlowp-12fed0cd7cfcd9e169bf1925bc3a7a58725fdcc3.zip 935628 BLAKE2B 09bd1dd1e367b1e5c4fe9a9525449c09460f19168bc03c57ad563aa27ec74a18f08b620a3dab6da4fa1a1448cdc7ce8bc8a089becc033b53fff4cc07593315ce SHA512 bd42a364f718f3b077cd9c840d626bfc1fd92ace98ce43329d8fa54700c9d28cb47568b6567e9b081b340af7be621c44be1ce06d4b8dccc4fb5d82c35b12f951 |
31 |
@@ -29,3 +30,4 @@ DIST tensorflow-1.15.0-python-license.rst.txt 45132 BLAKE2B 8a8ace4b64fd9eda1932 |
32 |
DIST tensorflow-1.15.0.tar.gz 46183168 BLAKE2B 5ccd87d5b53d6c543728e9c7ef96f12149bd8729620838b07a28f752fd5c57ef88350c2b62e8c44ec3561469eda38d8a64cbfca5e1ac7850674e1dd2c4d605d6 SHA512 f8683a950be6a3fa234d42e0f46bd9e049280e0c1d203c5c00ad44cf4728d894f2c301939d6d71203f815130a1ead53127adcc7565009589ff5aafc878b4dba7 |
33 |
DIST tensorflow-2.0.0.tar.gz 46176997 BLAKE2B 761e91a774f61b50459240a83fe52f723a76c2cc3eaf7c371684849cb7365b94fb3743ab8de9adffe64bef0f70da2bb0b7e6426a3f402e436eff841d3edefe8d SHA512 c3a2cb9673d2ade8e83961f7d944165123d0e4d94e6ca4d8fdee4f02f536893b879bbbf78d3ad5e557467b58f97f7194f2d9cfc64d7d05540969be732070eecc |
34 |
DIST tensorflow-2.1.0_rc0.tar.gz 41625131 BLAKE2B bc286f2f8d76a38973706acf3d839a4f54843b2f37ba5485eb149beacd55c7ac13660592acbc2413f8e60242c4a34b90cea8f1d9708d3f086e7cae73cc338832 SHA512 1481af1ed43ff03f6a0eeaad9896e2ec021d763bc69bf03e33353d6188074fdfaa1b6c3bba1d4e2496207f468e6a555fe2c962ae15eddd30ef872a3d76e19911 |
35 |
+DIST tensorflow-2.1.0_rc1.tar.gz 41637604 BLAKE2B 05a3de065de55fdd1086f75c8a6b69f227c02f43198a45e9d3312eacb903cca9ebf03784fa83881f54986d039fc54fbd88e85dfaa241d9b7455501ab8cfc26cc SHA512 e5b07da84a5ec7cd5f329b81a59835a858a7d1d43398555f1d81d6746829f6132d4153fd2506f97985f4424bab0b21b7c7938c873dc5e5e4004756bc5a885c12 |
36 |
|
37 |
diff --git a/sci-libs/tensorflow/files/tensorflow-2.1.0-python3.8-pywrap_tensor.patch b/sci-libs/tensorflow/files/tensorflow-2.1.0-python3.8-pywrap_tensor.patch |
38 |
new file mode 100644 |
39 |
index 00000000000..bc53750b070 |
40 |
--- /dev/null |
41 |
+++ b/sci-libs/tensorflow/files/tensorflow-2.1.0-python3.8-pywrap_tensor.patch |
42 |
@@ -0,0 +1,11 @@ |
43 |
+--- a/tensorflow/python/eager/pywrap_tensor.cc.ORIG 2019-09-27 23:56:33.000000000 +0200 |
44 |
++++ b/tensorflow/python/eager/pywrap_tensor.cc 2019-11-27 16:39:00.996713385 +0100 |
45 |
+@@ -839,7 +839,7 @@ |
46 |
+ sizeof(EagerTensor), /* tp_basicsize */ |
47 |
+ 0, /* tp_itemsize */ |
48 |
+ (destructor)EagerTensor_dealloc, /* tp_dealloc */ |
49 |
+- nullptr, /* tp_print */ |
50 |
++ 0, /* tp_vectorcall_offset */ |
51 |
+ nullptr, /* tp_getattr */ |
52 |
+ nullptr, /* tp_setattr */ |
53 |
+ nullptr, /* tp_compare */ |
54 |
|
55 |
diff --git a/sci-libs/tensorflow/tensorflow-2.1.0_rc1.ebuild b/sci-libs/tensorflow/tensorflow-2.1.0_rc1.ebuild |
56 |
new file mode 100644 |
57 |
index 00000000000..5d17a640e00 |
58 |
--- /dev/null |
59 |
+++ b/sci-libs/tensorflow/tensorflow-2.1.0_rc1.ebuild |
60 |
@@ -0,0 +1,372 @@ |
61 |
+# Copyright 1999-2019 Gentoo Authors |
62 |
+# Distributed under the terms of the GNU General Public License v2 |
63 |
+ |
64 |
+EAPI=7 |
65 |
+ |
66 |
+DISTUTILS_OPTIONAL=1 |
67 |
+PYTHON_COMPAT=( python{3_5,3_6,3_7} ) |
68 |
+MY_PV=${PV/_rc/-rc} |
69 |
+MY_P=${PN}-${MY_PV} |
70 |
+ |
71 |
+inherit bazel check-reqs cuda distutils-r1 flag-o-matic toolchain-funcs |
72 |
+ |
73 |
+DESCRIPTION="Computation framework using data flow graphs for scalable machine learning" |
74 |
+HOMEPAGE="https://www.tensorflow.org/" |
75 |
+ |
76 |
+LICENSE="Apache-2.0" |
77 |
+SLOT="0" |
78 |
+KEYWORDS="~amd64" |
79 |
+IUSE="cuda mpi +python xla" |
80 |
+CPU_USE_FLAGS_X86="sse sse2 sse3 sse4_1 sse4_2 avx avx2 fma3 fma4" |
81 |
+for i in $CPU_USE_FLAGS_X86; do |
82 |
+ IUSE+=" cpu_flags_x86_$i" |
83 |
+done |
84 |
+ |
85 |
+# distfiles that bazel uses for the workspace, will be copied to basel-distdir |
86 |
+bazel_external_uris=" |
87 |
+ https://storage.googleapis.com/mirror.tensorflow.org/www.kurims.kyoto-u.ac.jp/~ooura/fft2d.tgz -> oourafft2d-20061228.tgz |
88 |
+ https://gitlab.com/libeigen/eigen/-/archive/4e696901f873a2347f76d931cf2f701e31e15d05/eigen-4e696901f873a2347f76d931cf2f701e31e15d05.tar.gz -> eigen-4e696901f873a2347f76d931cf2f701e31e15d05.tar.gz |
89 |
+ https://github.com/abseil/abseil-cpp/archive/43ef2148c0936ebf7cb4be6b19927a9d9d145b8f.tar.gz -> abseil-cpp-43ef2148c0936ebf7cb4be6b19927a9d9d145b8f.tar.gz |
90 |
+ https://github.com/bazelbuild/bazel-skylib/releases/download/0.8.0/bazel-skylib.0.8.0.tar.gz |
91 |
+ https://github.com/bazelbuild/bazel-skylib/releases/download/0.9.0/bazel_skylib-0.9.0.tar.gz |
92 |
+ https://github.com/bazelbuild/bazel-toolchains/archive/92dd8a7a518a2fb7ba992d47c8b38299fe0be825.tar.gz -> bazel-toolchains-92dd8a7a518a2fb7ba992d47c8b38299fe0be825.tar.gz |
93 |
+ https://github.com/bazelbuild/rules_cc/archive/0d5f3f2768c6ca2faca0079a997a97ce22997a0c.zip -> bazelbuild-rules_cc-0d5f3f2768c6ca2faca0079a997a97ce22997a0c.zip |
94 |
+ https://github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz -> bazelbuild-rules_closure-308b05b2419edb5c8ee0471b67a40403df940149.tar.gz |
95 |
+ https://github.com/bazelbuild/rules_docker/releases/download/v0.10.0/rules_docker-v0.10.0.tar.gz -> bazelbuild-rules_docker-v0.10.0.tar.gz |
96 |
+ https://github.com/bazelbuild/rules_swift/releases/download/0.12.1/rules_swift.0.12.1.tar.gz -> bazelbuild-rules_swift.0.12.1.tar.gz |
97 |
+ https://github.com/google/farmhash/archive/816a4ae622e964763ca0862d9dbd19324a1eaf45.tar.gz -> farmhash-816a4ae622e964763ca0862d9dbd19324a1eaf45.tar.gz |
98 |
+ https://github.com/google/gemmlowp/archive/12fed0cd7cfcd9e169bf1925bc3a7a58725fdcc3.zip -> gemmlowp-12fed0cd7cfcd9e169bf1925bc3a7a58725fdcc3.zip |
99 |
+ https://github.com/google/highwayhash/archive/fd3d9af80465e4383162e4a7c5e2f406e82dd968.tar.gz -> highwayhash-fd3d9af80465e4383162e4a7c5e2f406e82dd968.tar.gz |
100 |
+ https://github.com/google/re2/archive/506cfa4bffd060c06ec338ce50ea3468daa6c814.tar.gz -> re2-506cfa4bffd060c06ec338ce50ea3468daa6c814.tar.gz |
101 |
+ https://github.com/mborgerding/kissfft/archive/36dbc057604f00aacfc0288ddad57e3b21cfc1b8.tar.gz -> kissfft-36dbc057604f00aacfc0288ddad57e3b21cfc1b8.tar.gz |
102 |
+ https://github.com/pybind/pybind11/archive/v2.3.0.tar.gz -> pybind11-v2.3.0.tar.gz |
103 |
+ https://github.com/llvm/llvm-project/archive/ecc999101aadc8dc7d4af9fd88be10fe42674aa0.tar.gz -> llvm-ecc999101aadc8dc7d4af9fd88be10fe42674aa0.tar.gz |
104 |
+ cuda? ( |
105 |
+ https://github.com/nvidia/nccl/archive/0ceaec9cee96ae7658aa45686853286651f36384.tar.gz -> nvidia-nccl-0ceaec9cee96ae7658aa45686853286651f36384.tar.gz |
106 |
+ https://github.com/NVlabs/cub/archive/1.8.0.zip -> cub-1.8.0.zip |
107 |
+ ) |
108 |
+ python? ( |
109 |
+ https://github.com/intel/ARM_NEON_2_x86_SSE/archive/1200fe90bb174a6224a525ee60148671a786a71f.tar.gz -> ARM_NEON_2_x86_SSE-1200fe90bb174a6224a525ee60148671a786a71f.tar.gz |
110 |
+ https://storage.googleapis.com/mirror.tensorflow.org/docs.python.org/2.7/_sources/license.rst.txt -> tensorflow-1.15.0-python-license.rst.txt |
111 |
+ https://pypi.python.org/packages/bc/cc/3cdb0a02e7e96f6c70bd971bc8a90b8463fda83e264fa9c5c1c98ceabd81/backports.weakref-1.0rc1.tar.gz |
112 |
+ )" |
113 |
+ |
114 |
+SRC_URI="https://github.com/${PN}/${PN}/archive/v${MY_PV}.tar.gz -> ${P}.tar.gz |
115 |
+ ${bazel_external_uris}" |
116 |
+ |
117 |
+RDEPEND=" |
118 |
+ app-arch/snappy |
119 |
+ dev-db/lmdb |
120 |
+ dev-db/sqlite |
121 |
+ dev-libs/double-conversion |
122 |
+ dev-libs/icu |
123 |
+ ~dev-libs/jsoncpp-1.9.1 |
124 |
+ dev-libs/libpcre |
125 |
+ dev-libs/nsync |
126 |
+ dev-libs/openssl:0= |
127 |
+ >=dev-libs/protobuf-3.8.0:= |
128 |
+ >=dev-libs/re2-0.2019.06.01 |
129 |
+ media-libs/giflib |
130 |
+ media-libs/libjpeg-turbo |
131 |
+ media-libs/libpng:0 |
132 |
+ >=net-libs/grpc-1.22.0 |
133 |
+ net-misc/curl |
134 |
+ sys-libs/zlib |
135 |
+ >=sys-apps/hwloc-2 |
136 |
+ cuda? ( |
137 |
+ >=dev-util/nvidia-cuda-toolkit-9.1[profiler] |
138 |
+ dev-libs/cudnn |
139 |
+ ) |
140 |
+ mpi? ( virtual/mpi ) |
141 |
+ python? ( |
142 |
+ ${PYTHON_DEPS} |
143 |
+ >=dev-libs/flatbuffers-1.8.0 |
144 |
+ dev-python/absl-py[${PYTHON_USEDEP}] |
145 |
+ >=dev-python/astor-0.7.1[${PYTHON_USEDEP}] |
146 |
+ dev-python/gast[${PYTHON_USEDEP}] |
147 |
+ >=dev-python/numpy-1.16[${PYTHON_USEDEP}] |
148 |
+ dev-python/google-pasta[${PYTHON_USEDEP}] |
149 |
+ dev-python/opt-einsum[${PYTHON_USEDEP}] |
150 |
+ >=dev-python/protobuf-python-3.8.0[${PYTHON_USEDEP}] |
151 |
+ dev-python/six[${PYTHON_USEDEP}] |
152 |
+ dev-python/termcolor[${PYTHON_USEDEP}] |
153 |
+ >=dev-python/grpcio-1.22.0[${PYTHON_USEDEP}] |
154 |
+ >=dev-python/wrapt-1.11.1[${PYTHON_USEDEP}] |
155 |
+ >=net-libs/google-cloud-cpp-0.10.0 |
156 |
+ >=sci-libs/keras-applications-1.0.8[${PYTHON_USEDEP}] |
157 |
+ >=sci-libs/keras-preprocessing-1.1.0[${PYTHON_USEDEP}] |
158 |
+ >=sci-visualization/tensorboard-2.0.0[${PYTHON_USEDEP}] |
159 |
+ )" |
160 |
+DEPEND="${RDEPEND} |
161 |
+ python? ( |
162 |
+ dev-python/mock |
163 |
+ dev-python/setuptools |
164 |
+ )" |
165 |
+PDEPEND="python? ( |
166 |
+ >=sci-libs/tensorflow-estimator-2.0.0[${PYTHON_USEDEP}] |
167 |
+ )" |
168 |
+BDEPEND=" |
169 |
+ app-arch/unzip |
170 |
+ >=dev-libs/protobuf-3.8.0 |
171 |
+ dev-java/java-config |
172 |
+ dev-lang/swig |
173 |
+ || ( |
174 |
+ =dev-util/bazel-0.27* |
175 |
+ =dev-util/bazel-0.29* |
176 |
+ ) |
177 |
+ cuda? ( |
178 |
+ >=dev-util/nvidia-cuda-toolkit-9.1[profiler] |
179 |
+ ) |
180 |
+ !python? ( dev-lang/python ) |
181 |
+ python? ( |
182 |
+ dev-python/cython |
183 |
+ dev-python/mock |
184 |
+ >=dev-python/grpcio-tools-1.22.0 |
185 |
+ )" |
186 |
+REQUIRED_USE="python? ( ${PYTHON_REQUIRED_USE} )" |
187 |
+ |
188 |
+S="${WORKDIR}/${MY_P}" |
189 |
+ |
190 |
+PATCHES=( |
191 |
+ "${FILESDIR}/tensorflow-1.15.0_rc0-0001-WORKSPACE-add-rules-docker-http_archive-bazel-toolch.patch" |
192 |
+ "${FILESDIR}/tensorflow-2.1.0-external_libs.patch" |
193 |
+ "${FILESDIR}/tensorflow-2.1.0-python3.8-pywrap_tensor.patch" |
194 |
+) |
195 |
+DOCS=( AUTHORS CONTRIBUTING.md ISSUE_TEMPLATE.md README.md RELEASE.md ) |
196 |
+CHECKREQS_MEMORY="5G" |
197 |
+CHECKREQS_DISK_BUILD="10G" |
198 |
+ |
199 |
+get-cpu-flags() { |
200 |
+ local i f=() |
201 |
+ # Keep this list in sync with tensorflow/core/platform/cpu_feature_guard.cc. |
202 |
+ for i in sse sse2 sse3 sse4_1 sse4_2 avx avx2 fma4; do |
203 |
+ use cpu_flags_x86_${i} && f+=( -m${i/_/.} ) |
204 |
+ done |
205 |
+ use cpu_flags_x86_fma3 && f+=( -mfma ) |
206 |
+ echo "${f[*]}" |
207 |
+} |
208 |
+ |
209 |
+pkg_setup() { |
210 |
+ ewarn "TensorFlow 2.0 is a major release that contains some incompatibilities" |
211 |
+ ewarn "with TensorFlow 1.x. For more information about migrating to TF2.0 see:" |
212 |
+ ewarn "https://www.tensorflow.org/guide/migrate" |
213 |
+ |
214 |
+ local num_pythons_enabled |
215 |
+ num_pythons_enabled=0 |
216 |
+ count_impls(){ |
217 |
+ num_pythons_enabled=$((${num_pythons_enabled} + 1)) |
218 |
+ } |
219 |
+ use python && python_foreach_impl count_impls |
220 |
+ |
221 |
+ # 10G to build C/C++ libs, 5G per python impl |
222 |
+ CHECKREQS_DISK_BUILD="$((10 + 6 * ${num_pythons_enabled}))G" |
223 |
+ check-reqs_pkg_setup |
224 |
+} |
225 |
+ |
226 |
+src_unpack() { |
227 |
+ # Only unpack the main distfile |
228 |
+ unpack "${P}.tar.gz" |
229 |
+ bazel_load_distfiles "${bazel_external_uris}" |
230 |
+} |
231 |
+ |
232 |
+src_prepare() { |
233 |
+ export JAVA_HOME=$(java-config --jre-home) # so keepwork works |
234 |
+ |
235 |
+ append-flags $(get-cpu-flags) |
236 |
+ bazel_setup_bazelrc |
237 |
+ |
238 |
+ if ver_test "$(cuda_toolkit_version)" -ge "10.2"; then |
239 |
+ eapply "${FILESDIR}/tensorflow-2.1.0-cuda_10.2_support_bin2c.patch" |
240 |
+ fi |
241 |
+ |
242 |
+ default |
243 |
+ use python && python_copy_sources |
244 |
+ |
245 |
+ use cuda && cuda_add_sandbox |
246 |
+} |
247 |
+ |
248 |
+src_configure() { |
249 |
+ export JAVA_HOME=$(java-config --jre-home) # so keepwork works |
250 |
+ |
251 |
+ do_configure() { |
252 |
+ export CC_OPT_FLAGS=" " |
253 |
+ export TF_ENABLE_XLA=$(usex xla 1 0) |
254 |
+ export TF_NEED_OPENCL_SYCL=0 |
255 |
+ export TF_NEED_OPENCL=0 |
256 |
+ export TF_NEED_COMPUTECPP=0 |
257 |
+ export TF_NEED_ROCM=0 |
258 |
+ export TF_NEED_MPI=$(usex mpi 1 0) |
259 |
+ export TF_SET_ANDROID_WORKSPACE=0 |
260 |
+ |
261 |
+ if use python; then |
262 |
+ python_export PYTHON_SITEDIR |
263 |
+ export PYTHON_BIN_PATH="${PYTHON}" |
264 |
+ export PYTHON_LIB_PATH="${PYTHON_SITEDIR}" |
265 |
+ else |
266 |
+ export PYTHON_BIN_PATH="$(which python)" |
267 |
+ export PYTHON_LIB_PATH="$(python -c 'from distutils.sysconfig import *; print(get_python_lib())')" |
268 |
+ fi |
269 |
+ |
270 |
+ export TF_NEED_CUDA=$(usex cuda 1 0) |
271 |
+ export TF_DOWNLOAD_CLANG=0 |
272 |
+ export TF_CUDA_CLANG=0 |
273 |
+ export TF_NEED_TENSORRT=0 |
274 |
+ if use cuda; then |
275 |
+ export TF_CUDA_PATHS="${EPREFIX%/}/opt/cuda" |
276 |
+ export GCC_HOST_COMPILER_PATH="$(cuda_gccdir)/$(tc-getCC)" |
277 |
+ export TF_CUDA_VERSION="$(cuda_toolkit_version)" |
278 |
+ export TF_CUDNN_VERSION="$(cuda_cudnn_version)" |
279 |
+ einfo "Setting CUDA version: $TF_CUDA_VERSION" |
280 |
+ einfo "Setting CUDNN version: $TF_CUDNN_VERSION" |
281 |
+ |
282 |
+ if [[ -z "$TF_CUDA_COMPUTE_CAPABILITIES" ]]; then |
283 |
+ ewarn "WARNING: Tensorflow is being built with its default CUDA compute capabilities: 3.5 and 7.0." |
284 |
+ ewarn "These may not be optimal for your GPU." |
285 |
+ ewarn "" |
286 |
+ ewarn "To configure Tensorflow with the CUDA compute capability that is optimal for your GPU," |
287 |
+ ewarn "set TF_CUDA_COMPUTE_CAPABILITIES in your make.conf, and re-emerge tensorflow." |
288 |
+ ewarn "For example, to use CUDA capability 7.5 & 3.5, add: TF_CUDA_COMPUTE_CAPABILITIES=7.5,3.5" |
289 |
+ ewarn "" |
290 |
+ ewarn "You can look up your GPU's CUDA compute capability at https://developer.nvidia.com/cuda-gpus" |
291 |
+ ewarn "or by running /opt/cuda/extras/demo_suite/deviceQuery | grep 'CUDA Capability'" |
292 |
+ fi |
293 |
+ fi |
294 |
+ |
295 |
+ # com_googlesource_code_re2 weird branch using absl, doesnt work with released re2 |
296 |
+ local SYSLIBS=( |
297 |
+ absl_py |
298 |
+ astor_archive |
299 |
+ boringssl |
300 |
+ com_github_googleapis_googleapis |
301 |
+ com_github_googlecloudplatform_google_cloud_cpp |
302 |
+ com_google_protobuf |
303 |
+ curl |
304 |
+ cython |
305 |
+ double_conversion |
306 |
+ enum34_archive |
307 |
+ flatbuffers |
308 |
+ functools32_archive |
309 |
+ gast_archive |
310 |
+ gif |
311 |
+ grpc |
312 |
+ hwloc |
313 |
+ icu |
314 |
+ libjpeg_turbo |
315 |
+ jsoncpp_git |
316 |
+ keras_applications_archive |
317 |
+ lmdb |
318 |
+ nasm |
319 |
+ nsync |
320 |
+ opt_einsum_archive |
321 |
+ org_sqlite |
322 |
+ pasta |
323 |
+ pcre |
324 |
+ png |
325 |
+ six_archive |
326 |
+ snappy |
327 |
+ swig |
328 |
+ termcolor_archive |
329 |
+ wrapt |
330 |
+ zlib_archive |
331 |
+ ) |
332 |
+ |
333 |
+ export TF_SYSTEM_LIBS="${SYSLIBS[@]}" |
334 |
+ export TF_IGNORE_MAX_BAZEL_VERSION=1 |
335 |
+ |
336 |
+ # This is not autoconf |
337 |
+ ./configure || die |
338 |
+ |
339 |
+ echo 'build --config=noaws --config=nohdfs' >> .bazelrc || die |
340 |
+ echo 'build --define tensorflow_mkldnn_contraction_kernel=0' >> .bazelrc || die |
341 |
+ echo 'build --incompatible_no_support_tools_in_action_inputs=false' >> .bazelrc || die |
342 |
+ } |
343 |
+ if use python; then |
344 |
+ python_foreach_impl run_in_build_dir do_configure |
345 |
+ else |
346 |
+ do_configure |
347 |
+ fi |
348 |
+} |
349 |
+ |
350 |
+src_compile() { |
351 |
+ export JAVA_HOME=$(java-config --jre-home) # so keepwork works |
352 |
+ |
353 |
+ if use python; then |
354 |
+ python_setup |
355 |
+ BUILD_DIR="${S}-${EPYTHON/./_}" |
356 |
+ cd "${BUILD_DIR}" |
357 |
+ fi |
358 |
+ |
359 |
+ # fail early if any deps are missing |
360 |
+ ebazel build --nobuild \ |
361 |
+ //tensorflow:libtensorflow_framework.so \ |
362 |
+ //tensorflow:libtensorflow.so \ |
363 |
+ //tensorflow:libtensorflow_cc.so \ |
364 |
+ $(usex python '//tensorflow/tools/pip_package:build_pip_package' '') |
365 |
+ |
366 |
+ ebazel build \ |
367 |
+ //tensorflow:libtensorflow_framework.so \ |
368 |
+ //tensorflow:libtensorflow.so |
369 |
+ ebazel build //tensorflow:libtensorflow_cc.so |
370 |
+ |
371 |
+ do_compile() { |
372 |
+ ebazel build //tensorflow/tools/pip_package:build_pip_package |
373 |
+ } |
374 |
+ BUILD_DIR="${S}" |
375 |
+ cd "${BUILD_DIR}" |
376 |
+ use python && python_foreach_impl run_in_build_dir do_compile |
377 |
+ ebazel shutdown |
378 |
+} |
379 |
+ |
380 |
+src_install() { |
381 |
+ local i j |
382 |
+ export JAVA_HOME=$(java-config --jre-home) # so keepwork works |
383 |
+ |
384 |
+ do_install() { |
385 |
+ einfo "Installing ${EPYTHON} files" |
386 |
+ local srcdir="${T}/src-${MULTIBUILD_VARIANT}" |
387 |
+ mkdir -p "${srcdir}" || die |
388 |
+ bazel-bin/tensorflow/tools/pip_package/build_pip_package --src "${srcdir}" || die |
389 |
+ cd "${srcdir}" || die |
390 |
+ esetup.py install |
391 |
+ |
392 |
+ # libtensorflow_framework.so is in /usr/lib already |
393 |
+ python_export PYTHON_SITEDIR PYTHON_SCRIPTDIR |
394 |
+ rm -f "${D}/${PYTHON_SITEDIR}"/${PN}/lib${PN}_framework.so* || die |
395 |
+ rm -f "${D}/${PYTHON_SITEDIR}"/${PN}_core/lib${PN}_framework.so* || die |
396 |
+ python_optimize |
397 |
+ } |
398 |
+ |
399 |
+ if use python; then |
400 |
+ python_foreach_impl run_in_build_dir do_install |
401 |
+ |
402 |
+ # Symlink to python-exec scripts |
403 |
+ for i in "${ED}"/usr/lib/python-exec/*/*; do |
404 |
+ n="${i##*/}" |
405 |
+ [[ -e "${ED}/usr/bin/${n}" ]] || dosym ../lib/python-exec/python-exec2 "/usr/bin/${n}" |
406 |
+ done |
407 |
+ |
408 |
+ python_setup |
409 |
+ local BUILD_DIR="${S}-${EPYTHON/./_}" |
410 |
+ cd "${BUILD_DIR}" || die |
411 |
+ fi |
412 |
+ |
413 |
+ einfo "Installing headers" |
414 |
+ ebazel build //tensorflow:install_headers |
415 |
+ ebazel shutdown |
416 |
+ insinto /usr/include/${PN}/ |
417 |
+ doins -r bazel-genfiles/tensorflow/include/* |
418 |
+ |
419 |
+ einfo "Installing libs" |
420 |
+ # Generate pkg-config file |
421 |
+ ${PN}/c/generate-pc.sh --prefix="${EPREFIX}"/usr --libdir=$(get_libdir) --version=${MY_PV} || die |
422 |
+ insinto /usr/$(get_libdir)/pkgconfig |
423 |
+ doins ${PN}.pc ${PN}_cc.pc |
424 |
+ |
425 |
+ for l in libtensorflow{,_framework,_cc}.so; do |
426 |
+ dolib.so bazel-bin/tensorflow/${l} |
427 |
+ dolib.so bazel-bin/tensorflow/${l}.$(ver_cut 1) |
428 |
+ dolib.so bazel-bin/tensorflow/${l}.$(ver_cut 1-3) |
429 |
+ done |
430 |
+ |
431 |
+ einstalldocs |
432 |
+} |