Use pytest-rerunfailures in mpi_gpu step

PennyLaneAI · Oct 25, 2023 · 0e7ab41 · 0e7ab41
1 parent 0bae517
commit 0e7ab41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/tests_linux_x86_mpi_gpu.yml b/.github/workflows/tests_linux_x86_mpi_gpu.yml
@@ -223,7 +223,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh && module use /opt/modules/ && module load ${{ matrix.mpilib }}
           python -m pip install -r requirements-dev.txt
-          python -m pip install cmake custatevec-cu11 mpi4py openfermionpyscf
+          python -m pip install cmake custatevec-cu11 mpi4py openfermionpyscf pytest-rerunfailures
           SKIP_COMPILATION=True PL_BACKEND=lightning_qubit python -m pip install -e . -vv
 
       - name: Install Stable PennyLane
@@ -243,7 +243,7 @@ jobs:
       - name: Run unit tests for MPI-enabled lightning.gpu device
         run: |
           source /etc/profile.d/modules.sh && module use /opt/modules/ && module load ${{ matrix.mpilib }}
-          PL_DEVICE=lightning.gpu /opt/mpi/${{ matrix.mpilib }}/bin/mpirun -np 2 python -m pytest ./mpitests $COVERAGE_FLAGS
+          PL_DEVICE=lightning.gpu /opt/mpi/${{ matrix.mpilib }}/bin/mpirun -np 2 python -m pytest --reruns 5 ./mpitests $COVERAGE_FLAGS
           mv coverage.xml coverage-${{ github.job }}-lightning_gpu_${{ matrix.mpilib }}-main.xml
           # PL_DEVICE=lightning.gpu /opt/mpi/${{ matrix.mpilib }}/bin/mpirun --oversubscribe -n 4 pytest -s -x mpitests/test_device.py -k test_create_device $COVERAGE_FLAGS