Skip to content

Commit d0e3464

Browse files
committed
xfail meta device init with fsdp2 test
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent d1399cc commit d0e3464

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

recipes/esm2_native_te_mfsdp/test_train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,7 @@ def test_sanity_convergence_fsdp2_eager(mock_distributed_config, tmp_path):
222222
assert final_loss < 3.0, f"Final loss {final_loss} is too high"
223223

224224

225+
@pytest.mark.xfail(reason="This passes on my local 5090 but fails on CI (L4) (BIONEMO-2719)")
225226
def test_sanity_convergence_fsdp2_eager_meta_device(mock_distributed_config, tmp_path):
226227
"""Test that the main function can be invoked wrapping the model in FSDP2 and using meta-device init."""
227228

0 commit comments

Comments
 (0)