[Composable] Add more sharding strategies to runtime test (#105205)
Add more sharding strategies to ensure equivalence
Differential Revision: [D47462392](https://our.internmc.facebook.com/intern/diff/D47462392/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105205
Approved by: https://github.com/awgu