}Yh |UddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m cm Z ddlmZejZgZeeed<e jeaedZedZd eeeffd Zd eeeffd Z dd eded eeeefgeeefffdZdefdZ dS)N)AnyCallableTypeVar) ParamSpec)uuid4)DCP_LOGGER_NAME__all___T_Preturnci}|dd}|dd}|dd}|dd}|s|p|x}rt|dd}|t|n||d<ttj|d<|r|jj|d<|r|jj|d<|r|jj|d<|S)z0 Extracts log data from dcp method args storage_writerNstorage_readerplanner checkpoint_iduuid)getgetattrstrrint __class____name__)argskwargsmsg_dictrrrr serializers u/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/distributed/checkpoint/logger.py_msg_dict_from_dcp_method_argsrsHZZ 0$77NZZ 0$77NjjD))GJJ55M CN,LnLjC OTBB ,7M] _ 577;''HVG%3%=%F!"G%3%=%F!"9%/8 Ocnt|i|}|tj|g|Ri||S)N)rupdate c10d_logger _get_msg_dict) func_namerrrs rr#r#;sF-t>v>>H OOK-iI$III&IIJJJ OrFlog_exceptionswrapper_kwargsc Jdtttfffd }|S)zKThis method decorator logs the start, end, and exception of wrapped events.funcctjdtjdtjdt ffd }|S)Nrrr ctjg|Rii |}d|d<tj}||d<|d<t| |i|}nP#t $rC}r;d|d<||d<tj|d<t|d}~wwxYwd|d<tj}tj|d<||z |d<t||S) Nstarteventtimer% exceptionerrorend times_spent)r#rr-time_ns _dcp_loggerdebug BaseExceptionr/) rrrt0resultr/t1r(r%r&s rwrapperz6_dcp_method_logger..decorator..wrapperHsH$  $(D>(DV(DH !(HW B!HV )7H% &   h ' ' ' t.v..    !0(3HW%+0 HW%'+|~~HV$%%h///  !&HW B#|~~HV &(2gH] #   h ' ' 'MsA!! B.+>B))B.) functoolswrapsr rrr )r(r9r%r&s` r decoratorz%_dcp_method_logger..decoratorGsb    27 bi B          @r)rr r )r%r&r<s`` r_dcp_method_loggerr=Bs@ "R("""""""H rrankc@ttjtj}|tjtjd|d}||t|dS)N[z6] %(asctime)s - %(name)s - %(levelname)s - %(message)s)loggersetLevelloggingINFO StreamHandler Formatter setFormatter addHandler)r>ch formatters r _init_loggerrKns OOGL!!!   BKK !HDHHHIOOI br)F)!r:rCr-typingrrrtyping_extensionsrrrtorch.distributed.c10d_logger distributedr"-torch.distributed.checkpoint.logging_handlersr getLoggerrAr listr__annotations___get_or_create_loggerr3r r dictrr#boolr=rrKrrrXs ))))))))))''''''333333333IIIIII    c0k/@@  WT]]Yt__tCH~Dc3h!)))47) xB (2r6"223))))Xsr