diff --git a/.github/workflows/nv-mii.yml b/.github/workflows/nv-mii.yml index 2b101b023d67..86de2a3b0bcb 100644 --- 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" ./