o
    k9:jF  ã                   @   sL   d dl mZ d dlZG dd„ dƒZdejjdejjejjB defdd	„ZdS )
é    )ÚTracebackTypeNc                   @   sf   e Zd ZdejjdejjejjB ddfdd„Zddd„Z	d	e
e dB d
edB dedB ddfdd„ZdS )Ú_InsertPointÚinsert_point_graphÚinsert_pointÚreturnNc                 C   s   || _ || _d | _d S ©N)r   ÚgÚguard)Úselfr   r   © r   úZ/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/jit/_ir_utils.pyÚ__init__   s   
z_InsertPoint.__init__c                 C   s   | j  ¡ | _| j  | j¡ d S r   )r   ÚinsertPointÚprev_insert_pointÚsetInsertPointr   )r
   r   r   r   Ú	__enter__   s   z_InsertPoint.__enter__Úexc_typeÚexc_valÚexc_tbc                 C   s   | j  | j¡ d S r   )r   r   r   )r
   r   r   r   r   r   r   Ú__exit__   s   z_InsertPoint.__exit__)r   N)Ú__name__Ú
__module__Ú__qualname__ÚtorchÚ_CÚGraphÚNodeÚBlockr   r   ÚtypeÚBaseExceptionr   r   r   r   r   r   r      s$    þý
ü
	
þýüûr   r
   r   r   c                 C   s
   t | |ƒS r   )r   )r
   r   r   r   r   Úinsert_point_guard   s   
r    )	Útypesr   r   r   r   r   r   r   r    r   r   r   r   Ú<module>   s    ÿÿþ