diff options
author | Anna R <annarev@google.com> | 2018-09-21 00:50:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 00:54:51 -0700 |
commit | 04d86e99547f21947ca7a88f32fa2a0449ba4a5e (patch) | |
tree | 0e534d403451cc203b6f2cffe36b166502be5602 /tensorflow/tools/ci_build | |
parent | 347201aa1e866f5899d000f40251c4090bcf3c73 (diff) |
Internal change.
PiperOrigin-RevId: 213948394
Diffstat (limited to 'tensorflow/tools/ci_build')
4 files changed, 5 insertions, 5 deletions
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 |