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