pytorch

e3bd38d2 - [DTensor] fix test_device_mesh failure on GPU (#91783)

[DTensor] fix test_device_mesh failure on GPU (#91783) Pull Request resolved: https://github.com/pytorch/pytorch/pull/91783 Approved by: https://github.com/wanchaol