pytorch

4114c1ea - Revert "[dtensor] remove typing hack of DeviceMesh (#94526)"

Revert "[dtensor] remove typing hack of DeviceMesh (#94526)" This reverts commit 70b063db0e2b55b24c096884b2375376c0925453. Reverted https://github.com/pytorch/pytorch/pull/94526 on behalf of https://github.com/atalman due to breaking internal builds