From d7032dae83d3253363a155daa053cbb6dbf886aa Mon Sep 17 00:00:00 2001 From: Vincent Chen Date: Mon, 1 Jul 2024 17:34:31 -0700 Subject: [PATCH] Pytest skip 2 (#3448) * test * test * test * test * test * test * fix * sleep before skip * fix * pull request target * revert * revery pr_target branches * sleep 1 * 10 sec * uncomment * dist barrier * test * dist works! * update 0.0.9 * mihir comment Co-authored-by: Mihir Patel --------- Co-authored-by: v-chen_data Co-authored-by: Mihir Patel --- tests/trainer/test_fsdp_checkpoint.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py index 3b4f26024c..154ed6b282 100644 --- a/tests/trainer/test_fsdp_checkpoint.py +++ b/tests/trainer/test_fsdp_checkpoint.py @@ -836,6 +836,7 @@ def test_fsdp_partitioned_state_dict_load( if weights_only and autoresume: pytest.skip('Weights only with autoresume is not supported') if (use_tp or use_hsdp) and version.parse(torch.__version__) < version.parse('2.3.0'): + dist.barrier() # Sync to avoid race conditions on cleaning up tmp_path pytest.skip('HSDP and TP require torch 2.3.0 or later') load_ignore_keys = [] if load_ignore_keys is None else load_ignore_keys