File tree Expand file tree Collapse file tree 2 files changed +6
-11
lines changed
tensorflow_lite_support/tools/pip_package Expand file tree Collapse file tree 2 files changed +6
-11
lines changed Original file line number Diff line number Diff 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.
Original file line number Diff line number Diff line change 1717set -ex
1818
1919bazel 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
2121bazel 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
You can’t perform that action at this time.
0 commit comments