pytorch

781b7ebe - [DeviceMesh] Expose init_device_mesh (#107969)

[DeviceMesh] Expose init_device_mesh (#107969) As title. Pull Request resolved: https://github.com/pytorch/pytorch/pull/107969 Approved by: https://github.com/wanchaol, https://github.com/fduwjj