pytorch

0a008580 - Implement checks for vmap escaped errors (#89585)

Implement checks for vmap escaped errors (#89585) Follow on to https://github.com/pytorch/pytorch/pull/89077 Pull Request resolved: https://github.com/pytorch/pytorch/pull/89585 Approved by: https://github.com/zou3519