Reduction should not coalesce_dimensions when splitting for 32bit indexing (#37788)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/37583
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37788
Differential Revision: D21387325
Pulled By: ngimel
fbshipit-source-id: dbd0f5a23e06d8c4cc68cd21b09b4b0221c4bba7