pytorch

7a348a1d - Fix internal breakage caused by #82134 (#84363)

Fix internal breakage caused by #82134 (#84363) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84363 Approved by: https://github.com/rohan-varma, https://github.com/mehtanirav