mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-05 00:14:54 +08:00
35f36363ec
This reverts commit 2ee6b97464d17fcf4c1fc67c29868fa30d0c16e1. Reverted https://github.com/pytorch/pytorch/pull/133113 on behalf of https://github.com/wanchaol due to looks like it break some internal type imports ([comment](https://github.com/pytorch/pytorch/pull/133113#issuecomment-2295670911))
10 lines
190 B
Python
10 lines
190 B
Python
from torch.distributed.device_mesh import ( # noqa: F401
|
|
_get_device_handle,
|
|
_mesh_resources,
|
|
DeviceMesh,
|
|
init_device_mesh,
|
|
)
|
|
|
|
|
|
__all__ = ["init_device_mesh", "DeviceMesh"]
|