Vh5 `UddlZddlZddlmZmZmZddlmZddlZddl m Z ddl m Z ddlmZgZeeed<dZefded ej*fd Zefded eej0effd Zead eeeffd Zed ZedZdeeefd eeeffdZdeeefd eeeffdZ y)N)AnyCallableTypeVar) ParamSpec) _log_handlers) _WaitCounter__all__default destinationreturnc t|\}}tjd|}|jtjtj d}|j |d|_|j||S)Nzc10d-za%(asctime)s %(filename)s:%(lineno)s %(levelname)s p:%(processName)s t:%(threadName)s: %(message)sF) _get_logging_handlerlogging getLoggersetLevelDEBUG Formatter setFormatter propagate addHandler)r logging_handlerlog_handler_namelogger formatters M/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/c10d_logger.py_get_or_create_loggerrsy(<[(I%O%   '7&89 :F OOGMM"!!kI  +F o& McPt|}t|jd|}||fS)N-)rtype__name__)r log_handlerrs rrr's7 ,K{+445Q{mD ) **rc <tjr|jdxs|jd}|tj|jdtj|tj tj |tj tj |d}|ddk(rCtjjj}djd|D|d <|Sd |i}|S) Ngroup process_grouppg) func_namepg_namebackend world_size group_size global_rank local_rankr)nccl.c32K|]}t|yw)N)str).0vs r z _get_msg_dict..As/M1A/Ms nccl_versionr') distis_initializedget_get_process_group_name get_backendget_world_sizeget_ranktorchcudar.versionjoin)r'argskwargsr$msg_dictr5s r _get_msg_dictrD3s  7#Bvzz/'B%;66vzz$7GHI**512!0023!0078"mmo.!]]512  I & ( ::??224L'*xx/M /M'MH^ $ O I;  Or_T_Pfuncctjdtjdtjdt ffd }|S)NrArBr c |i|S#t$r:}tjg|i|}||d<tj |d}~wwxYw)Nerror) ExceptionrDr! _c10d_loggerdebug)rArBrJrCrGs rwrapperz"_exception_logger..wrapperNs] (( ( $T]]DTDVDH#('HW    x (   s A5A  A functoolswrapsrFrArBrErGrNs` r_exception_loggerrSMsB__Trww")) Nrctjdtjdtjdt ffd }|S)NrArBr ctdjj5|i|}ddd|S#1swYSxYw)Nzpytorch.wait_counter.c10d.)rr!guard)rArB func_returnrGs rrNz_time_logger..wrapper\sK 6t}}oF G M M O 0//K 0 0s ;ArOrRs` r _time_loggerrX[sB__Trww")) Nr)!rPrtypingrrrtyping_extensionsrr=torch.distributed distributedr6"torch.distributed.logging_handlersr torch.monitorrr listr1__annotations___DEFAULT_DESTINATIONLoggerrtupleHandlerrrLdictrDrErFrSrXrrrgs))' <&c .B s gnn ,++ 7??C  +%& c3h, T]t_ HRV, "b&1A xB'HRV,<r