
    &Vjiq                     b    d dl Z d dlmZ d dlmZ  e j        e          Zed             Zd Z	dS )    N)contextmanager)castc              #   r  K   d}|rdnd}d| d| }	 |                      |d|                                          }||k    r2d V  |                     |d           |                     |d           d S 	 |                     |g           n+# t          $ r t
                              d	||            w xY w)
NRpcGroupManagementTokenjoinleave
Token_for__T DonezBGroup membership token %s timed out waiting for %s to be released.)compare_setdecodesetwaitRuntimeErrorloggererror)storenameis_join	token_keyjoin_or_leavemy_tokenreturneds          V/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/distributed/rpc/_utils.py_group_membership_managementr   
   s      )I%2FF7M2D22=22H$$YH==DDFFxEEE IIi$$$IIh'''E

H:&&&&   X  
 !s   6B (B5c                     ddl m}m} t          ||                                          }|                    | |||          }|S )N   )apiTensorPipeAgent)r   r   r    r   _get_current_rpc_agent_update_group_membership)worker_info
my_devicesreverse_device_mapr   r   r    agentrets           r   r"   r"   (   s[    &&&&&&&&#"<"<">">??E

(
(Z!3W C J    )
logging
contextlibr   typingr   	getLogger__name__r   r   r"    r(   r   <module>r/      sz     % % % % % %       
	8	$	$   :    r(   