
    [Th5                     n   % S SK r S SKrS SKJrJrJr  S SKJr  S SKrS SK	J
r  S SKJr  S SKJr  / r\\   \S'   Sr\4S\S	\R*                  4S
 jjr\4S\S	\\R0                  \4   4S jjr\" 5       qS	\\\4   4S jr\" S5      r\" S5      rS\\\4   S	\\\4   4S jrS\\\4   S	\\\4   4S jr g)    N)AnyCallableTypeVar)	ParamSpec)_log_handlers)_WaitCounter__all__defaultdestinationreturnc                    [        U 5      u  p[        R                  " SU 35      nUR                  [        R                  5        [        R
                  " S5      nUR                  U5        SUl        UR                  U5        U$ )Nzc10d-za%(asctime)s %(filename)s:%(lineno)s %(levelname)s p:%(processName)s t:%(threadName)s: %(message)sF)	_get_logging_handlerlogging	getLoggersetLevelDEBUG	FormattersetFormatter	propagate
addHandler)r   logging_handlerlog_handler_namelogger	formatters        U/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/c10d_logger.py_get_or_create_loggerr      sv    (<[(I%O'7&89:F
OOGMM"!!kI   +F
o&M    c                 N    [         U    n[        U5      R                   SU  3nX4$ )N-)r   type__name__)r   log_handlerr   s      r   r   r   '   s4      ,K{+445Q{mD**r   c           	      `   [         R                  " 5       (       Ga  UR                  S5      =(       d    UR                  S5      nU  [         R                  " UR                  S5      5       [         R                  " U5       [         R
                  " 5        [         R
                  " U5       [         R                  " 5        [         R                  " U5       S.nUS   S:X  aC  [        R                  R                  R                  5       nSR                  S U 5       5      US	'   U$ S
U  0nU$ )Ngroupprocess_grouppg)	func_namepg_namebackend
world_size
group_sizeglobal_rank
local_rankr)   nccl.c              3   8   #    U  H  n[        U5      v   M     g 7f)N)str).0vs     r   	<genexpr> _get_msg_dict.<locals>.<genexpr>A   s     /M1As   nccl_versionr'   )distis_initializedget_get_process_group_nameget_backendget_world_sizeget_ranktorchcudar.   versionjoin)r'   argskwargsr$   msg_dictr6   s         r   _get_msg_dictrE   3   s    

7#Bvzz/'B%;66vzz$7GHI**512!0023!0078"mmo.!]]512
 I&( ::??224L'*xx/M/M'MH^$
 O I;
 Or   _T_Pfuncc                    ^  [         R                  " T 5      S[        R                  S[        R                  S[
        4U 4S jj5       nU$ )NrB   rC   r   c                     >  T" U 0 UD6$ ! [          a:  n[        TR                  /U Q70 UD6nU US'   [        R	                  U5        e S nAff = f)Nerror)	ExceptionrE   r!   _c10d_loggerdebug)rB   rC   rK   rD   rH   s       r   wrapper"_exception_logger.<locals>.wrapperN   s]    	((( 	$T]]DTDVDH#('HWx(		s    
A5A

A	functoolswrapsrG   rB   rC   rF   rH   rO   s   ` r   _exception_loggerrU   M   sB    __Trww "))    Nr   c                    ^  [         R                  " T 5      S[        R                  S[        R                  S[
        4U 4S jj5       nU$ )NrB   rC   r   c                     > [        STR                   35      R                  5          T" U 0 UD6nS S S 5        U$ ! , (       d  f       W$ = f)Nzpytorch.wait_counter.c10d.)r   r!   guard)rB   rC   func_returnrH   s      r   rO   _time_logger.<locals>.wrapper\   sK    6t}}oFGMMO//K P POs	   	;
A
rQ   rT   s   ` r   _time_loggerr[   [   sB    __Trww "))   
 Nr   )!rR   r   typingr   r   r   typing_extensionsr   r>   torch.distributeddistributedr7   "torch.distributed.logging_handlersr   torch.monitorr   r	   listr1   __annotations___DEFAULT_DESTINATIONLoggerr   tupleHandlerr   rM   dictrE   rF   rG   rU   r[    r   r   <module>rj      s      ) ) '    < & c    .B 
s 
gnn 
 ,++
7??C + %&c3h , T]t_HRV, "b&1A xB' HRV,< r   