o
    hq                     @   sB   d dl Z d dlmZ d dlmZ e eZedd Zdd Z	dS )    N)contextmanager)castc                 c   s    d}|rdnd}d| d| }	 |  |d| }||kr0d V  | |d | |d d S z| |g W n tyG   td	||  w q)
NZRpcGroupManagementTokenjoinZleaveZ
Token_for__T DonezBGroup membership token %s timed out waiting for %s to be released.)Zcompare_setdecodesetwaitRuntimeErrorloggererror)storenameis_joinZ	token_keyZjoin_or_leaveZmy_tokenZreturned r   b/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torch/distributed/rpc/_utils.py_group_membership_management
   s,   r   c                 C   s2   ddl m}m} t|| }|| |||}|S )N   )apiTensorPipeAgent)r   r   r   r   Z_get_current_rpc_agent_update_group_membership)Zworker_infoZ
my_devicesZreverse_device_mapr   r   r   Zagentretr   r   r   r   (   s   r   )
logging
contextlibr   typingr   	getLogger__name__r   r   r   r   r   r   r   <module>   s   

