pr in test infra: https://github.com/pytorch/test-infra/pull/366
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78772
Approved by: https://github.com/suo, https://github.com/janeyx99