From 7c0c0abab5b07528bae982d69257ebf4a8c077cb Mon Sep 17 00:00:00 2001 From: Anna R Date: Tue, 2 Oct 2018 16:14:32 -0700 Subject: Internal change. PiperOrigin-RevId: 215477724 --- tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/tools') 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 -- cgit v1.2.3