Signed-off-by: Eli Uriegas <eliuriegasfb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77370

Approved by: https://github.com/suo, https://github.com/janeyx99