blob: 58cb76fc8cfd7e384b547d410e50a2a83cbeb48e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
include tensorflow.inc
SRC_URI += " \
file://0001-disable-avxvnni-and-avx512fp16-for-x86.patch \
"
inherit native
do_configure:append () {
cat >> "${S}/bazelrc" <<-EOF
build --conlyopt=-Wno-stringop-overflow
build --conlyopt=-Wno-dangling-pointer
EOF
SED_COMMAND="s#%%WORKDIR%%#${WORKDIR}#g"
sed -i "${SED_COMMAND}" ${S}/configure.py
${TF_CONFIG} \
./configure
}
do_compile () {
unset CC
${BAZEL} build \
${TF_ARGS_EXTRA} \
-c opt \
--subcommands --explain=${T}/explain.log \
--verbose_explanations --verbose_failures \
--repo_env=TF_PYTHON_VERSION=3.12 \
--verbose_failures \
//tensorflow/tools/pip_package:wheel
}
do_install() {
echo "Installing pip package"
install -d ${D}/${PYTHON_SITEPACKAGES_DIR}
${STAGING_BINDIR_NATIVE}/pip3 install --disable-pip-version-check -v --no-deps \
-t ${D}/${PYTHON_SITEPACKAGES_DIR} --no-cache-dir \
${S}/bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow-${PV}*.whl
install -d ${D}${sbindir}
(
cd ${D}${PYTHON_SITEPACKAGES_DIR}/bin;
for app in `ls`; do
mv $app ${D}${sbindir}
done
)
}
|
