o
    !i                     @   s  U d dl Z d dlmZ d dlmZmZmZ ddlmZ g dZ	i a
eeef ed< daedd	ed
d Zedd	dee fddZedd	dd Zedd	dd Zedd	dee fddZedd	defddZedd	edd Zedd	deeef fddZdS )    N)contextmanager)ListAnyDict   )compatibility)preserve_node_metahas_preserved_node_metaset_stack_traceset_grad_fn_seq_nrreset_grad_fn_seq_nrformat_stackset_current_metaget_current_metacurrent_metaF)Zis_backward_compatiblec                  c   s"    t } z
da d V  W | a d S | a w )NTshould_preserve_node_meta)Zsaved_should_preserve_node_meta r   a/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/fx/traceback.pyr      s   r   stackc                 C   s"   t r| rd| td< d S d S d S )N stack_trace)r   joinr   )r   r   r   r   r
      s   r
   c                 C   s<   t rtdd td< tdd td< | td< dtd< d S d S )Ngrad_fn_seq_nrprev_grad_fn_seq_nr
in_grad_fnprev_in_grad_fnT)r   r   get)Zseq_nrr   r   r   r   #   s   r   c                   C   sL   t r$td d u rtd d u sJ td= td= td td< td td< d S d S )Nr   r   r   r   )r   r   r   r   r   r   r   /   s   r   returnc                   C   s(   t r	tddgS tt d d S )Nr   r   )r   r   r   	tracebackformat_listextract_stackr   r   r   r   r   >   s   r   c                   C      t S Nr   r   r   r   r   r	   G      r	   c                 c   s    t r=| jr=t}z1| j adtvr| j| jfgtd< ntd d d | jkr2td | j| jf d V  W |ad S |aw d V  d S )NZ	from_noder   r   )r   metar   copynametargetappend)nodeZ
saved_metar   r   r   r   L   s   


r   c                   C   r#   r$   )r   r   r   r   r   r   b   r%   r   )r    
contextlibr   typingr   r   r   Z_compatibilityr   __all__r   str__annotations__r   r   r
   r   r   r   boolr	   r   r   r   r   r   r   <module>   s2   
 

