Unify checks for `normal` (#70087)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70087
Test Plan: Imported from OSS
Reviewed By: davidberard98
Differential Revision: D34089965
Pulled By: bdhirsh
fbshipit-source-id: 17d7eab3d8d60d03ca8ee63875ff2813bb2992c8
(cherry picked from commit 6cd1e9f53135a36bb46089dc8db6cd40d11eb80f)