From 7024590cf9b61a13680947dd8c55d983ae7bdf32 Mon Sep 17 00:00:00 2001 From: Jirka B Date: Thu, 9 Jan 2025 01:16:40 +0900 Subject: [PATCH] batch --- .azure/gpu-tests-fabric.yml | 1 + .azure/gpu-tests-pytorch.yml | 1 + tests/run_standalone_tests.sh | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml index 43880fc5ba4ff..0dede598b58b3 100644 --- a/.azure/gpu-tests-fabric.yml +++ b/.azure/gpu-tests-fabric.yml @@ -49,6 +49,7 @@ jobs: FREEZE_REQUIREMENTS: "1" PIP_CACHE_DIR: "/var/tmp/pip" PL_RUN_CUDA_TESTS: "1" + PL_STANDALONE_TESTS_BATCH_SIZE: "2" container: image: $(image) # default shm size is 64m. Increase it to avoid: diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml index 317913bc6f1b1..7e52df9216027 100644 --- a/.azure/gpu-tests-pytorch.yml +++ b/.azure/gpu-tests-pytorch.yml @@ -67,6 +67,7 @@ jobs: FREEZE_REQUIREMENTS: "1" PIP_CACHE_DIR: "/var/tmp/pip" PL_RUN_CUDA_TESTS: "1" + PL_STANDALONE_TESTS_BATCH_SIZE: "3" container: image: $(image) # default shm size is 64m. Increase it to avoid: diff --git a/tests/run_standalone_tests.sh b/tests/run_standalone_tests.sh index 75a52e16c57dc..136198ad4cf31 100755 --- a/tests/run_standalone_tests.sh +++ b/tests/run_standalone_tests.sh @@ -17,7 +17,7 @@ set -e # Batch size for testing: Determines how many standalone test invocations run in parallel # It can be set through the env variable PL_STANDALONE_TESTS_BATCH_SIZE and defaults to 6 if not set -test_batch_size="${PL_STANDALONE_TESTS_BATCH_SIZE:-3}" +test_batch_size="${PL_STANDALONE_TESTS_BATCH_SIZE:-5}" source="${PL_STANDALONE_TESTS_SOURCE:-"lightning"}" # this is the directory where the tests are located test_dir=$1 # parse the first argument