a
    d>                     @   s   d dl Z d dlZd dlZd dlmZmZ de_eejd dZdd Z	dd	 Z
d
d ZddddiiddddddiddgddddgdddddZd!ddZe dZeszd dlZdZW n ey   dZY n0 eedd dZdd  ZdS )"    N)	_Faketqdmtqdm   CODETc                   C   s   t dt dgS )Ntorch._dynamotorch._inductor)logging	getLogger r
   r
   ^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_dynamo/logging.pyget_loggers   s    r   c                 C   s   t  D ]}||  qdS )zWrite current log levelN)r   setLevel)levelloggerr
   r
   r   set_loggers_level   s    
r   c                   C   s   t  d jS )zRead current log levelr   )r   r   r
   r
   r
   r   get_loggers_level   s    r      Ztorchdynamo_formatformatz3[%(asctime)s] %(name)s: [%(levelname)s] %(message)sZtorchdynamo_consolezlogging.StreamHandlerDEBUGzext://sys.stderr)classr   	formatterstreamF)r   handlers	propagate)r   r   )version
formattersr   loggersdisable_existing_loggersc                 C   s   dt jvrtjt |d urHt|}||  t D ]}|	| q8t
t jddrddlm} tj} t j| d}t j|st | tt j|d}|tj td}|	| t|  d S )	NZPYTEST_CURRENT_TESTZTORCH_COMPILE_DEBUGFr   )get_debug_dirZtorchdynamoz	debug.logr   )osenvironr   config
dictConfigLOGGING_CONFIGFileHandlerr   r   
addHandlerboolgetutilsr   r   pathjoinexistsmakedirsr	   r   )Z	log_levelZlog_file_nameZlog_filer   r   Zlog_pathr
   r
   r   init_loggingB   s$    






r-         ztorch.compile())totaldescdelayc                    s@   t s&td ttts&t j  tt fdd}|S )Nr   c                    s     | d d|  d S )NzStep z: )log)r   msgr   stepr
   r   r3   ~   s    zget_step_logger.<locals>.log)	disable_progresspbarupdate
isinstancer   Zset_postfix_strnamenext_step_counter)r   r3   r
   r5   r   get_step_loggerv   s    

r>   )N)	itertoolsr   r   Z	torch.hubr   r   r   addLevelNamer7   r   r   r   r#   r-   countr=   ZtritonZ	num_stepsImportErrorr8   r>   r
   r
   r
   r   <module>   sT   
 
$

