split workflow

huggingface · Dec 12, 2024 · ce2613c · ce2613c
1 parent 07423f1
commit ce2613c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/test_inf2.yml b/.github/workflows/test_inf2.yml
@@ -60,4 +60,8 @@ jobs:
       - name: Run other generation tests
         run: |
           source aws_neuron_venv_pytorch/bin/activate
-          HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} pytest -m is_inferentia_test tests/generation
+          HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} pytest -m is_inferentia_test --ignore=tests/generation/test_parallel.py tests/generation
+      - name: Run parallel tests
+        run: |
+          source aws_neuron_venv_pytorch/bin/activate
+          HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} pytest -m is_inferentia_test tests/generation/test_parallel.py