
    9iq                     ^    S SK r S SKJr  S SKJr  \ R
                  " \5      r\S 5       rS r	g)    N)contextmanager)castc              #   J  #    SnU(       a  SOSnSU SU 3n U R                  USU5      R                  5       nXe:X  a)  S v   U R                  US5        U R                  US5        g  U R                  U/5        Md  ! [         a    [
        R                  SUU5        e f = f7f)	NRpcGroupManagementTokenjoinleave
Token_for__ 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          \/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/torch/distributed/rpc/_utils.py_group_membership_managementr   
   s     )I%F7MD6=/2H
$$YH=DDF IIi$IIh'

H:&     X
 s   A&B#)A= ;B#=#B  B#c                 l    SSK JnJn  [        XTR	                  5       5      nUR                  XX#5      nU$ )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"   (   s5    &"<"<">?E

(
(!3C J    )
logging
contextlibr   typingr   	getLogger__name__r   r   r"    r(   r   <module>r/      s9     %  
		8	$  :r(   