experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a binary size check for CPU binaries as well.

Browse Source 
PiperOrigin-RevId: 306472255
Change-Id: Id1c760d984311a1037c71b167d1ccbb110bff9ce
This commit is contained in:
Amit Patankar 2020-04-14 10:54:46 -07:00 committed by TensorFlower Gardener
parent bb14228bf0
commit 9f693e35a2
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tensorflow/tools/ci_build/builds/nightly_release_smoke_test.sh
View File

@ -18,7 +18,8 @@
set -e set -e
set -x set -x
MAX_WHL_SIZE=550M CPU_MAX_WHL_SIZE=190M
GPU_MAX_WHL_SIZE=510M
function run_smoke_test() { function run_smoke_test() {
VENV_TMP_DIR=$(mktemp -d) VENV_TMP_DIR=$(mktemp -d)
@ -82,10 +83,22 @@ function test_tf_imports() {
} }
function test_tf_whl_size() { function test_tf_whl_size() {
if [[ $(find $WHL_NAME -type f -size +${MAX_WHL_SIZE}) ]]; then # We do not need a separate check for MacOS regular binaries.
echo "The whl size has exceeded 550MB. To keep within pypi's CDN # We check for the `_cpu` string in the whl file name.
distribution limit, we must not exceed that threshold." if [[ "$WHL_NAME" == *"_cpu"* ]]; then
return 1 # Check CPU whl size.
if [[ $(find $WHL_NAME -type f -size +${CPU_MAX_WHL_SIZE}) ]]; then
echo "The CPU whl size has exceeded ${CPU_MAX_WHL_SIZE}MB. To keep
within pypi's CDN distribution limit, we must not exceed that threshold."
return 1
fi
else
# Check GPU whl size.
if [[ $(find $WHL_NAME -type f -size +${GPU_MAX_WHL_SIZE}) ]]; then
echo "The GPU whl size has exceeded ${GPU_MAX_WHL_SIZE}MB. To keep
within pypi's CDN distribution limit, we must not exceed that threshold."
return 1
fi
fi fi
} }