diff options
Diffstat (limited to 'tensorflow/tools/ci_build/ci_parameterized_build.sh')
-rwxr-xr-x | tensorflow/tools/ci_build/ci_parameterized_build.sh | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/tools/ci_build/ci_parameterized_build.sh b/tensorflow/tools/ci_build/ci_parameterized_build.sh index 351e9f0e96..50d61d8c31 100755 --- a/tensorflow/tools/ci_build/ci_parameterized_build.sh +++ b/tensorflow/tools/ci_build/ci_parameterized_build.sh @@ -18,7 +18,7 @@ # ci_parameterized_build.sh # # The script obeys the following required environment variables: -# TF_BUILD_CONTAINER_TYPE: (CPU | GPU | ANDROID) +# TF_BUILD_CONTAINER_TYPE: (CPU | GPU | ANDROID | ANDROID_NIGHTLY) # TF_BUILD_PYTHON_VERSION: (PYTHON2 | PYTHON3 | PYTHON3.5) # TF_BUILD_IS_PIP: (NO_PIP | PIP | BOTH) # @@ -127,6 +127,7 @@ PIP_CMD="${CI_BUILD_DIR}/builds/pip.sh" PIP_TEST_TUTORIALS_FLAG="--test_tutorials" PIP_INTEGRATION_TESTS_FLAG="--integration_tests" ANDROID_CMD="${CI_BUILD_DIR}/builds/android.sh" +ANDROID_NIGHTLY_CMD="${CI_BUILD_DIR}/builds/android_nightly.sh" TF_GPU_COUNT=${TF_GPU_COUNT:-8} PARALLEL_GPU_TEST_CMD='//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute' @@ -236,7 +237,7 @@ elif [[ ${CTYPE} == "gpu" ]]; then echo "" fi fi -elif [[ ${CTYPE} == "android" ]]; then +elif [[ ${CTYPE} == "android" ]] || [[ ${CTYPE} == "android_nightly" ]]; then : else die "Unrecognized value in TF_BUILD_CONTAINER_TYPE: "\ @@ -343,6 +344,9 @@ if [[ ${TF_BUILD_IS_PIP} == "no_pip" ]] || elif [[ ${CTYPE} == "android" ]]; then # Run android specific script for android build. NO_PIP_MAIN_CMD="${ANDROID_CMD} ${OPT_FLAG} " + elif [[ ${CTYPE} == "android_nightly" ]]; then + # Run android specific script for android nightly build. + NO_PIP_MAIN_CMD="${ANDROID_NIGHTLY_CMD} ${OPT_FLAG} " fi fi |