pytorch

d93b1b9c - Address feedback from previous PR (#86622)

Address feedback from previous PR (#86622) Pull Request resolved: https://github.com/pytorch/pytorch/pull/86622 Approved by: https://github.com/albanD