
    [Th                     \   % S SK r 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	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)AnyCallableTypeVar)	ParamSpec)uuid4)DCP_LOGGER_NAME__all___T_Preturnc                     0 nUR                  SS5      nUR                  SS5      nUR                  SS5      nUR                  S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_writerNstorage_readerplannercheckpoint_iduuid)getgetattrstrr   int	__class____name__)argskwargsmsg_dictr   r   r   r   
serializers           [/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/checkpoint/logger.py_msg_dict_from_dcp_method_argsr      s     H ZZ 0$7NZZ 0$7NjjD)GJJ5M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>.wrapperH   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>.decoratorG   sH    			277 	bii 	B 	 	 
	@ r   )r   r   r
   )r%   r&   r=   s   `` r   _dcp_method_loggerr?   B   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   n   si    
OOGLL!				 BKK!!
D6GHI OOI
br   )F)!r;   rE   r-   typingr   r   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>rZ      s       ) ) '  3 3 I 
			 c  //@T]t_tCH~ Dc3h  !))47)xB (2r6"223)Xs r   