Fix some integer comparisons (#60894)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60894
Test Plan: Sandcastle
Reviewed By: malfet
Differential Revision: D29431512
fbshipit-source-id: b0ef7656806f378ad823e503e7c27cc563d3dc7d
