Reland split unsafe version (#41484)
Summary:
Reland of https://github.com/pytorch/pytorch/pull/39299
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41484
Reviewed By: glaringlee
Differential Revision: D22552377
Pulled By: albanD
fbshipit-source-id: 1d1b713d2429ae162e04bda845ef0838c52df789