Make some comparisons explicit (#60505)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60505
Test Plan: Sandcastle
Reviewed By: ngimel
Differential Revision: D29313240
fbshipit-source-id: 3f558e68cbb0328326d7540e2b3bd0c2e12ba3e2
