trying to fix testX (#16370)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16370
passed locally but seems testX has some problem
Reviewed By: ezyang
Differential Revision: D13820250
fbshipit-source-id: e4ad9d1ec99508867d4ead46753a7fb7019c50bd