Mjq^ddlZddlmZddlmZej eZedZdZ y)N)contextmanager)castc#<Kd}|rdnd}d|d|} |j|d|j}||k(r)d|j|d|j|dy |j|gc#t$rt j d||wxYww) NRpcGroupManagementTokenjoinleave Token_for__DonezBGroup membership token %s timed out waiting for %s to be released.) compare_setdecodesetwait RuntimeErrorloggererror)storenameis_join token_key join_or_leavemy_tokenreturneds [/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/distributed/rpc/_utils.py_group_membership_managementr s)I%F7MD6=/2H $$YH=DDF x   IIi $ IIh '   H:&     X   sA!B$A76B7"BBcrddlm}m}t||j }|j ||||}|S)N)apiTensorPipeAgent)r rr r_get_current_rpc_agent_update_group_membership) worker_info my_devicesreverse_device_maprrr agentrets rr"r"(s;& #"<"<"> ?E  ( (Z!3W C J) logging contextlibrtypingr getLogger__name__rrr"r(rr/s;%   8 $:r(