Skip to content

Commit 437321f

Browse files
authored
Revert "[Collective][PR 4/6] NCCL Communicator caching and preliminary stream management (ray-project#13030)"
This reverts commit ed6b212.
1 parent 61733d6 commit 437321f

29 files changed

+276
-1930
lines changed
Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,11 @@
1-
from ray.util.collective.collective import nccl_available, gloo_available, \
1+
from ray.util.collective.collective import nccl_available, mpi_available, \
22
is_group_initialized, init_collective_group, destroy_collective_group, \
3-
declare_collective_group, get_rank, get_world_size, allreduce, \
4-
allreduce_multigpu, barrier, reduce, reduce_multigpu, broadcast, \
5-
broadcast_multigpu, allgather, allgather_multigpu, reducescatter, \
6-
reducescatter_multigpu, send, send_multigpu, recv, recv_multigpu
3+
get_rank, get_world_size, allreduce, barrier, reduce, broadcast, \
4+
allgather, reducescatter, send, recv
75

86
__all__ = [
9-
"nccl_available", "gloo_available", "is_group_initialized",
10-
"init_collective_group", "destroy_collective_group",
11-
"declare_collective_group", "get_rank", "get_world_size", "allreduce",
12-
"allreduce_multigpu", "barrier", "reduce", "reduce_multigpu", "broadcast",
13-
"broadcast_multigpu", "allgather", "allgather_multigpu", "reducescatter",
14-
"reducescatter_multigpu", "send", "send_multigpu", "recv", "recv_multigpu"
7+
"nccl_available", "mpi_available", "is_group_initialized",
8+
"init_collective_group", "destroy_collective_group", "get_rank",
9+
"get_world_size", "allreduce", "barrier", "reduce", "broadcast",
10+
"allgather", "reducescatter", "send", "recv"
1511
]

0 commit comments

Comments
 (0)