From 619c5785813d87e785423a9767218e767e1fd516 Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Mon, 20 Jan 2020 19:57:06 -0800 Subject: [PATCH 1/2] Disable the gpu on cpu tests as they were added for 2.1 --- .../release/ubuntu_16/gpu_pip_on_cpu/build.sh | 38 +------------------ .../release/windows/gpu_pip_on_cpu/build.bat | 6 +-- 2 files changed, 2 insertions(+), 42 deletions(-) diff --git a/tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh b/tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh index 8161bed36e206a..ff7cd318a295f8 100755 --- a/tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh +++ b/tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh @@ -16,40 +16,4 @@ set -e set -x -source tensorflow/tools/ci_build/release/common.sh -update_bazel_linux - -install_ubuntu_16_pip_deps pip3.6 - -# Run configure. -export TF_NEED_GCP=1 -export TF_NEED_HDFS=1 -export TF_NEED_S3=1 -export TF_NEED_CUDA=1 -export TF_CUDA_VERSION=10 -export TF_CUDNN_VERSION=7 -export TF_NEED_TENSORRT=1 -export TENSORRT_INSTALL_PATH=/usr/local/tensorrt -export CC_OPT_FLAGS='-mavx' -export PYTHON_BIN_PATH=$(which python3.6) -export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib" -export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0 - -yes "" | "$PYTHON_BIN_PATH" configure.py - -######################## -## Build GPU pip package -######################## -bazel build --config=opt \ - --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \ - tensorflow/tools/pip_package:build_pip_package - -PIP_WHL_DIR=whl -mkdir -p ${PIP_WHL_DIR} -PIP_WHL_DIR=$(readlink -f ${PIP_WHL_DIR}) # Get absolute path -bazel-bin/tensorflow/tools/pip_package/build_pip_package "${PIP_WHL_DIR}" -WHL_PATH=$(ls "${PIP_WHL_DIR}"/*.whl) - -cp "${WHL_PATH}" "$(pwd)"/. -chmod +x tensorflow/tools/ci_build/builds/docker_cpu_pip.sh -docker run -e "BAZEL_VERSION=${BAZEL_VERSION}" -e "CI_BUILD_USER=$(id -u -n)" -e "CI_BUILD_UID=$(id -u)" -e "CI_BUILD_GROUP=$(id -g -n)" -e "CI_BUILD_GID=$(id -g)" -e "CI_BUILD_HOME=/bazel_pip" -v "$(pwd)":/bazel_pip tensorflow/tensorflow:devel-py3 "./bazel_pip/tensorflow/tools/ci_build/builds/with_the_same_user" "./bazel_pip/tensorflow/tools/ci_build/builds/docker_cpu_pip.sh" +echo "This build not enabled for 2.0" diff --git a/tensorflow/tools/ci_build/release/windows/gpu_pip_on_cpu/build.bat b/tensorflow/tools/ci_build/release/windows/gpu_pip_on_cpu/build.bat index 479bb15bff8b9d..34c5ac7923c242 100644 --- a/tensorflow/tools/ci_build/release/windows/gpu_pip_on_cpu/build.bat +++ b/tensorflow/tools/ci_build/release/windows/gpu_pip_on_cpu/build.bat @@ -13,8 +13,4 @@ :: limitations under the License. :: ============================================================================= -SET PYTHON_DIRECTORY=Python36 - -CALL tensorflow\tools\ci_build\release\common_win.bat - -echo "This build does not exist in TF 2.0" +echo "This build not enabled on 2.0" From 63aedd7d84370f31d549d8b508a52a7a79db49b8 Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Mon, 20 Jan 2020 20:09:07 -0800 Subject: [PATCH 2/2] Disable test that times out on mac non pip builds --- tensorflow/python/kernel_tests/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 3baa43e23c8bf2..2d00b05982e2bd 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -3275,6 +3275,7 @@ cuda_py_test( "//tensorflow/python:framework_for_generated_wrappers", ], tags = [ + "no_mac", # timeouts on GPU "no_pip", "notap", # http://b/31080670 ],