mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-06 12:20:11 +01:00
Remove the no longer used build.sh file to avoid confusion.
PiperOrigin-RevId: 764880850
This commit is contained in:
parent
8db8fdb6d8
commit
d60f47b8f4
|
|
@ -1,65 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
|
|
||||||
#
|
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
# you may not use this file except in compliance with the License.
|
|
||||||
# You may obtain a copy of the License at
|
|
||||||
#
|
|
||||||
# http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
#
|
|
||||||
# Unless required by applicable law or agreed to in writing, software
|
|
||||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
# See the License for the specific language governing permissions and
|
|
||||||
# limitations under the License.
|
|
||||||
# ==============================================================================
|
|
||||||
|
|
||||||
# Builds the following Docker images for Linux ARM64. See the accompanying
|
|
||||||
# Dockerfile for more details:
|
|
||||||
# - us-central1-docker.pkg.dev/tensorflow-sigs/ml-build-arm64:latest-multi-python
|
|
||||||
|
|
||||||
set -exo pipefail
|
|
||||||
|
|
||||||
function is_continuous_or_release() {
|
|
||||||
[[ "$KOKORO_JOB_TYPE" == "CONTINUOUS_INTEGRATION" ]] || [[ "$KOKORO_JOB_TYPE" == "RELEASE" ]]
|
|
||||||
}
|
|
||||||
|
|
||||||
# Move into the directory of the script
|
|
||||||
cd "$(dirname "$0")"
|
|
||||||
|
|
||||||
if is_continuous_or_release || [[ -z "$KOKORO_BUILD_ID" ]]; then
|
|
||||||
# A continuous job is the only one to publish to latest
|
|
||||||
TAG="latest"
|
|
||||||
else
|
|
||||||
TAG=$(head -n 1 "$KOKORO_PIPER_DIR/presubmit_request.txt" | grep -oP "^presubmit_changelist: +\K(\d+)")
|
|
||||||
fi
|
|
||||||
|
|
||||||
AR_IMAGE_PATH="us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build-arm64"
|
|
||||||
|
|
||||||
# Build for both JAX and TF usage. We do these in one place because they share
|
|
||||||
# almost all of the same cache layers
|
|
||||||
export DOCKER_BUILDKIT=1
|
|
||||||
AR_IMAGE="$AR_IMAGE_PATH:$TAG"
|
|
||||||
docker pull "$AR_IMAGE" || true
|
|
||||||
# Due to some flakiness of resources pulled in the build, allow the docker
|
|
||||||
# command to reattempt build a few times in the case of failure (b/302558736)
|
|
||||||
set +e
|
|
||||||
for i in $(seq 1 5)
|
|
||||||
do
|
|
||||||
docker build \
|
|
||||||
--build-arg REQUIREMENTS_FILE=jax.requirements.txt \
|
|
||||||
--cache-from "$AR_IMAGE" \
|
|
||||||
-t "$AR_IMAGE" . && break
|
|
||||||
done
|
|
||||||
final=$?
|
|
||||||
if [ $final -ne 0 ]; then
|
|
||||||
exit $final
|
|
||||||
fi
|
|
||||||
set -e
|
|
||||||
|
|
||||||
INFRA_PUBLIC_TAG=infrastructure-public-image-$(docker images "$AR_IMAGE" --quiet)
|
|
||||||
AR_IMAGE_INFRA_PUBLIC="$AR_IMAGE_PATH:$INFRA_PUBLIC_TAG"
|
|
||||||
docker image tag "$AR_IMAGE" "$AR_IMAGE_INFRA_PUBLIC"
|
|
||||||
|
|
||||||
gcloud auth configure-docker us-central1-docker.pkg.dev
|
|
||||||
docker push "$AR_IMAGE_PATH" --all-tags
|
|
||||||
Loading…
Reference in New Issue
Block a user