aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-05 01:00:50 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-05 01:03:35 -0700
commitbf8d058ccaf30bc05bce5d4b13133d14aca42dfe (patch)
tree45d15a6fd50e541d4db5ddb603ce50e012780200
parentd660ab0c392562be89f02400e492bd54a7f9d6b0 (diff)
Windows: Refactor bazel_test_lib.sh and common_env.sh
- Removed workaround for https://github.com/bazelbuild/bazel/issues/2182 since it's fixed - Removed setting CUDA related environment variables. Assume they are already set. If not, configure.py will set default values for them. - Removed obsolete variables for cc_test targets. PiperOrigin-RevId: 199256482
-rw-r--r--tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh116
-rw-r--r--tensorflow/tools/ci_build/windows/bazel/common_env.sh5
2 files changed, 3 insertions, 118 deletions
diff --git a/tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh b/tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh
index 582188fc00..a3e07737a4 100644
--- a/tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh
+++ b/tensorflow/tools/ci_build/windows/bazel/bazel_test_lib.sh
@@ -14,130 +14,20 @@
# limitations under the License.
# ==============================================================================
#
-# C++ tests
-failing_cpu_cc_tests="\
- //tensorflow/core/kernels:control_flow_ops_test + \
- //tensorflow/core:example_example_parser_configuration_test + \
- //tensorflow/core:lib_core_status_test + \
- //tensorflow/core:lib_monitoring_collection_registry_test + \
- //tensorflow/core:lib_strings_numbers_test + \
- //tensorflow/core/platform/hadoop:hadoop_file_system_test + \
- //tensorflow/core:platform_file_system_test + \
- //tensorflow/core:platform_logging_test + \
- //tensorflow/core:util_sparse_sparse_tensor_test + \
- //tensorflow/cc:framework_gradient_checker_test + \
- //tensorflow/cc:framework_gradients_test + \
- //tensorflow/cc:gradients_array_grad_test + \
- //tensorflow/cc:gradients_math_grad_test + \
- //tensorflow/cc:gradients_nn_grad_test + \
- //tensorflow/cc/saved_model:loader_test \
-"
-
-broken_cpu_cc_tests="\
- //tensorflow/cc:framework_cc_ops_test + \
- //tensorflow/core/platform/cloud:time_util_test + \
- //tensorflow/core/platform/cloud:oauth_client_test + \
- //tensorflow/core/platform/cloud:http_request_test + \
- //tensorflow/core/platform/cloud:google_auth_provider_test + \
- //tensorflow/core/platform/cloud:gcs_file_system_test + \
- //tensorflow/core/kernels/cloud:bigquery_table_accessor_test + \
- //tensorflow/core/kernels/hexagon:graph_transferer_test + \
- //tensorflow/core/kernels:remote_fused_graph_execute_utils_test + \
- //tensorflow/core/kernels:requantize_op_test + \
- //tensorflow/core/kernels:requantization_range_op_test + \
- //tensorflow/core/kernels:quantized_reshape_op_test + \
- //tensorflow/core/kernels:quantized_pooling_ops_test + \
- //tensorflow/core/kernels:quantized_matmul_op_test + \
- //tensorflow/core/kernels:quantized_conv_ops_test + \
- //tensorflow/core/kernels:quantized_concat_op_test + \
- //tensorflow/core/kernels:quantized_bias_add_op_test + \
- //tensorflow/core/kernels:quantized_batch_norm_op_test + \
- //tensorflow/core/kernels:quantized_activation_ops_test + \
- //tensorflow/core/kernels:quantize_op_test + \
- //tensorflow/core/kernels:quantize_down_and_shrink_range_op_test + \
- //tensorflow/core/kernels:quantize_and_dequantize_op_test_gpu + \
- //tensorflow/core/kernels:quantize_and_dequantize_op_test + \
- //tensorflow/core/kernels:quantization_utils_test + \
- //tensorflow/core/kernels:debug_ops_test + \
- //tensorflow/core/distributed_runtime/rpc:rpc_rendezvous_mgr_test_gpu + \
- //tensorflow/core/distributed_runtime/rpc:rpc_rendezvous_mgr_test + \
- //tensorflow/core/distributed_runtime/rpc:grpc_tensor_coding_test + \
- //tensorflow/core/distributed_runtime/rpc:grpc_session_test_gpu + \
- //tensorflow/core/distributed_runtime/rpc:grpc_session_test + \
- //tensorflow/core/distributed_runtime/rpc:grpc_channel_test_gpu + \
- //tensorflow/core/distributed_runtime/rpc:grpc_channel_test + \
- //tensorflow/core/distributed_runtime:remote_device_test_gpu + \
- //tensorflow/core/distributed_runtime:remote_device_test + \
- //tensorflow/core/distributed_runtime:executor_test_gpu + \
- //tensorflow/core/distributed_runtime:executor_test + \
- //tensorflow/core/debug:debug_gateway_test + \
- //tensorflow/core/debug:debug_grpc_io_utils_test + \
- //tensorflow/core:util_reporter_test + \
- //tensorflow/core:util_memmapped_file_system_test + \
- //tensorflow/core:platform_subprocess_test + \
- //tensorflow/core:platform_profile_utils_cpu_utils_test + \
- //tensorflow/core:lib_jpeg_jpeg_mem_unittest + \
- //tensorflow/core/debug:debug_io_utils_test \
-"
-
-# lib_core_threadpool_test is timeout, but it passes when running alone
-extra_failing_gpu_cc_tests="\
- //tensorflow/core:lib_core_threadpool_test + \
- //tensorflow/core:cuda_libdevice_path_test + \
- //tensorflow/core:common_runtime_direct_session_test + \
- //tensorflow/core:common_runtime_direct_session_with_tracking_alloc_test + \
- //tensorflow/core:device_tracer_test + \
- //tensorflow/core:ops_math_grad_test \
-"
-
-exclude_cpu_cc_tests="${failing_cpu_cc_tests} + ${broken_cpu_cc_tests}"
-
-exclude_gpu_cc_tests="${extra_failing_gpu_cc_tests} + ${exclude_cpu_cc_tests}"
function run_configure_for_cpu_build {
- # Due to a bug in Bazel: https://github.com/bazelbuild/bazel/issues/2182
- # yes "" | ./configure doesn't work on Windows, so we set all the
- # environment variables in advance to avoid interact with the script.
- export TF_NEED_CUDA=0
- if [ -z "$TF_ENABLE_XLA" ]; then
- export TF_ENABLE_XLA=0
- fi
- if [ -z "$TF_NEED_MKL" ]; then
- export TF_NEED_MKL=0
- fi
- export TF_NEED_VERBS=0
- export TF_NEED_GCP=1
- export TF_NEED_HDFS=0
- export TF_NEED_OPENCL_SYCL=0
- echo "" | ./configure
+ yes "" | ./configure
}
function run_configure_for_gpu_build {
- # Due to a bug in Bazel: https://github.com/bazelbuild/bazel/issues/2182
- # yes "" | ./configure doesn't work on Windows, so we set all the
- # environment variables in advance to avoid interact with the script.
+ # Enable CUDA support
export TF_NEED_CUDA=1
- export TF_CUDA_VERSION=9.0
- export CUDA_TOOLKIT_PATH="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0"
- export TF_CUDNN_VERSION=7.0
- if [ -z "$CUDNN_INSTALL_PATH" ]; then
- export CUDNN_INSTALL_PATH="C:/tools/cuda"
- fi
- export TF_CUDA_COMPUTE_CAPABILITIES="3.7"
- if [ -z "$TF_ENABLE_XLA" ]; then
- export TF_ENABLE_XLA=0
- fi
- export TF_NEED_VERBS=0
- export TF_NEED_MKL=0
- export TF_NEED_GCP=0
- export TF_NEED_HDFS=0
- export TF_NEED_OPENCL_SYCL=0
# TODO(pcloudy): Remove this after TensorFlow uses its own CRSOOTOOL
# for GPU build on Windows
export USE_MSVC_WRAPPER=1
- echo "" | ./configure
+ yes "" | ./configure
}
function set_gcs_remote_cache_options {
diff --git a/tensorflow/tools/ci_build/windows/bazel/common_env.sh b/tensorflow/tools/ci_build/windows/bazel/common_env.sh
index 0e6c0227b7..eefa8ee2d5 100644
--- a/tensorflow/tools/ci_build/windows/bazel/common_env.sh
+++ b/tensorflow/tools/ci_build/windows/bazel/common_env.sh
@@ -49,8 +49,3 @@ export PATH="/c/Program Files/Git/cmd:$PATH"
# Make sure we have pip in PATH
export PATH="/c/${PYTHON_BASE_PATH}/Scripts:$PATH"
-
-# Add Cuda and Cudnn dll directories into PATH
-export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/bin:$PATH"
-export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/extras/CUPTI/libx64:$PATH"
-export PATH="/c/tools/cuda/bin:$PATH"