o `Û·ikã@sPddlmZddlmZddlZddlmZeGdd„dƒƒZGdd„dƒZdS) é)Ú dataclass)ÚListN)ÚBackendc@s*eZdZUdZeed<eed<eed<dS)Ú Communicatorz= A handle to a communicator that we are a member of. ÚnameÚrankÚbackendN)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__Úint©rrú^/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/experimental/collective/communicator.pyrs  rc@s~eZdZdZdeejjdedefdd„Z dejjfdd „Z e d eejjfd d „ƒZ e d efd d„ƒZ e d efdd„ƒZdS)ÚCommunicatorHandlezj A communicator handle used by the driver to store handles to the actors in the communicator. ÚactorsrrcCs||_||_t|ƒ|_dS)a„ Initializes the CommunicatorHandle with the given actor handles. Assumes that the communicator has already been initialized on all actors. Args: actors: A list of actor handles to be stored. name: Name of the communicator. backend: Communicator backend. See ray.util.collective.types for valid values. N)Ú_actorsÚ_namerÚ_backend)ÚselfrrrrrrÚ__init__s zCommunicatorHandle.__init__ÚactorcCs(t|jƒD] \}}||kr|SqdS)Néÿÿÿÿ)Ú enumerater)rrÚiÚarrrÚget_rank,s ÿzCommunicatorHandle.get_rankÚreturncCs|jdd…S)z@ Return all actor handles in this communicator. N)r©rrrrr2szCommunicatorHandle.actorscCó|jS©N)rr rrrr9ózCommunicatorHandle.namecCr!r")rr rrrr=r#zCommunicatorHandle.backendN)r r r r rÚrayrÚ ActorHandler rrÚpropertyrrrrrrrrsr) Ú dataclassesrÚtypingrr$Úray.util.collective.typesrrrrrrrÚs