This reverts commit 24f7dcd816. Reverted https://github.com/pytorch/pytorch/pull/77213 on behalf of https://github.com/davidberard98
24f7dcd816