xla
2b16a718 - [FSDP] Don't use detach in _rebuild_full_params (#4813)

Commit
2 years ago
[FSDP] Don't use detach in _rebuild_full_params (#4813) Summary: Somehow the detach_copy implementation has some issues with FSDP that it messed up with FSDP's memory management mechanism. Let's not use it. Test Plan: CI
Author
Parents
Loading