pytorch

fc380a2b - [ez] Minor refactors (#107656)

[ez] Minor refactors (#107656) Pull Request resolved: https://github.com/pytorch/pytorch/pull/107656 Approved by: https://github.com/angelayi