tj3 UddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m cmZddlmZejZgZeeed<ejeaedZe d Zd eeeffd Zd eeeffd Z ddeded eeeefgeeefffdZde fdZ!dS)N)Callable)AnyTypeVar) ParamSpec)uuid4)DCP_LOGGER_NAME__all___T_Preturnci}|d}|d}|d}|d}|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_writerstorage_readerplanner checkpoint_idNuuid)getgetattrstrrint __class____name__)argskwargsmsg_dictrrrr serializers a/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/distributed/checkpoint/logger.py_msg_dict_from_dcp_method_argsrs HZZ 011NZZ 011Njj##GJJ//M 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..wrapperKsH$  $(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..decoratorJsb    27 bi B          @r)rr r )r%r&r<s`` r_dcp_method_loggerr=Es@ "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_loggerrKqs OOGL!!!   BKK !HDHHHIOOI br)F)"r:rCr-collections.abcrtypingrrtyping_extensionsrrrtorch.distributed.c10d_logger distributedr"-torch.distributed.checkpoint.logging_handlersr getLoggerrAr listr__annotations___get_or_create_loggerr3r r dictrr#boolr=rrKrrrYs $$$$$$''''''333333333IIIIII    c0k/@@  WT]]Yt__ tCH~    Fc3h!)))47) xB (2r6"223))))Xsr