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

Read the Docs v: v0.4.0
Versions
latest
stable
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.