Fixes #62636 (#62670)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/62636.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62670
Reviewed By: ezyang
Differential Revision: D30102179
Pulled By: soulitzer
fbshipit-source-id: 38480463ef354f2c12ed83e6678aed26b0b96efe
