pytorch

fc6f2f6e - [spmd] simplify data parallel tests (#99901)

[spmd] simplify data parallel tests (#99901) As titled Pull Request resolved: https://github.com/pytorch/pytorch/pull/99901 Approved by: https://github.com/awgu, https://github.com/mrshenli