
    IЦi                     ,   % S SK r S SKrS SKJrJr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\\   \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g)    N)AnyCallableDictList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           b/var/www/html/ai-image-ml/venv/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%   7   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BaseExceptionr1   )
r   r   r   t0resultr1   t1r*   r'   r(   s
          r   wrapper6_dcp_method_logger.<locals>.decorator.<locals>.wrapperD   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*   r;   r'   r(   s   ` r   	decorator%_dcp_method_logger.<locals>.decoratorC   sH    			277 	bii 	B 	 	 
	@ r!   )r   r   r   )r'   r(   r?   s   `` r   _dcp_method_loggerrA   >   s&    
"R( " "H r!   )F)r=   r/   typingr   r   r   r   r   typing_extensionsr   r   r	   torch.distributed.c10d_loggerdistributedr$   -torch.distributed.checkpoint.logging_handlersr
   r   r   __annotations___get_or_create_loggerr5   r   r   r    r%   boolrA    r!   r   <module>rK      s      5 5 '  3 3 I c  //@T]t_tCH~ Dc3h  !))47)xB (2r6"223)r!   