DeepSpeed

[Bugfix] Resolve Rank index out of range during BWD when sp_size < world_size in Ulysses

#7809

Merged