Update deepspeed.yaml

Ascend · Oct 26, 2024 · 6c2ffbe · 6c2ffbe
1 parent 610f949
commit 6c2ffbe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml
@@ -141,7 +141,7 @@ jobs:
         source /root/.bashrc
         unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
         
-        cd DeepSpeed/tests/
+        cd DeepSpeed/tests/unit/
         # pytest --verbose $PYTEST_OPTS --forked -n 4  unit/
         # pytest $PYTEST_OPTS --forked -m 'sequential' unit/