pytorch

cce2c52b - [pt2] support vmap (#101707)

[pt2] support vmap (#101707) Teach dynamo about `vmap` Pull Request resolved: https://github.com/pytorch/pytorch/pull/101707 Approved by: https://github.com/zou3519