Internal change

PiperOrigin-RevId: 191769724

diff --git a/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh b/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh
index 438c5d5..5e9ae49 100644 (file)
--- a/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh
+++ b/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh
@@ -42,6 +42,14 @@ source "tensorflow/tools/ci_build/windows/bazel/common_env.sh" \
 source "tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh" \
   || { echo "Failed to source bazel_test_lib.sh" >&2; exit 1; }
 
+skip_test=0
+
+for ARG in "$@"; do
+  if [[ "$ARG" == --skip_test ]]; then
+    skip_test=1
+  fi
+done
+
 run_configure_for_cpu_build
 
 # --define=override_eigen_strong_inline=true speeds up the compiling of conv_grad_ops_3d.cc and conv_ops_3d.cc
@@ -49,6 +57,10 @@ run_configure_for_cpu_build
 BUILD_OPTS="--define=override_eigen_strong_inline=true"
 bazel build -c opt $BUILD_OPTS tensorflow/tools/pip_package:build_pip_package || exit $?
 
+if [[ "$skip_test" == 1 ]]; then
+  exit 0
+fi
+
 # Create a python test directory to avoid package name conflict
 PY_TEST_DIR="py_test_dir"
 create_python_test_dir "${PY_TEST_DIR}"