Fix a casting warning (#42451)
Summary:
Fix an annoying casting warning
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42451
Reviewed By: yf225
Differential Revision: D22993194
Pulled By: ailzhang
fbshipit-source-id: f317a212d4e768d49d24f50aeff9c003be2fd30a
