a h5 @sUddlZddlZddlmZmZmZddlmZddlZddl m Z ddl m Z ddlmZgZeeed<dZefeejdd d Zefeeejefdd d Zeaeeefd ddZedZedZeeefeeefdddZeeefeeefdddZ dS)N)AnyCallableTypeVar) ParamSpec) _log_handlers) _WaitCounter__all__default) destinationreturncCsPt|\}}td|}|tjtd}||d|_|||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 Zlogging_handlerlog_handler_namelogger formatterrK/var/www/auris/lib/python3.9/site-packages/torch/distributed/c10d_logger.py_get_or_create_loggers    rcCs$t|}t|jd|}||fS)N-)rtype__name__)r Z log_handlerrrrrr 'sr )r c Ostr|dp|d}|t|dt|tt|tt|d}|ddkrtjj }d dd |D|d <n d |i}|S) NgroupZ process_groupZpg) func_nameZpg_namebackendZ world_sizeZ group_sizeZ global_rankZ local_rankr nccl.css|]}t|VqdS)N)str).0vrrr Az _get_msg_dict.. nccl_versionr) distZis_initializedgetZ_get_process_group_name get_backendZget_world_sizeZget_ranktorchZcudar!versionjoin)rargskwargsrmsg_dictr(rrr _get_msg_dict3s      r2_T_P)funcr cs(ttjtjtdfdd }|S)Nr/r0r c shz|i|WStyb}z:tjg|Ri|}||d<t|WYd}~n d}~00dS)Nerror) Exceptionr2r _c10d_loggerdebug)r/r0r7r1r5rrwrapperNs  z"_exception_logger..wrapper functoolswrapsr4r/r0r3r5r<rr;r_exception_loggerMs rAcs(ttjtjtdfdd }|S)Nr6csFtdj|i|}Wdn1s80Y|S)Nzpytorch.wait_counter.c10d.)rrZguard)r/r0Z func_returnr;rrr<\s,z_time_logger..wrapperr=r@rr;r _time_logger[srB)!r>r typingrrrZtyping_extensionsrr,Ztorch.distributedZ distributedr)Z"torch.distributed.logging_handlersrZ torch.monitorrrlistr#__annotations__Z_DEFAULT_DESTINATIONLoggerrtupleHandlerr r9dictr2r3r4rArBrrrr s*