
    wiH                        U d dl Z d dlZd dlmZmZmZmZmZmZ d dl	m
Z
 d dlZd dlmZ d dlmZ d dlmZ g Zee         ed<   dZefded	ej        fd
Zefded	eej        ef         fdZ e            ad	eeef         fdZ ed          Z e
d          Zdeeef         d	eeef         fdZdeeef         d	eeef         fdZ dS )    N)AnyCallableDictListTupleTypeVar)	ParamSpec)_log_handlers)_WaitCounter__all__defaultdestinationreturnc                     t          |           \  }}t          j        d|           }|                    t          j                   t          j        d          }|                    |           d|_        |                    |           |S )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        m/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/torch/distributed/c10d_logger.py_get_or_create_loggerr      s    (<[(I(I%O%9'799::F
OOGM"""!k I   +++F
o&&&M    c                 V    t           |          }t          |          j         d|  }||fS )N-)r
   type__name__)r   log_handlerr   s      r   r   r   '   s9      ,K{++4DD{DD)**r    c           	      L   t          j                    r
|                    d          p|                    d          }|  t          j        |                    d                     t          j        |           t          j                     t          j        |           t          j                     t          j        |           d}|d         dk    rEt          j        j	        
                                }d                    d |D                       |d	<   nd
|  i}|S )Ngroupprocess_grouppg)	func_namepg_namebackend
world_size
group_sizeglobal_rank
local_rankr,   nccl.c              3   4   K   | ]}t          |          V  d S )N)str).0vs     r   	<genexpr>z _get_msg_dict.<locals>.<genexpr>A   s(      /M/M1A/M/M/M/M/M/Mr    nccl_versionr*   )distis_initializedget_get_process_group_nameget_backendget_world_sizeget_ranktorchcudar1   versionjoin)r*   argskwargsr'   msg_dictr8   s         r   _get_msg_dictrG   3   s    


7##Bvzz/'B'B%6vzz$7G7GHHJ*5113!0224!0779"moo/!]5113
 
 I&(( :?2244L'*xx/M/M/M/M/M'M'MH^$ I
 Or    _T_Pfuncc                      t          j                   dt          j        dt          j        dt
          f fd            }|S )NrD   rE   r   c                      	  | i |S # t           $ r;}t          j        g| R i |}| |d<   t                              |            d }~ww xY w)Nerror)	ExceptionrG   r$   _c10d_loggerdebug)rD   rE   rM   rF   rJ   s       r   wrapperz"_exception_logger.<locals>.wrapperN   s~    	4(((( 	 	 	$T]DTDDDVDDH#(
HWx(((		s    
A6AA	functoolswrapsrI   rD   rE   rH   rJ   rQ   s   ` r   _exception_loggerrV   M   sV    _Trw ")        Nr    c                      t          j                   dt          j        dt          j        dt
          f fd            }|S )NrD   rE   r   c                      t          dj                                                   5   | i |}d d d            n# 1 swxY w Y   |S )Nzpytorch.wait_counter.c10d.)r   r$   guard)rD   rE   func_returnrJ   s      r   rQ   z_time_logger.<locals>.wrapper\   s    Ft}FFGGMMOO 	0 	0$///K	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0s   	A  AArR   rU   s   ` r   _time_loggerr[   [   sV    _Trw ")       
 Nr    )!rS   r   typingr   r   r   r   r   r   typing_extensionsr	   r@   torch.distributeddistributedr9   "torch.distributed.logging_handlersr
   torch.monitorr   r   r4   __annotations___DEFAULT_DESTINATIONLoggerr   Handlerr   rO   rG   rH   rI   rV   r[    r    r   <module>rg      s         < < < < < < < < < < < < < < < < ' ' ' ' ' '              < < < < < < & & & & & & c      .B 
 
s 
gn 
 
 
 
 ,+ ++
7?C + + + + %$&&c3h    , WT]]Yt__HRV, "b&1A    xB' HRV,<      r    