add new unit tests

microsoft · Nov 16, 2023 · c2090be · c2090be
1 parent ce5e56a
commit c2090be
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/nv-mii.yml b/.github/workflows/nv-mii.yml
@@ -54,5 +54,7 @@ jobs:
           cd DeepSpeed-MII
           pip install .[dev]
           unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
-          cd tests/legacy
+          cd tests
+          pytest $PYTEST_OPTS ./
+          cd legacy
           pytest $PYTEST_OPTS --forked -m "deepspeed" ./