Shortcuts

mmengine.dist.get_comm_device

mmengine.dist.get_comm_device(group=None)[source]

Return the device for communication among groups.

Parameters

group (ProcessGroup, optional) – The process group to work on.

Returns

The device of backend.

Return type

torch.device