pytorch
1859e5f0 - [FSDP] Enforce wrapper_cls as a mandatory kwarg in enable_wrap. (#69358)

Commit
3 years ago
[FSDP] Enforce wrapper_cls as a mandatory kwarg in enable_wrap. (#69358) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/69358 Enforces and raises error earlier if wrapper_cls is not provided as an arg into enable_wrap() function. Also improves the documentation. ghstack-source-id: 144807950 Test Plan: CI Reviewed By: zhaojuanmao Differential Revision: D32826963 fbshipit-source-id: d1b98df021e86d3d87a626e82facf6230b571a55
Author
Parents
Loading