Test Plan:
- wait for CI
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82000
Approved by: https://github.com/janeyx99