new_group¶
dragon.distributed.
new_group
(
ranks=None,
backend=None,
verbose=False
)[source]¶Create a new communication group.
The
ranks
can be set to None to create an empty group for disabling the distributed utilities.If
backend
is None, select as: NCCL > MPI.Note that this function should be called from all processes, even if they are not going to be included in this group.
- Parameters:
- ranks (Sequence[int], optional) – The rank of processes to be included.
- backend ({'AUTO', 'MPI', 'NCCL'}, optional) – The optional backend.
- verbose (bool, optional, default=False) –
True
to log the group info.