Add more CPU tests (#47369)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47369
Test Plan: Imported from OSS
Reviewed By: ansley
Differential Revision: D24805251
Pulled By: eellison
fbshipit-source-id: f1a8210ffdc3cc88354cb4896652151d83a0345a
