
    קg<                         d dl Z d dlZd dlmZmZ dZd Z e j        d          Zes#	 d dl	Z	dZ
n# e$ r dZ
Y nw xY w ee
dd 	          Zd
 ZdS )    N)	_FaketqdmtqdmTc                  x    t          j        d          t          j        d          t          j        d          gS )Nz%torch.fx.experimental.symbolic_shapesztorch._dynamoztorch._inductor)logging	getLogger     Q/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_dynamo/logging.pyget_loggersr      s8    ABB/**+,, r	            ztorch.compile())totaldescdelayc                      t           sTt                              d           t          t          t                    s t                               j                    t          t                     fd}|S )Nr   c                 *     j         | d|fi | d S )NzStep %s: %s)log)levelmsgkwargsloggersteps      r
   r   zget_step_logger.<locals>.log8   s)    
5-s==f=====r	   )	disable_progresspbarupdate
isinstancer   set_postfix_strnamenext_step_counter)r   r   r   s   ` @r
   get_step_loggerr"   0   sx     3A$	** 	3  FK!1222D> > > > > > Jr	   )	itertoolsr   	torch.hubr   r   r   r   countr!   triton	num_stepsImportErrorr   r"   r   r	   r
   <module>r)      s         % % % % % % % %    &  	""  B		   			4i&7qAAAD    s   0 ::