mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 19:17:04 +08:00
Hao Zhang
12 lines
502 B
Python
12 lines
502 B
Python
from ray.util.collective.collective import nccl_available, mpi_available, \
|
|
is_group_initialized, init_collective_group, destroy_collective_group, \
|
|
get_rank, get_world_size, allreduce, barrier, reduce, broadcast, \
|
|
allgather, reducescatter
|
|
|
|
__all__ = [
|
|
"nccl_available", "mpi_available", "is_group_initialized",
|
|
"init_collective_group", "destroy_collective_group", "get_rank",
|
|
"get_world_size", "allreduce", "barrier", "reduce", "broadcast",
|
|
"allgather", "reducescatter"
|
|
]
|