U
    9%e                     @   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/Darija-Ai-API/env/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   

