
    ȅi                     ^    S SK r S SKJr  S\4S jrS\4S jrSS\S\R                  4S jjr	g)    Nreturnc                  :    [        [        R                  S   5      $ )NRANKintosenviron     X/home/james-whalen/.local/lib/python3.13/site-packages/torch/distributed/debug/_store.pyget_rankr      s    rzz&!""r   c                  :    [        [        R                  S   5      $ )N
WORLD_SIZEr   r
   r   r   get_world_sizer   
   s    rzz,'((r   prefixc                     [         R                  S   n[        [         R                  S   5      n[        R                  " UUSS9nU (       a  [        R
                  " X5      nU$ )NMASTER_ADDRMASTER_PORTF)	host_nameport	is_master)r   r	   r   distTCPStorePrefixStore)r   r   r   stores       r   tcpstore_clientr      sT    **]+Kbjj/0KMME
   /Lr   )debug_server)
r   torch.distributeddistributedr   r   r   r   strStorer   r
   r   r   <module>r"      s:    	  ## #) )C TZZ r   