U
    ªÊ¹j  ã                   @   s~   d dl Z d dlZd dlmZmZ dZdd„ Ze  d¡Zesrzd dl	Z	dZ
W n ek
rb   dZ
Y nX ee
d	d d
�Zdd„ ZdS )é    N)Ú	_FaketqdmÚtqdmTc                   C   s   t  d¡t  d¡t  d¡gS )Nz%torch.fx.experimental.symbolic_shapesztorch._dynamoztorch._inductor)ÚloggingÚ	getLogger© r   r   úT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_dynamo/logging.pyÚget_loggers   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 S )NzStep %s: %s)Úlog)ÚlevelÚmsg©ÚloggerÚstepr   r   r   6   s    zget_step_logger.<locals>.log)	Údisable_progressÚpbarÚupdateÚ
isinstancer   Zset_postfix_strÚnameÚnextÚ_step_counter)r   r   r   r   r   Úget_step_logger.   s    

r   )Ú	itertoolsr   Z	torch.hubr   r   r   r   Úcountr   ZtritonZ	num_stepsÚImportErrorr   r   r   r   r   r   Ú<module>   s   

