polish engine unitest

2025-10-12 02:08:03 +00:00 · 2022-03-09 12:03:49 +08:00
parent 354c0f9047
commit d271f2596b
2 changed files with 11 additions and 8 deletions
--- a/tests/test_zero_data_parallel/test_shard_model_v2.py
+++ b/tests/test_zero_data_parallel/test_shard_model_v2.py
@@ -76,6 +76,7 @@ def run_dist(rank, world_size, port):
                check_grads(model, zero_model, loose=True)


+@pytest.mark.skip(reason="Under development")
@pytest.mark.dist
@pytest.mark.parametrize("world_size", [1, 2, 4])
 def test_shard_model_v2(world_size):