[Test] Sync DeviceMesh tests with MeshLayout changes-pytorch-AtomGit

Star860

Star860

已合并

[Test] Sync DeviceMesh tests with MeshLayout changes #43811

MengJiayi创建于 8月5日从Jiayi_Meng/pytorch： master 合入到 Ascend/pytorch： master

[Test] Sync DeviceMesh tests with MeshLayout changes #43811

已合并

MengJiayi创建于 8月5日

从Jiayi_Meng/pytorch master合入到Ascend/pytorch master