Temporarily skip thunder test due to timeout (Lightning-AI#1304)

mcolletta · Apr 16, 2024 · 02b04ba · 02b04ba
1 parent 3f2f70a
commit 02b04ba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_thunder_fsdp.py b/tests/test_thunder_fsdp.py
@@ -263,6 +263,8 @@ def set_up_planner(self, state_dict, metadata, is_coordinator):
 
 @RunIf(min_cuda_gpus=2, thunder=True, standalone=True)
 def test_save_load_sharded_checkpoint(tmp_path):
+    pytest.skip("Temporarily disabled, often exceeds 5 min timeout")
+
     strategy = ThunderFSDPStrategy(state_dict_type="sharded", broadcast_from=0)
     fabric = Fabric(accelerator="cuda", devices=2, strategy=strategy)
     fabric.launch()