__init__.py 479 B

123456789101112131415161718192021
  1. from ray.experimental.collective.collective import (
  2. create_collective_group,
  3. destroy_all_collective_groups,
  4. destroy_collective_group,
  5. get_collective_groups,
  6. )
  7. from ray.experimental.collective.operations import (
  8. allgather,
  9. allreduce,
  10. reducescatter,
  11. )
  12. __all__ = [
  13. "allgather",
  14. "allreduce",
  15. "reducescatter",
  16. "get_collective_groups",
  17. "create_collective_group",
  18. "destroy_collective_group",
  19. "destroy_all_collective_groups",
  20. ]