pytorch

60cd2a43 - [DeviceMesh] Add support for nD slicing (#119752)

[DeviceMesh] Add support for nD slicing (#119752) Fixes one of the issue mentioned in #118639 @mvpatel2000 Pull Request resolved: https://github.com/pytorch/pytorch/pull/119752 Approved by: https://github.com/wanchaol