diff options
author | Anna R <annarev@google.com> | 2018-10-02 16:14:32 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-02 16:19:00 -0700 |
commit | 7c0c0abab5b07528bae982d69257ebf4a8c077cb (patch) | |
tree | 2c04c480a7760589147272f7953afbe71d6277ef /tensorflow/tools | |
parent | 6c487cddd3503ef72c015c5c283fff81328282e5 (diff) |
Internal change.
PiperOrigin-RevId: 215477724
Diffstat (limited to 'tensorflow/tools')
-rwxr-xr-x | tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh | 2 |
1 files changed, 1 insertions, 1 deletions
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 cd7206baf8..9c6390070c 100755 --- a/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh +++ b/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh @@ -29,7 +29,7 @@ TF_TESTS_PER_GPU=${TF_TESTS_PER_GPU:-8} # p100 has minimum 12G memory. Therefore, we should limit each test to 1.5G. # To leave some room in case we want to run more tests in parallel in the # future and to use a rounder number, we set it to 1G. -export TF_PER_DEVICE_MEMORY_LIMIT_MB=1024 +export TF_PER_DEVICE_MEMORY_LIMIT_MB=${TF_PER_DEVICE_MEMORY_LIMIT_MB:-1024} # ******************************************************************* # This section of the script is needed to |