partial revert of D32522826 (#68889)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68889
Reviewed By: cpuhrsch, ejguan
Differential Revision: D32650385
Pulled By: Krovatkin
fbshipit-source-id: 2c4a30cfc729a023b592b6b6e1959bbd2ad6f7cf