o
    ߗi                     @   s`   U d dl Z d dlmZmZmZ erd dlZg aeeedgdf  ed< e j	dd Z
dd ZdS )	    N)CallableListTYPE_CHECKINGztorch.TensorINTERMEDIATE_HOOKSc                 c   s,    t |  z
d V  W t   d S t   w N)r   appendpop)fn r
   S/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_inductor/hooks.pyintermediate_hook   s
   
r   c                 C   s.   t }g a z|D ]}|| | qW |a d S |a w r   )r   )namevalhookshookr
   r
   r   run_intermediate_hooks   s   r   )
contextlibtypingr   r   r   torchr   str__annotations__contextmanagerr   r   r
   r
   r
   r   <module>   s   

