Switch order to check status of MII tests

microsoft · Oct 28, 2024 · d8b4840 · d8b4840
1 parent 637ab2c
commit d8b4840
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml
@@ -54,12 +54,6 @@ jobs:
       - name: Python environment
         run: |
           python -m pip list
-      - name: Unit tests
-        run: |
-          unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
-          cd tests
-          python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2' unit/ --torch_ver="2.1" --cuda_ver="12"
-          python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2_ops' unit/ --torch_ver="2.1" --cuda_ver="12"
       - name: MII unit tests
         run: |
           BRANCH="main"
@@ -72,3 +66,9 @@ jobs:
           pip install .[dev]
           cd tests
           python -m pytest --color=yes --durations=0 --verbose -rF ./
+      - name: Unit tests
+        run: |
+          unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
+          cd tests
+          python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2' unit/ --torch_ver="2.1" --cuda_ver="12"
+          python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2_ops' unit/ --torch_ver="2.1" --cuda_ver="12"