diff --git a/python/pylibwholegraph/pylibwholegraph/torch/comm.py b/python/pylibwholegraph/pylibwholegraph/torch/comm.py index 4796ef1c8..aa15d3a0a 100644 --- a/python/pylibwholegraph/pylibwholegraph/torch/comm.py +++ b/python/pylibwholegraph/pylibwholegraph/torch/comm.py @@ -44,6 +44,7 @@ def reset_communicators(): all_comm_local_rank = 0 all_comm_local_size = 1 + def 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,