2 Commits

Author	SHA1	Message	Date
Yanli Zhao	2336571cb7	make fsdp folder to be public (#72084) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72084 make fsdp folder to be public ghstack-source-id: 148173447 Test Plan: unit tests Reviewed By: mrshenli Differential Revision: D33903417 fbshipit-source-id: 7852a2adc4af09af48a5ffa52ebf210489f834d5 (cherry picked from commit bd06513cfe)	2022-02-02 15:50:14 +00:00
Yanli Zhao	df3f82a1ef	Add more FSDP unit tests to cover core logic, freezing weights and flatten parameter wrapper (#66904) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66904 Add more FSDP unit tests to cover core logic, freezing weights and flatten parameter wrappe, these unit tests are refactored to be aligned with PyTorch commonly used test classes ghstack-source-id: 141335614 Test Plan: unit tests Reviewed By: mrshenli Differential Revision: D31779565 fbshipit-source-id: c727110d1d7570c0ec49e42cadfc9e9a5e440073	2021-10-22 16:50:52 -07:00