Fix smoke test jobs (#39638)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/39626.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/39638
Differential Revision: D21924224
Pulled By: ezyang
fbshipit-source-id: 8da75e401bfbff5e11ceeccefd77d0fad81356e4
