This reverts commit 1e2c4a6e0e. Reverted https://github.com/pytorch/pytorch/pull/84190 on behalf of https://github.com/malfet due to Needs internal changes, has to be landed via co-dev
1e2c4a6e0e