&`iFdZddlZddlZddlZejeZejGddZejGddZ dS)z#Some utility class for Collectives.Nc*eZdZdZdZdZdZdZdS)NCCLUniqueIDStorezNCCLUniqueID Store as a named actor class. Args: name: the unique name for this named actor. Attributes: name: the unique name for this named actor. nccl_id: the NCCLUniqueID held in this store. cR||_d|_tj|_dS)N)namenccl_idasyncioEventevent)selfrs l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/collective/util.py__init__zNCCLUniqueIDStore.__init__s!  ]__ cJK||_|j|S)z Initialize the NCCL unique ID for this store. Args: uid: the unique ID generated via the NCCL generate_communicator_id API. Returns: The NCCL unique ID set. )rr set)r uids r set_idzNCCLUniqueIDStore.set_ids&   rcRK|jd{V|jS)z4Wait for the NCCL unique ID to be set and return it.N)r waitrr s r wait_and_get_idz!NCCLUniqueIDStore.wait_and_get_id)s2joo|rc|js2td|j|jS)z*Get the NCCL unique ID held in this store.z.The NCCL ID has not been set yet for store {}.)rloggerwarningformatrrs r get_idzNCCLUniqueIDStore.get_id.s?|  NN@GG RR   |rN)__name__ __module__ __qualname____doc__r rrrrr rr sZ%%%    rrc$eZdZdZdZdZdZdS)InfozrStore the group information created via `create_collective_group`. Note: Should be used as a NamedActor. cLd|_d|_d|_d|_d|_dS)Ni0uids world_sizerankbackend gloo_timeoutrs r r z Info.__init__>s,  !rcL||_||_||_||_||_dS)zStore collective information.Nr%)r r&r'r(r)r*s r set_infoz Info.set_infoEs,$  (rcB|j|j|j|j|jfS)z-Get previously stored collective information.r%rs r get_infoz Info.get_infoMs( H O I L     rN)rrrrr r,r.r rr r"r"7sK """)))     rr") rrloggingray getLoggerrrremoterr"r rr r3s))  8 $ $))))))) )X           r