Run CI unittests in parallel

.circleci/unittest/linux/scripts/environment.yml

@@ -6,6 +6,8 @@ channels:
 dependencies:
   - pytest
   - pytest-cov
+  - pytest-xdist
+  - pytest-timeout
   - codecov
   - pip
   - libpng

.circleci/unittest/linux/scripts/run_test.sh

@@ -6,5 +6,20 @@ eval "$(./conda/bin/conda shell.bash hook)"
 conda activate ./env
 
 export PYTORCH_TEST_WITH_SLOW='1'
+if [ "${CU_VERSION:-}" == cpu ] ; then
+    NUMPROCESSES="auto"
+    export OMP_NUM_THREADS="1"
+else
+    NUMPROCESSES="1"
+fi
+
 python -m torch.utils.collect_env
-pytest --cov=torchvision --junitxml=test-results/junit.xml -v --durations 20 test --ignore=test/test_datasets_download.py
+pytest \
+    --numprocesses=$NUMPROCESSES \
+    --timeout=10 \
+    --cov=torchvision \
+    --junitxml=test-results/junit.xml \
+    --verbose \
+    --durations 20 \
+    --ignore=test/test_datasets_download.py \
+    test

.circleci/unittest/windows/scripts/environment.yml

@@ -5,6 +5,7 @@ channels:
   - conda-forge
 dependencies:
   - pytest
+  - pytest-xdist
   - pytest-cov
   - codecov
   - pip

.circleci/unittest/windows/scripts/run_test.sh

@@ -6,5 +6,19 @@ eval "$(./conda/Scripts/conda.exe 'shell.bash' 'hook')"
 conda activate ./env
 
 export PYTORCH_TEST_WITH_SLOW='1'
+if [ "${CU_VERSION:-}" == cpu ] ; then
+    NUMPROCESSES="auto"
+    export OMP_NUM_THREADS="1"
+else
+    NUMPROCESSES="1"
+fi
+
 python -m torch.utils.collect_env
-pytest --cov=torchvision --junitxml=test-results/junit.xml -v --durations 20 test --ignore=test/test_datasets_download.py
+pytest \
+    --numprocesses=$NUMPROCESSES \
+    --cov=torchvision \
+    --junitxml=test-results/junit.xml \
+    --verbose \
+    --durations 20 \
+    --ignore=test/test_datasets_download.py \
+    test