
    }Yh5                        U d dl Z d dlZd dl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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        o/var/www/tools.fuzzalab.pt/emblema-extractor/venv/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cudar.   versionjoin)r'   argskwargsr$   msg_dictr5   s         r   _get_msg_dictrD   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 )NrA   rB   r   c                      	  | i |S # t           $ r;}t          j        g| R i |}| |d<   t                              |            d }~ww xY w)Nerror)	ExceptionrD   r!   _c10d_loggerdebug)rA   rB   rJ   rC   rG   s       r   wrapperz"_exception_logger.<locals>.wrapperN   s~    	4(((( 	 	 	$T]DTDDDVDDH#(
HWx(((		s    
A6AA	functoolswrapsrF   rA   rB   rE   rG   rN   s   ` r   _exception_loggerrS   M   sV    _Trw ")        Nr   c                      t          j                   dt          j        dt          j        dt
          f fd            }|S )NrA   rB   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)rA   rB   func_returnrG   s      r   rN   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rO   rR   s   ` r   _time_loggerrX   [   sV    _Trw ")       
 Nr   )!rP   r   typingr   r   r   typing_extensionsr   r=   torch.distributeddistributedr6   "torch.distributed.logging_handlersr   torch.monitorr   r	   listr1   __annotations___DEFAULT_DESTINATIONLoggerr   tupleHandlerr   rL   dictrD   rE   rF   rS   rX    r   r   <module>rg      s         ) ) ) ) ) ) ) ) ) ) ' ' ' ' ' '              < < < < < < & & & & & & c      .B 
 
s 
gn 
 
 
 
 ,+ ++
7?C + + + + %$&&c3h    , WT]]Yt__HRV, "b&1A    xB' HRV,<      r   