Skip to content

Commit 21199fa

Browse files
terryheotflite-support-robot
authored andcommitted
Update ARM wheel platform to manylinux2014
PiperOrigin-RevId: 413023530
1 parent 612a65f commit 21199fa

File tree

2 files changed

+6
-11
lines changed

2 files changed

+6
-11
lines changed

tensorflow_lite_support/tools/pip_package/build_pip_package.sh

+4-9
Original file line numberDiff line numberDiff line change
@@ -211,15 +211,10 @@ function main() {
211211
PKG_NAME_FLAG="--project_name tflite_support_nightly"
212212
fi
213213

214-
# Set package name flags for ARM builds.
215-
case "${TENSORFLOW_TARGET}" in
216-
armhf)
217-
PKG_NAME_FLAG="${PKG_NAME_FLAG} --plat-name=linux-armv7l"
218-
;;
219-
aarch64)
220-
PKG_NAME_FLAG="${PKG_NAME_FLAG} --plat-name=linux-aarch64"
221-
;;
222-
esac
214+
# Set additional package name flags (for ARM builds).
215+
if [[ -n ${EXTRA_PKG_NAME_FLAG} ]]; then
216+
PKG_NAME_FLAG="${PKG_NAME_FLAG} ${EXTRA_PKG_NAME_FLAG}"
217+
fi
223218

224219
if [[ ${NIGHTLY_BUILD} == "1" ]]; then
225220
# we use a script to update versions to avoid any tool differences on different platforms.

tensorflow_lite_support/tools/pip_package/rpi/build_arm_pip_package.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,6 @@
1717
set -ex
1818

1919
bazel build -c opt --config=elinux_armhf tensorflow_lite_support/tools/pip_package:build_pip_package
20-
TENSORFLOW_TARGET=armhf ./bazel-bin/tensorflow_lite_support/tools/pip_package/build_pip_package --dst wheels --nightly_flag
20+
EXTRA_PKG_NAME_FLAG="--plat-name=manylinux2014-armv7l" ./bazel-bin/tensorflow_lite_support/tools/pip_package/build_pip_package --dst wheels --nightly_flag
2121
bazel build -c opt --config=elinux_aarch64 tensorflow_lite_support/tools/pip_package:build_pip_package
22-
TENSORFLOW_TARGET=aarch64 ./bazel-bin/tensorflow_lite_support/tools/pip_package/build_pip_package --dst wheels --nightly_flag
22+
EXTRA_PKG_NAME_FLAG="--plat-name=manylinux2014-aarch64" ./bazel-bin/tensorflow_lite_support/tools/pip_package/build_pip_package --dst wheels --nightly_flag

0 commit comments

Comments
 (0)