pytorch
f6696c5a - export CPUOffload in _fsdp package (#68308)

Commit
3 years ago
export CPUOffload in _fsdp package (#68308) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68308 export CPUOffload in _fsdp package, as cpu_offload config in FSDP API needs to import this class ghstack-source-id: 143560608 Test Plan: unit tests Reviewed By: rohan-varma Differential Revision: D32408719 fbshipit-source-id: ee5c40ec91a423fbd58872fbdeb5f2dda8a3d89e
Author
Parents
Loading