Shortcuts

mmengine.dist.get_comm_device

mmengine.dist.get_comm_device(group=None)[源代码]

Return the device for communication among groups.

参数:

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

返回:

The device of backend.

返回类型:

torch.device