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