Forward fix for FB internal breakage (manual export of internal diff D39421802) (#84871)
D39421802 is a forward-fix for D39419569 (corresponds to #84806).
The forward-fix is mostly internal-facing, but has a single line that has to be exported to GH first. Manual export was required, because automatic export failed due to diff dependecies.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84871
Approved by: https://github.com/mehtanirav