diff --git a/python/pylibwholegraph/pylibwholegraph/torch/comm.py b/python/pylibwholegraph/pylibwholegraph/torch/comm.py index cc27fa41e..4796ef1c8 100644 --- a/python/pylibwholegraph/pylibwholegraph/torch/comm.py +++ b/python/pylibwholegraph/pylibwholegraph/torch/comm.py @@ -31,6 +31,7 @@ all_comm_local_rank = 0 all_comm_local_size = 1 + def reset_communicators(): global all_comm_world_rank, all_comm_world_size, all_comm_local_rank, all_comm_local_size global global_communicators, local_node_communicator, local_device_communicator