wehub-resource-sync
75 lines
2.9 KiB
Bash
Executable File
75 lines
2.9 KiB
Bash
Executable File
#!/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.
|
|
# ==============================================================================
|
|
if [[ "$TFCI_DOCKER_PULL_ENABLE" == 1 ]]; then
|
|
# Simple retry logic for docker-pull errors. Sleeps if a pull fails.
|
|
# Pulling an already-pulled container image will finish instantly, so
|
|
# repeating the command costs nothing.
|
|
docker pull "$TFCI_DOCKER_IMAGE" || sleep 15
|
|
docker pull "$TFCI_DOCKER_IMAGE" || sleep 30
|
|
docker pull "$TFCI_DOCKER_IMAGE" || sleep 60
|
|
docker pull "$TFCI_DOCKER_IMAGE"
|
|
fi
|
|
|
|
if [[ "$TFCI_DOCKER_REBUILD_ENABLE" == 1 ]]; then
|
|
DOCKER_BUILDKIT=1 docker build --cache-from "$TFCI_DOCKER_IMAGE" -t "$TFCI_DOCKER_IMAGE" $TFCI_DOCKER_REBUILD_ARGS
|
|
if [[ "$TFCI_DOCKER_REBUILD_UPLOAD_ENABLE" == 1 ]]; then
|
|
docker push "$TFCI_DOCKER_IMAGE"
|
|
fi
|
|
fi
|
|
|
|
# Keep the existing "tf" container if it's already present.
|
|
# The container is not cleaned up automatically! Remove it with:
|
|
# docker rm tf
|
|
if ! docker container inspect tf >/dev/null 2>&1 ; then
|
|
# Pass all existing TFCI_ variables into the Docker container
|
|
env_file=$(mktemp)
|
|
env | grep ^TFCI_ > "$env_file"
|
|
|
|
if [[ $(uname -s) == MSYS_NT* ]]; then
|
|
is_windows=true
|
|
else
|
|
is_windows=false
|
|
fi
|
|
|
|
WORKING_DIR="$TFCI_GIT_DIR"
|
|
if [[ "$is_windows" == true ]]; then
|
|
env_file=$(cygpath -m $env_file)
|
|
WORKING_DIR=$(replace_drive_letter_with_prefix "$TFCI_GIT_DIR" "$TFCI_OUTPUT_WIN_DOCKER_DIR")
|
|
echo "GCE_METADATA_HOST=$IP_ADDR" >> $env_file
|
|
fi
|
|
|
|
docker run $TFCI_DOCKER_ARGS --name tf -w "$WORKING_DIR" -itd --rm \
|
|
-v "$TFCI_GIT_DIR:$WORKING_DIR" \
|
|
--env-file "$env_file" \
|
|
"$TFCI_DOCKER_IMAGE" \
|
|
bash
|
|
|
|
if [[ "$is_windows" == true ]]; then
|
|
# Allow requests from the container.
|
|
# Additional setup is contained in ci/official/envs/rbe.
|
|
CONTAINER_IP_ADDR=$(docker inspect -f '{{range .NetworkSettings.Networks}}{{.IPAddress}}{{end}}' tf)
|
|
netsh advfirewall firewall add rule name="Allow Metadata Proxy" dir=in action=allow protocol=TCP localport=80 remoteip="$CONTAINER_IP_ADDR"
|
|
|
|
# Stop non-essential indexing and link tracking services that
|
|
# may lock new files or symlinks.
|
|
# They may be causing sporadic "Permission denied" errors during Bazel builds.
|
|
# b/461500885
|
|
docker exec tf powershell -NoProfile -Command 'Stop-Service -Name SysMain,DiagTrack -Force -ErrorAction SilentlyContinue'
|
|
fi
|
|
|
|
fi
|
|
tfrun() { docker exec tf "$@"; }
|