pytorch
705ad36c - Dynamo asserts FSDP wrapped modules use_orig_param (#89523)

Commit
2 years ago
Dynamo asserts FSDP wrapped modules use_orig_param (#89523) - This is a strict requirement given the way dynamo+FSDP is implemented, but isn't convenient to assert. - By plumbing use_orig_param field on all wrapped modules, we can do this assertion inside dynamo Pull Request resolved: https://github.com/pytorch/pytorch/pull/89523 Approved by: https://github.com/awgu
Author
Committer
Parents
Loading