o !iI @sUddlZddlZddlZddlmZmZddlmZddlm Z gZ ee e d<dejfddZdd e deeje ffd d Zead d ZddZdS)N)ListTuple) _log_handlers__all__returncCsNt\}}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)Zlogging_handlerlog_handler_namelogger formatterrl/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/distributed/c10d_logger.py_get_or_create_loggers    rdefault destinationcCst|}t|j}||fS)N)rtype__name__)rZ log_handlerrrrrr!s rctfdd}|S)Nc sz|i|WStyW}zCtr?j|d|t|dt|dtt|d|d}nj|d||d}t |d}~ww)N, group) func_nameargsbackend world_size global_rank local_rankerror)rrr#) Exceptiondistis_initializedr get_backendgetget_world_sizeget_rank _c10d_loggerdebug)rkwargsr#Zerror_msg_dictfuncrrwrapper,s(   z"_exception_logger..wrapper functoolswrapsr/r0rr.r_exception_logger+sr5cr)Nc st}|i|}t}trBj|d|t|dt|dtt|d||dd}nj|d|||d}t ||S)Nrrns)rrrr r!r" time_spent)rrr7) timetime_nsr%r&rr'r(r)r*r+r,)rr-t1Z func_returnt2Zmsg_dictr.rrr0Gs$    z_time_logger..wrapperr1r4rr.r _time_loggerFsr<)r)r2rr8typingrrZ"torch.distributed.logging_handlersrZtorch.distributed distributedr%rstr__annotations__LoggerrHandlerrr+r5r<rrrrs