From 77bed6590d239c3eee2f25e6f483094cf282ecf6 Mon Sep 17 00:00:00 2001 From: Yifei Feng Date: Thu, 23 Jan 2020 14:47:15 -0800 Subject: [PATCH] Only update default docker images for official TF release versions. PiperOrigin-RevId: 291249321 Change-Id: I7676cde9b79c22702b21c9d7bfd8ca8f64f721a7 --- .../release/ubuntu_16/custom_op/release.sh | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh b/tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh index 5d167caeb52..da4928b2354 100644 --- a/tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh +++ b/tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh @@ -21,8 +21,6 @@ if [[ -z "${TF_VERSION}" ]]; then exit fi -UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:custom-op-ubuntu16" -UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:custom-op-gpu-ubuntu16" VERSIONED_UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:${TF_VERSION}-custom-op-ubuntu16" VERSIONED_UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:${TF_VERSION}-custom-op-gpu-ubuntu16" @@ -37,11 +35,18 @@ docker login -u "${TF_DOCKER_USERNAME}" -p "${TF_DOCKER_PASSWORD}" docker push "${VERSIONED_UBUNTU16_CPU_IMAGE}" docker push "${VERSIONED_UBUNTU16_GPU_IMAGE}" -# Tag and push the default images -docker tag "${VERSIONED_UBUNTU16_CPU_IMAGE}" "${UBUNTU16_CPU_IMAGE}" -docker push "${UBUNTU16_CPU_IMAGE}" +# Tag and push the default images for official TF releases +if [[ ${TF_VERSION} == *"rc"* ]]; then + echo "Do not update default images as ${TF_VERSION} is a release candidate." +else + UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:custom-op-ubuntu16" + UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:custom-op-gpu-ubuntu16" -docker tag "${VERSIONED_UBUNTU16_GPU_IMAGE}" "${UBUNTU16_GPU_IMAGE}" -docker push "${UBUNTU16_GPU_IMAGE}" + docker tag "${VERSIONED_UBUNTU16_CPU_IMAGE}" "${UBUNTU16_CPU_IMAGE}" + docker push "${UBUNTU16_CPU_IMAGE}" + + docker tag "${VERSIONED_UBUNTU16_GPU_IMAGE}" "${UBUNTU16_GPU_IMAGE}" + docker push "${UBUNTU16_GPU_IMAGE}" +fi docker logout#!/bin/bash