Skip to content

Commit 52f51a9

Browse files
tohtanaamaurya
authored andcommitted
Move modal tests to tests/v1 (deepspeedai#7557)
This PR moves active tests under `tests/unit/v1` to clarify which tests are run on modal. --------- Signed-off-by: Masahiro Tanaka <[email protected]>
1 parent 5008ac1 commit 52f51a9

File tree

6 files changed

+2
-2
lines changed

6 files changed

+2
-2
lines changed

ci/torch_latest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
def pytest():
3434
import subprocess
3535
subprocess.run(
36-
"pytest -n 4 --verbose tests/unit/runtime/zero/test_zero.py tests/unit/runtime/half_precision/test_bf16.py tests/unit/runtime/zero/test_zero_autocast.py --torch_ver=2.6 --cuda_ver=12.4".split(),
36+
"pytest -n 4 --verbose tests/unit/v1/ --torch_ver=2.6 --cuda_ver=12.4".split(),
3737
check=True,
3838
cwd=ROOT_PATH / ".",
3939
)

tests/unit/runtime/compile/test_compile_zero.py renamed to tests/unit/v1/compile/test_compile_zero.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from deepspeed.utils.torch import required_torch_version
1111
from deepspeed.accelerator import get_accelerator
1212

13-
from unit.runtime.compile.util import compare_loss
13+
from unit.v1.compile.util import compare_loss
1414
from unit.common import DistributedTest
1515
from unit.util import bf16_required_version_check, skip_on_arch
1616
import deepspeed
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)