batch

Lightning-AI · Jan 8, 2025 · 7024590 · 7024590
1 parent 165d38a
commit 7024590
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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