o
    ߗi                     @   s   d dl Z d dlZd dlmZmZmZ d dlmZmZ dZ	deej
 fddZe dZe	sHzd dlZd	ZW n ey@   d
ZY nw eedd dZdej
ded fddZdS )    N)AnyCallableList)	_FaketqdmtqdmTreturnc                   C   s   t dt dt dgS )Nz%torch.fx.experimental.symbolic_shapesztorch._dynamoztorch._inductor)logging	getLogger r
   r
   S/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_dynamo/logging.pyget_loggers   s   r            ztorch.compile())totaldescdelaylogger).Nc                    sR   t std tttst j  ttdt	dt
dtdd f fdd}|S )Nr   levelmsgkwargsr   c                    s,   d|vrd|d<  j | d|fi | d S )N
stacklevelr   zStep %s: %s)log)r   r   r   r   stepr
   r   r   8   s   zget_step_logger.<locals>.log)disable_progresspbarupdate
isinstancer   set_postfix_strnamenext_step_counterintstrr   )r   r   r
   r   r   get_step_logger0   s   

 r%   )	itertoolsr   typingr   r   r   	torch.hubr   r   r   Loggerr   countr"   triton	num_stepsImportErrorr   r%   r
   r
   r
   r   <module>   s     
