Gemfury - distributed/_tensor/device_mesh.py · team-dmm / torch v2.4.0 - python package

team-dmm / torch python

Repository URL to install this package:

Details

torch / distributed / _tensor / device_mesh.py

from torch.distributed.device_mesh import (  # noqa: F401
    _get_device_handle,
    _mesh_resources,
    DeviceMesh,
    init_device_mesh,
)