pytorch
26322544 - Add limited FSDP correctness to torchdynamo benchmark (#89469)

Commit
2 years ago
Add limited FSDP correctness to torchdynamo benchmark (#89469) - Does not do recursive wrapping - Only supports accuracy bench - Mainly useful for sweeping over models for correctness, in part to evaluate whether dynamo support for FSDP is breaking anywhere Pull Request resolved: https://github.com/pytorch/pytorch/pull/89469 Approved by: https://github.com/davidberard98, https://github.com/aazzolini
Author
Committer
Parents
Loading