pytorch
3dfa755a - [MTPG] Enable for some tests in test_fsdp_misc (#102043)

Commit
1 year ago
[MTPG] Enable for some tests in test_fsdp_misc (#102043) Enables MTPG for some FSDP tests in this file. Tests that need the backward pass and warning logging are left as follow up work. Backward pass issue: It seems that there is a hang with all_gather. Will sync with @kumpera on this. Warning issue: We have a couple tests that regex check on warnings, but in the multithreaded scenario these warnings are somehow not logged. Differential Revision: [D43209769](https://our.internmc.facebook.com/intern/diff/D43209769/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/102043 Approved by: https://github.com/awgu
Author
Committer
Parents
Loading