We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3330d29 commit 9ea1e67Copy full SHA for 9ea1e67
1 file changed
test/test_low_bit_optim.py
@@ -533,6 +533,7 @@ def world_size(self) -> int:
533
534
@skip_if_lt_x_gpu(_FSDP_WORLD_SIZE)
535
@skip_if_rocm("ROCm enablement in progress")
536
+ @pytest.mark.skipif(torch_version_at_least("2.11.0"), reason="Failing in CI")
537
def test_fsdp2(self):
538
# we do this to avoid all combinations
539
args_list = [
0 commit comments