batch

Author: Borda

.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:

.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:

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