From bcdabf44aeeb57d3fea75ecb91ca446c9372fbb9 Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Thu, 16 Nov 2023 11:39:51 -0800 Subject: [PATCH] Add new MII unit tests (#4693) --- .github/workflows/nv-a6000.yml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml index e1f8ed41cebe..e9c33c55f8ff 100644 --- a/.github/workflows/nv-a6000.yml +++ b/.github/workflows/nv-a6000.yml @@ -54,3 +54,10 @@ jobs: cd tests python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2' unit/ --torch_ver="2.0" --cuda_ver="12" python -m pytest --color=yes --durations=0 --verbose -rF -m 'inference_v2_ops' unit/ --torch_ver="2.0" --cuda_ver="12" + - name: MII unit tests + run: | + git clone https://github.com/microsoft/DeepSpeed-MII.git + cd DeepSpeed-MII + pip install .[dev] + cd tests + python -m pytest --color=yes --durations=0 --verbose -rF ./