File size: 199 Bytes
f4cade0 |
1 2 3 4 5 6 7 8 9 10 |
from torch.distributed.device_mesh import ( # noqa: F401
_get_device_handle,
_mesh_resources,
DeviceMesh,
init_device_mesh,
)
__all__ = ["init_device_mesh", "DeviceMesh"]
|