Skip to content

GitLab

Unverified Commit 4975b05e authored by Min Xu's avatar Min Xu Committed by GitHub
Browse files

reenable a test (#1081) 


Co-authored-by: default avatarMin Xu <min.xu.public@gmail.com>
parent e71d2570
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/nn/data_parallel/test_fsdp.py
View file @ 4975b05e
...@@ -481,7 +481,6 @@ class TestReduceScatterProcessGroup(DistributedTest): ...@@ -481,7 +481,6 @@ class TestReduceScatterProcessGroup(DistributedTest):
return FullyShardedDataParallel(model, group, **config) return FullyShardedDataParallel(model, group, **config)
@pytest.mark.skip(reason="Recently flaky and not reproducible locally.")
class TestSerialization(DistributedTest): class TestSerialization(DistributedTest):
@parameterized.expand([[False, False], [True, False], [True, True], [False, True]], name_func=rename_test) @parameterized.expand([[False, False], [True, False], [True, True], [False, True]], name_func=rename_test)
def test_pickle(self, mixed_precision, cpu_offload): def test_pickle(self, mixed_precision, cpu_offload):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment