We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 91dc371 commit 261d0a4Copy full SHA for 261d0a4
test/prototype/test_low_bit_optim.py
@@ -230,6 +230,7 @@ def world_size(self) -> int:
230
return 2
231
232
@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")
234
@skip_if_lt_x_gpu(2)
235
def test_fsdp2(self):
236
optim_classes = [low_bit_optim.Adam8bit, low_bit_optim.Adam4bit]
0 commit comments