From e8a6ce4f9f34b1bfdd3cf96d71d4c19b4219a50c Mon Sep 17 00:00:00 2001 From: Brad Rees <34135411+BradReesWork@users.noreply.github.com> Date: Fri, 19 Jan 2024 12:51:46 -0500 Subject: [PATCH] Update python/pylibwholegraph/pylibwholegraph/torch/comm.py --- python/pylibwholegraph/pylibwholegraph/torch/comm.py | 1 + 1 file changed, 1 insertion(+) 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,