MacOS libtensorflow fix.

PiperOrigin-RevId: 354374100
Change-Id: I2ae1464dd370e43d3d54d0844b8f02199e8f0f0c
This commit is contained in:
Amit Patankar 2021-01-28 12:52:51 -08:00 committed by TensorFlower Gardener
parent bd9288e665
commit 828d3611aa
2 changed files with 7 additions and 4 deletions

View File

@ -58,7 +58,9 @@ function build_libtensorflow_tarball() {
export TF_NEED_ROCM=0
export TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75,compute_80"
else
BAZEL_OPTS="${BAZEL_OPTS} --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010:toolchain"
if [ "$(uname)" != "Darwin" ]; then
BAZEL_OPTS="${BAZEL_OPTS} --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010:toolchain"
fi
fi
bazel clean --expunge
yes "" | ./configure

View File

@ -14,9 +14,11 @@
# limitations under the License.
# ==============================================================================
set -ex
if [[ "$IS_NIGHTLY" -eq 1 ]]; then
echo "chmod go+w lib_package/*" >> tensorflow/tools/ci_build/linux/libtensorflow.sh
echo "bazel clean --expunge" >> tensorflow/tools/ci_build/linux/libtensorflow.sh
echo "chmod go+w lib_package/*" >> tensorflow/tools/ci_build/osx/libtensorflow.sh
echo "bazel clean --expunge" >> tensorflow/tools/ci_build/osx/libtensorflow.sh
# Install latest bazel
source tensorflow/tools/ci_build/release/common.sh
@ -41,7 +43,6 @@ if [[ "$IS_NIGHTLY" -eq 1 ]]; then
gsutil cp macos_cpu_libtensorflow_binaries.tar.gz gs://libtensorflow-nightly/prod/tensorflow/release/macos/latest
else
set -ex
# Install latest bazel
source tensorflow/tools/ci_build/release/common.sh
install_bazelisk