pytorch
4635f56d - [FSDP()][18/N] Refactor `pre_forward_unshard()` (#87931)

Commit

2 years ago

[FSDP()][18/N] Refactor `pre_forward_unshard()` (#87931) Pull Request resolved: https://github.com/pytorch/pytorch/pull/87931 Approved by: https://github.com/mrshenli

Author

awgu

Committer

pytorchmergebot

Parents

0a752688

pytorch 4635f56d - [FSDP()][18/N] Refactor `pre_forward_unshard()` (#87931)

pytorch
4635f56d - [FSDP()][18/N] Refactor `pre_forward_unshard()` (#87931)