Skip to content

Commit 9ea1e67

Browse files
authored
Skip test_fsdp2 if PyTorch version is 2.11.0 or higher (#4168)
* Skip test_fsdp2 if PyTorch version is 2.11.0 or higher * fix ruff
1 parent 3330d29 commit 9ea1e67

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

test/test_low_bit_optim.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -533,6 +533,7 @@ def world_size(self) -> int:
533533

534534
@skip_if_lt_x_gpu(_FSDP_WORLD_SIZE)
535535
@skip_if_rocm("ROCm enablement in progress")
536+
@pytest.mark.skipif(torch_version_at_least("2.11.0"), reason="Failing in CI")
536537
def test_fsdp2(self):
537538
# we do this to avoid all combinations
538539
args_list = [

0 commit comments

Comments
 (0)