Refactor check macros to reuse code (#38126)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38126
Test Plan: Imported from OSS
Differential Revision: D21595300
Pulled By: albanD
fbshipit-source-id: 53805053a7a1ad35e93f335e889718e699a5dce1