Fix some comparison warnings (#60875)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60875
Test Plan: Sandcastle
Reviewed By: ngimel
Differential Revision: D29406593
fbshipit-source-id: 0eb070ef05c1cd343c9e835786b42014d0553aa5