frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099)"

Browse Source 
This reverts commit eaa64339d640ed1d36520ada379213f8361be5ff.
This commit is contained in:
Nikita Shulga
2023-12-05 08:59:36 -08:00
parent 0a9819e3e1
commit a827ac71f2
33 changed files with 101 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/testing/_internal/distributed/distributed_test.py
View File

@ -10046,7 +10046,7 @@ class DistributedTest:
"""
world_size = int(os.environ["WORLD_SIZE"])
from torch.distributed.device_mesh import init_device_mesh
from torch.distributed._device_mesh import init_device_mesh
device_mesh = init_device_mesh("cuda", (world_size,))
pg = _get_default_group()