pytorch/test/distributed/_composable
wz337 83e6ec2ccd [FSDP2+TP] Disable 2D state_dict (#129519)
Fixes #ISSUE_NUMBER

Gonna fill in the RFC but just want to run CI to see if anything else breaks.

Test:
```
python test/distributed/_composable/fsdp/test_fully_shard_training.py -k test_raise_not_implemented_state_dict_if_2d
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/129519
Approved by: https://github.com/awgu
2024-07-02 01:26:14 +00:00
..
fsdp [FSDP2+TP] Disable 2D state_dict (#129519) 2024-07-02 01:26:14 +00:00
fully_shard
test_checkpoint.py Adding kwargs to composable AC API to enable full capabilities (#128516) 2024-06-15 00:23:48 +00:00
test_compose.py
test_contract.py
test_replicate.py
test_replicate_with_compiler.py [Horace's PR #126446] Prevent partitioner from ever saving views (#129039) 2024-06-19 23:21:16 +00:00