viable/strict/1789822279: [FSDP2] Extract copy preparation helpers (#197372)
- PyTorch: 1199 events in the last 90 days
- PyTorch: 1185th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1789818637
What happened
pure code refactoring - no perf inidication, no new feature the goal is to let user register custom Ag/RS copy fn in next PR for better Shard(1) perf Pull Request resolved: #197372 Approved by: https://github.com/anijain2305 , https://github.com/anshul-si ghstack dependencies: #197459
Summary assembled by rule from the sources below