o
    iv                     @   s`   d dl mZ d dlmZmZ d dlZG dd dZdejjdeejj	ejj
f defd	d
ZdS )    )TracebackType)OptionalUnionNc                   @   sf   e Zd Zdejjdeejjejjf fddZ	dddZ
d	eee  d
ee dee ddfddZdS )_InsertPointinsert_point_graphinsert_pointc                 C   s   || _ || _d | _d S N)r   gguard)selfr   r    r   P/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/jit/_ir_utils.py__init__   s   
z_InsertPoint.__init__returnNc                 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Graphr   NodeBlockr   r   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   typingr   r   r   r   r   r   r   r   r"   r   r   r   r   <module>   s    