wehub-resource-sync
53 lines
2.7 KiB
Plaintext
53 lines
2.7 KiB
Plaintext
# 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.
|
|
# ==============================================================================
|
|
# Enables RBE as long as RBE is available for the selected platform.
|
|
# The "selected" platform is considered to be
|
|
# TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX. "Available" means there is
|
|
# an entry in tensorflow/.bazelrc for build:rbe_TFCI_BAZEL_TARGET...
|
|
# ^^^^
|
|
# This env is only valid when RBE is available on the selected platform, since
|
|
# it enables a derived --config setting. If RBE is not available (i.e. there
|
|
# is no --config setting), bazel would fail and quit. This script does a quick
|
|
# check This script checks for such errors early
|
|
if ! grep "rbe_$TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX" .bazelrc; then
|
|
cat <<EOF
|
|
ERROR: RBE was enabled via the 'rbe' env in the 'TFCI' variable.
|
|
TFCI: $TFCI
|
|
TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX is "$TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX".
|
|
But in .bazelrc, there is no "rbe_$TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX".
|
|
That setting must exist for this script to function correctly.
|
|
EOF
|
|
exit 1
|
|
fi
|
|
|
|
TFCI_BAZEL_COMMON_ARGS="$TFCI_BAZEL_COMMON_ARGS --config rbe_$TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX"
|
|
|
|
if [[ `uname -s | grep -P '^MSYS_NT'` ]]; then
|
|
if [[ "$TFCI_GITHUB_ACTIONS" != "true" ]]; then
|
|
# Docker on Windows doesn't support the `host` networking mode, and so
|
|
# port-forwarding is required for the container to detect it's running on GCE.
|
|
export IP_ADDR=$(powershell -command "(Get-NetIPAddress -AddressFamily IPv4 -InterfaceAlias 'vEthernet (nat)').IPAddress")
|
|
netsh interface portproxy add v4tov4 listenaddress=$IP_ADDR listenport=80 connectaddress=169.254.169.254 connectport=80
|
|
# A local firewall rule for the container is added in
|
|
# ci/official/utilities/setup_docker.sh.
|
|
fi
|
|
else
|
|
# The volume mapping flag below shares the user's gcloud credentials, if any,
|
|
# with the container, in case the user has credentials stored there.
|
|
# This would allow Bazel to authenticate for RBE.
|
|
# Note: TF's CI does not have any credentials stored there.
|
|
TFCI_DOCKER_ARGS="$TFCI_DOCKER_ARGS -v $HOME/.config/gcloud:/root/.config/gcloud"
|
|
fi
|