pytorch
32d22edc - [FSDP()][27/N] Add forward hook registration (#88040)

Commit
2 years ago
[FSDP()][27/N] Add forward hook registration (#88040) This PR adds the forward hook registration to composable FSDP and adds a unit test for the runtime. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88040 Approved by: https://github.com/zhaojuanmao, https://github.com/rohan-varma
Author
Committer
Parents
Loading