o c²“iCã@sJdZddlZddlZe e¡ZejGdd„dƒƒZejGdd„dƒƒZdS)z#Some utility class for Collectives.éNc@ó(eZdZdZdd„Zdd„Zdd„ZdS) ÚNCCLUniqueIDStorezéNCCLUniqueID 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. cCs||_d|_dS)N)ÚnameÚnccl_id)Úselfr©rúL/home/ubuntu/.local/lib/python3.10/site-packages/ray/util/collective/util.pyÚ__init__s zNCCLUniqueIDStore.__init__cCs ||_|jS)zÅ Initialize the NCCL unique ID for this store. Args: uid: the unique ID generated via the NCCL generate_communicator_id API. Returns: None )r)rÚuidrrrÚset_ids zNCCLUniqueIDStore.set_idcCs|js t d |j¡¡|jS)z*Get the NCCL unique ID held in this store.z.The NCCL ID has not been set yet for store {}.)rÚloggerÚwarningÚformatr©rrrrÚget_id%s  ÿzNCCLUniqueIDStore.get_idN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r rrrrrrs   rc@r) ÚInfozrStore the group information created via `create_collective_group`. Note: Should be used as a NamedActor. cCs"d|_d|_d|_d|_d|_dS)Néÿÿÿÿi0u©ÚidsÚ world_sizeÚrankÚbackendÚ gloo_timeoutrrrrr 5s  z Info.__init__cCs"||_||_||_||_||_dS)zStore collective information.Nr)rrrrrrrrrÚset_info<s  z Info.set_infocCs|j|j|j|j|jfS)z-Get previously stored collective information.rrrrrÚget_infoDs ûz Info.get_infoN)rrrrr rrrrrrr.s  r) rÚrayÚloggingÚ getLoggerrr ÚremoterrrrrrÚs %