Skip to content

Commit 261d0a4

Browse files
authored
skip fsdp2 low bit optim test in nightlies (#653)
* skip fsdp2 low bit optim test in nightlies * Update test_low_bit_optim.py * Update test_low_bit_optim.py
1 parent 91dc371 commit 261d0a4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/prototype/test_low_bit_optim.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,7 @@ def world_size(self) -> int:
230230
return 2
231231

232232
@pytest.mark.skipif(not TORCH_VERSION_AFTER_2_4, reason="torch >= 2.4 required")
233+
@pytest.mark.skipif(TORCH_VERSION_AFTER_2_5, reason="https://github.com/pytorch/ao/issues/652")
233234
@skip_if_lt_x_gpu(2)
234235
def test_fsdp2(self):
235236
optim_classes = [low_bit_optim.Adam8bit, low_bit_optim.Adam4bit]

0 commit comments

Comments
 (0)