U
    yh(	                     @   s   d dl Z d dlZ d dlZd dlmZ d dlmZ dddiddidddd	d
ddddd
dddidddgddddgdddgdddddZe	ee j
 dddZdS )    N)Optional   formatz%%(name)s: [%(levelname)s] %(message)s)spmd_formatgraph_opt_formatzlogging.StreamHandlerDEBUGr   zext://sys.stdout)classlevel	formatterstreamr   r   zlogging.NullHandler)spmd_consolegraph_opt_consolenull_consoler   F)r	   handlers	propagater   r   )r   r   )Zspmd_expZ	graph_optnull_logger)version
formattersr   loggersdisable_existing_loggers)log_typereturnc                 C   s   ddl m} dtjkrtjt ttd  }| |ksNt	d|  d| t
 s`t| S t
 dkrt| }||j |jd k	rt|j}||j || n
td}|S tdS )Nr   )configZPYTEST_CURRENT_TESTr   zUnable to find z" in the available list of loggers r   )Ztorch.distributed._spmdr   osenvironlogging
dictConfigLOGGING_CONFIGlistkeysAssertionErrordistZis_initialized	getLoggerZget_ranksetLevelZ	log_levelZlog_file_nameFileHandler
addHandler)r   r   Zavail_loggersloggerZlog_file r'   S/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/_spmd/log_utils.py
get_logger5   s(    




r)   )r   logging.configr   typingr   Ztorch.distributeddistributedr!   r   strLoggerr)   r'   r'   r'   r(   <module>   sJ    ,