
    קgm                     r    d dl Z d dlmZ d dlmZ ddlmZmZ  e j        e	          Z
ed             Zd ZdS )    N)contextmanager)cast   )apiTensorPipeAgentc              #   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          X/var/www/html/ai-engine/env/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                     t          t          t          j                              }|                    | |||          }|S )N)r   r   r   _get_current_rpc_agent_update_group_membership)worker_info
my_devicesreverse_device_mapr   agentrets         r   r"   r"   *   s@    #"<">">??E

(
(Z!3W C J    )logging
contextlibr   typingr   r   r   r   	getLogger__name__r   r   r"    r(   r   <module>r/      s     % % % % % %       " " " " " " " " 
	8	$	$   :    r(   