
    #Ki3                     d   % S SK r S SKrS SKrS SKJr  S SKJrJr  S SKJ	r	  S SK
Jr  S SKJs  Jr  S SKJr  \R"                  " 5       r/ r\\   \S'   \R.                  " \5      q\" S5      r\	" S	5      rS
\\\4   4S jrS
\\\4   4S jr SS\S\S
\\\\4   /\\\4   4   4S jjrS\ 4S jr!g)    N)Callable)AnyTypeVar)	ParamSpec)uuid4)DCP_LOGGER_NAME__all___T_Preturnc                     0 nUR                  S5      nUR                  S5      nUR                  S5      nUR                  S5      nU(       d  U=(       d    U=n(       a  [        USS5      nUb  [        U5      OUUS'   [        [        5       R                  5      US'   U(       a  UR
                  R                  US'   U(       a  UR
                  R                  US'   U(       a  UR
                  R                  US'   U$ )z(
Extracts log data from dcp method args
storage_writerstorage_readerplannercheckpoint_idNuuid)getgetattrstrr   int	__class____name__)argskwargsmsg_dictr   r   r   r   
serializers           e/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/torch/distributed/checkpoint/logger.py_msg_dict_from_dcp_method_argsr      s     H ZZ 01NZZ 01Njj#GJJ/MN,LnLjL
OTB ,7M] _ 57;;'HV%3%=%=%F%F!"%3%=%=%F%F!"%//88O    c                 n    [        U0 UD6nUR                  [        R                  " U /UQ70 UD65        U$ )N)r   updatec10d_logger_get_msg_dict)	func_namer   r   r   s       r   r#   r#   >   s7    -t>v>HOOK--iI$I&IJOr   log_exceptionswrapper_kwargsc                 B   ^ ^ S[         [        [        4   4U U4S jjnU$ )zKThis method decorator logs the start, end, and exception of wrapped events.funcc                    >^  [         R                  " T 5      S[        R                  S[        R                  S[
        4U UU4S jj5       nU$ )Nr   r   r   c                    > [        TR                  /U Q70 0 T	EUED6nSUS'   [        R                  " 5       nX2S'   TUS'   [        R                  U5         T" U 0 UD6nSUS'   [        R                  " 5       n[        R                  " 5       US'   Xc-
  US'   [        R                  U5        U$ ! [         aE  nT(       a8  SUS'   U US'   [        R                  " 5       US'   [        R                  U5        e S nAff = f)	Nstarteventtimer%   	exceptionerrorendtimes_spent)r#   r   r-   time_ns_dcp_loggerdebugBaseExceptionr/   )
r   r   r   t0resultr/   t1r(   r%   r&   s
          r   wrapper6_dcp_method_logger.<locals>.decorator.<locals>.wrapperK   s   $ $(D>(DV(DH
 !(HWB!V)7H%&h't.v. !&HWB#||~HV&(gH]#h'M ! !(3HW%+0'HW%'+||~HV$%%h/s   B0 0
C?:A C::C?)	functoolswrapsr   r   r   r
   )r(   r9   r%   r&   s   ` r   	decorator%_dcp_method_logger.<locals>.decoratorJ   sH    			277 	bii 	B 	 	 
	@ r   )r   r   r
   )r%   r&   r=   s   `` r   _dcp_method_loggerr?   E   s&    
"R( " "H r   rankc                 2   [         R                  [        R                  5        [        R                  " 5       nUR                  [        R                  5        [        R
                  " SU  S35      nUR                  U5        [         R                  U5        g )N[z6] %(asctime)s - %(name)s - %(levelname)s - %(message)s)loggersetLevelloggingINFOStreamHandler	FormattersetFormatter
addHandler)r@   ch	formatters      r   _init_loggerrM   q   si    
OOGLL!				 BKK!!
D6GHI OOI
br   )F)"r;   rE   r-   collections.abcr   typingr   r   typing_extensionsr   r   r   torch.distributed.c10d_loggerdistributedr"   -torch.distributed.checkpoint.logging_handlersr   	getLoggerrC   r	   listr   __annotations___get_or_create_loggerr3   r
   r   dictr   r#   boolr?   r   rM    r   r   <module>r[      s       $  '  3 3 I 
			 c  //@T]t_ tCH~  Fc3h  !))47)xB (2r6"223)Xs r   