Fix for MII unit tests (microsoft#4652)

microsoft/DeepSpeed-MII#262 Moved the legacy-MII unit tests to a subdirectory. Reflecting that change here. @loadams
mauryaavinash95 · Feb 17, 2024 · 91a18c7 · 91a18c7
1 parent 8528abd
commit 91a18c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/nv-mii.yml b/.github/workflows/nv-mii.yml
@@ -54,5 +54,5 @@ jobs:
           cd DeepSpeed-MII
           pip install .[dev]
           unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
-          cd tests
+          cd tests/legacy
           pytest $PYTEST_OPTS --forked -m "deepspeed" ./