pytorch

d05a6e5a - Add missing DeviceMesh import (#109187)

Add missing DeviceMesh import (#109187) The test is broken after https://github.com/pytorch/pytorch/pull/107533#issuecomment-1709529759 Pull Request resolved: https://github.com/pytorch/pytorch/pull/109187 Approved by: https://github.com/clee2000