[tp] fix SP style regression (#111353)
[tp] fix SP style regression
Although we want to remove prepare_input/output, we should still keep
the old behavior for SequenceParallel
Pull Request resolved: https://github.com/pytorch/pytorch/pull/111353
Approved by: https://github.com/fduwjj
