diff --git a/.github/workflows/nvidia-rtx-3090-tests.yaml b/.github/workflows/nvidia-rtx-3090-tests.yaml index 888f75b..7764d07 100644 --- a/.github/workflows/nvidia-rtx-3090-tests.yaml +++ b/.github/workflows/nvidia-rtx-3090-tests.yaml @@ -50,6 +50,7 @@ jobs: run: | mpirun -mca orte_allowed_exit_without_sync 1 -n 2 pytest --with-mpi ./axonn/tests/test_intra_layer_conv.py - name: Run intra-layer Embedding unit tests + run: | mpirun -mca orte_allowed_exit_without_sync 1 -n 2 pytest --with-mpi ./axonn/tests/test_intra_layer_emb.py -k bw_pass mpirun -mca orte_allowed_exit_without_sync 1 -n 2 pytest --with-mpi ./axonn/tests/test_intra_layer_emb.py -k fw_pass - name: Uninstall AxoNN