dmlcloud.is_root

dmlcloud.is_root(group=None)

Check if the current rank is the root rank (rank 0).

Parameters:

group (ProcessGroup, optional) – The process group to work on. If None (default), the default process group will be used.