U
    9%e                     @   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 dddd	d
dddgZ	i a
eeef ed< daeddedd Zeddee dddZedddd	 Zedddd
 Zeddee dddZeddedddZeddedd Zeddeeef d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 5 | a X d S )NTshould_preserve_node_meta)Zsaved_should_preserve_node_meta r   Q/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/fx/traceback.pyr      s
    
stackc                 C   s   t r| rd| td< d S )N stack_trace)r   joinr   r   r   r   r   r
      s    c                 C   s8   t r4tdd td< tdd td< | td< dtd< 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
    c                   C   sH   t rDtd d kr,td d ks ttd= td= td td< td td< d S )Nr   r   r   r   )r   r   AssertionErrorr   r   r   r   r   /   s    )returnc                   C   s,   t rtddgS tt d d S d S )Nr   r   )r   r   r   	tracebackformat_listextract_stackr   r   r   r   r   >   s    c                   C   s   t S Nr   r   r   r   r   r	   G   s    c                 c   s~   t rt| jrtt}z\| j adtkr6| j| jfgtd< n,td d d | jkrbtd | j| jf d V  W 5 |aX nd V  d S )NZ	from_noder!   r   )r   metar   copynametargetappend)nodeZ
saved_metar   r   r   r   L   s    


c                   C   s   t S r%   )r   r   r   r   r   r   b   s    )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>   s@        

