From c2090be01d6a4208dc3a18965e1aeb6682b91063 Mon Sep 17 00:00:00 2001
From: Michael Wyatt <michaelwyatt@microsoft.com>
Date: Thu, 16 Nov 2023 09:45:28 -0800
Subject: [PATCH] add new unit tests

---
 .github/workflows/nv-mii.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/nv-mii.yml b/.github/workflows/nv-mii.yml
index 86de2a3b0bcb..25dd0a5d2e64 100644
--- a/.github/workflows/nv-mii.yml
+++ b/.github/workflows/nv-mii.yml
@@ -54,5 +54,7 @@ jobs:
           cd DeepSpeed-MII
           pip install .[dev]
           unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
-          cd tests/legacy
+          cd tests
+          pytest $PYTEST_OPTS ./
+          cd legacy
           pytest $PYTEST_OPTS --forked -m "deepspeed" ./