Commit 21199fa 1 parent 612a65f commit 21199fa Copy full SHA for 21199fa
File tree 2 files changed +6
-11
lines changed
tensorflow_lite_support/tools/pip_package
2 files changed +6
-11
lines changed Original file line number Diff line number Diff line change @@ -211,15 +211,10 @@ function main() {
211
211
PKG_NAME_FLAG=" --project_name tflite_support_nightly"
212
212
fi
213
213
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
223
218
224
219
if [[ ${NIGHTLY_BUILD} == " 1" ]]; then
225
220
# 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 17
17
set -ex
18
18
19
19
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
21
21
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
You can’t perform that action at this time.
0 commit comments