pylibwholegraph.torch.comm.set_world_info

pylibwholegraph.torch.comm.set_world_info(world_rank: int, world_size: int, local_rank: int, local_size: int)

Set the global world’s information. This is used for create common used communicators, like local node communicator, global communicator, or local device communicator.

Parameters:

• world_rank – world rank of current process.

• world_size – world size

• local_rank – local rank of current process in current machine node.

• local_size – local size of each machine node

Returns:

None