
    [Thm                     n    S SK r S SKJr  S SKJr  SSKJrJr  \ R                  " \	5      r
\S 5       rS rg)    N)contextmanager)cast   )apiTensorPipeAgentc              #   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          T/var/www/auris/envauris/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                 p    [        [        [        R                  " 5       5      nUR	                  XX#5      nU$ )N)r   r   r   _get_current_rpc_agent_update_group_membership)worker_info
my_devicesreverse_device_mapr   agentrets         r   r"   r"   *   s4    #"<"<">?E

(
(!3C J    )logging
contextlibr   typingr   r   r   r   	getLogger__name__r   r   r"    r(   r   <module>r/      s<     %  " 
		8	$  :r(   