diff options
-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 |