pytorch

3319dbcd - Update vmap guard to avoid recompilations (#119061)

Update vmap guard to avoid recompilations (#119061) Pull Request resolved: https://github.com/pytorch/pytorch/pull/119061 Approved by: https://github.com/zou3519