From 04d86e99547f21947ca7a88f32fa2a0449ba4a5e Mon Sep 17 00:00:00 2001 From: Anna R Date: Fri, 21 Sep 2018 00:50:20 -0700 Subject: Internal change. PiperOrigin-RevId: 213948394 --- tensorflow/tools/ci_build/builds/run_pip_tests.sh | 2 +- tensorflow/tools/ci_build/ci_parameterized_build.sh | 2 +- tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh | 4 ++-- tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) (limited to 'tensorflow') diff --git a/tensorflow/tools/ci_build/builds/run_pip_tests.sh b/tensorflow/tools/ci_build/builds/run_pip_tests.sh index 4b762bf258..17198a6560 100755 --- a/tensorflow/tools/ci_build/builds/run_pip_tests.sh +++ b/tensorflow/tools/ci_build/builds/run_pip_tests.sh @@ -64,7 +64,7 @@ while true; do fi done -TF_GPU_COUNT=${TF_GPU_COUNT:-8} +TF_GPU_COUNT=${TF_GPU_COUNT:-4} # PIP tests should have a "different" path. Different than the one we place # virtualenv, because we are deleting and recreating it here. diff --git a/tensorflow/tools/ci_build/ci_parameterized_build.sh b/tensorflow/tools/ci_build/ci_parameterized_build.sh index cc09784c1d..49a9048c03 100755 --- a/tensorflow/tools/ci_build/ci_parameterized_build.sh +++ b/tensorflow/tools/ci_build/ci_parameterized_build.sh @@ -147,7 +147,7 @@ PIP_INTEGRATION_TESTS_FLAG="--integration_tests" ANDROID_CMD="${CI_BUILD_DIR}/builds/android.sh" ANDROID_FULL_CMD="${CI_BUILD_DIR}/builds/android_full.sh" -TF_GPU_COUNT=${TF_GPU_COUNT:-8} +TF_GPU_COUNT=${TF_GPU_COUNT:-4} PARALLEL_GPU_TEST_CMD='//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute' BENCHMARK_CMD="${CI_BUILD_DIR}/builds/benchmark.sh" diff --git a/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh b/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh index 03a2a07fb1..cd7206baf8 100755 --- a/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh +++ b/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh @@ -21,8 +21,8 @@ # Required environment variables: # TF_GPU_COUNT = Number of GPUs available. -TF_GPU_COUNT=${TF_GPU_COUNT:-8} -TF_TESTS_PER_GPU=${TF_TESTS_PER_GPU:-4} +TF_GPU_COUNT=${TF_GPU_COUNT:-4} +TF_TESTS_PER_GPU=${TF_TESTS_PER_GPU:-8} # We want to allow running one of the following configs: # - 4 tests per GPU on k80 # - 8 tests per GPU on p100 diff --git a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh index 28d5565b98..34847e637a 100644 --- a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh +++ b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh @@ -122,7 +122,7 @@ fi PIP_NAME=$(ls ${PY_TEST_DIR}/tensorflow_gpu-*.whl) reinstall_tensorflow_pip ${PIP_NAME} -TF_GPU_COUNT=${TF_GPU_COUNT:-8} +TF_GPU_COUNT=${TF_GPU_COUNT:-4} # Define no_tensorflow_py_deps=true so that every py_test has no deps anymore, # which will result testing system installed tensorflow -- cgit v1.2.3