Split reduction compile units (#37205)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37205
Test Plan: Imported from OSS
Differential Revision: D21233254
Pulled By: ngimel
fbshipit-source-id: 68b37ebbdd715a30c616e425a39b6b21c01b37e2