update nightly version

galeone · Jun 13, 2022 · 6d02cd3 · 6d02cd3
1 parent cf41fa4
commit 6d02cd3
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/.github/workflows/arm-cd.yml b/.github/workflows/arm-cd.yml
@@ -44,14 +44,20 @@ jobs:
       - name: Clean repository
         shell: bash
         run: find /home/ubuntu/actions-runner/_work/tensorflow/tensorflow/. -name . -o -prune -exec sudo rm -rf -- {} + || true
-      - name: Checkout repository
+      - name: Checkout repository for nightly (skipped for releases)
+        if: ${{ github.event_name == 'schedule' }}
+        uses: actions/checkout@v3
+        with:
+          ref: 'nightly'
+      - name: Checkout repository for releases (skipped for nightly)
+        if: ${{ github.event_name == 'push' }}
         uses: actions/checkout@v3
       - name: Build and test pip wheel
         shell: bash
         run: |
-          tf_project_name='tensorflow_cpu_aws' && ${{ github.event_name == 'schedule' }} && tf_project_name='tf_nightly_cpu_aws'
+          is_nightly=0 && tf_project_name='tensorflow_cpu_aws' && ${{ github.event_name == 'schedule' }} && is_nightly=1 && tf_project_name='tf_nightly_cpu_aws'
           echo "PyPI project name:" $tf_project_name
-          CI_DOCKER_BUILD_EXTRA_PARAMS="--build-arg py_major_minor_version=${{ matrix.pyver }} --build-arg tf_project_name=${tf_project_name}" \
+          CI_DOCKER_BUILD_EXTRA_PARAMS="--build-arg py_major_minor_version=${{ matrix.pyver }} --build-arg is_nightly=${is_nightly} --build-arg tf_project_name=${tf_project_name}" \
           ./tensorflow/tools/ci_build/ci_build.sh cpu.arm64 bash tensorflow/tools/ci_build/rel/ubuntu/cpu_arm64_pip.sh
       - name: Upload pip wheel to PyPI
         shell: bash

diff --git a/tensorflow/tools/ci_build/Dockerfile.cpu.arm64 b/tensorflow/tools/ci_build/Dockerfile.cpu.arm64
@@ -37,6 +37,8 @@ RUN export PYTHON_VERSION=$(python -c 'import platform; print(platform.python_ve
 
 RUN pip3 install packaging
 
+ARG is_nightly=0
 ARG tf_project_name='tf_ci_cpu' # PyPI project name passed by CD GitHub workflow
 
+ENV IS_NIGHTLY=${is_nightly}
 ENV TF_PROJECT_NAME=${tf_project_name}
diff --git a/tensorflow/tools/ci_build/rel/ubuntu/cpu_arm64_pip.sh b/tensorflow/tools/ci_build/rel/ubuntu/cpu_arm64_pip.sh
@@ -90,10 +90,13 @@ export TF_TEST_TARGETS="${DEFAULT_BAZEL_TARGETS} \
     -//tensorflow/python/training:server_lib_test"
 export TF_PIP_TESTS="test_pip_virtualenv_clean"
 export TF_TEST_FILTER_TAGS="-no_oss,-oss_serial,-no_oss_py${py_ver},-gpu,-tpu,-benchmark-test,-v1only,-no_aarch64,-requires-gpu"
-export IS_NIGHTLY=0 # This variable is not used, but is required to be set
 export TF_PIP_TEST_ROOT="pip_test"
 export TF_AUDITWHEEL_TARGET_PLAT="manylinux2014"
 
+if [ ${IS_NIGHTLY} == 1 ]; then
+  ./tensorflow/tools/ci_build/update_version.py --nightly
+fi
+
 source tensorflow/tools/ci_build/builds/pip_new.sh
 
 # remove duplicate wheel and copy wheel to mounted volume for local access