o "’×i— ã @sæUddlZddlZddlmZmZmZmZmZddlm Z ddl m m Z ddl mZgZeeed<e  e¡aedƒZe dƒZdeeeffd d „Zdeeeffd d „Z ddededeeeefgeeefffdd„ZdS)éN)ÚAnyÚCallableÚDictÚListÚTypeVar)Ú ParamSpec)ÚDCP_LOGGER_NAMEÚ__all__Ú_TÚ_PÚreturncOs`i}| dd¡}| dd¡}| dd¡}|s"|p|}r"t|ddƒ}|dur*t|ƒn||d<|S)z0 Extracts log data from dcp method args Ústorage_writerNÚstorage_readerÚ checkpoint_id)ÚgetÚgetattrÚstr)ÚargsÚkwargsÚmsg_dictr rrÚ serializer©rúa/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/distributed/checkpoint/logger.pyÚ_msg_dict_from_dcp_method_argss    ÿrcOs*t|i|¤Ž}| tj|fi|¤Ž¡|S)N)rÚupdateÚ c10d_loggerÚ _get_msg_dict)Ú func_namerrrrrrr'srFÚlog_exceptionsÚwrapper_kwargsc s dtttff‡‡fdd„ }|S)zKThis method decorator logs the start, end, and exception of wrapped events.Úfunccs0t ˆ¡dtjdtjdtf‡‡‡fdd„ ƒ}|S)Nrrr c sÒtˆjg|¢Riiˆ¥|¥¤Ž}d|d<t ¡}||d<t |¡z ˆ|i|¤Ž}Wn#tyM}zˆrHd|d<|›|d<t ¡|d<t |¡‚d}~wwd|d<t ¡}t ¡|d<|||d<t |¡|S)NÚstartÚeventÚtimeÚ exceptionÚerrorÚendÚ times_spent)rÚ__name__r#Útime_nsÚ _dcp_loggerÚdebugÚ Exceptionr%)rrrÚt0Úresultr%Út1)r rrrrÚwrapper4s8ÿÿ ÿ    €ú    z6_dcp_method_logger..decorator..wrapper)Ú functoolsÚwrapsr rrr )r r0©rr)r rÚ decorator3s$z%_dcp_method_logger..decorator)rr r )rrr4rr3rÚ_dcp_method_logger.s#r5)F)r1r#ÚtypingrrrrrÚtyping_extensionsrÚtorch.distributed.c10d_loggerÚ distributedrÚ-torch.distributed.checkpoint.logging_handlersrr rÚ__annotations__Ú_get_or_create_loggerr*r r rrÚboolr5rrrrÚs(    ÿÿÿþ