From 6c2ffbea77d852d827f1b0e1bc11dda7fab15c58 Mon Sep 17 00:00:00 2001 From: jiahao su Date: Sat, 26 Oct 2024 16:31:50 +0800 Subject: [PATCH] Update deepspeed.yaml --- .github/workflows/deepspeed.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml index 9746f80a9..6c9688451 100644 --- 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/