Simplify check (#106044)
Summary: Simplify check / refactor for readability
Test Plan: sandcastle, github
Differential Revision: D47800732
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106044
Approved by: https://github.com/mikaylagawarecki
