
    Wj                         U d dl Z d dlmZ d dlmZ erd dlZg aeeedgdf                  e	d<   e j
        d             Zd ZdS )    N)Callable)TYPE_CHECKINGztorch.TensorINTERMEDIATE_HOOKSc              #      K   t                               |            	 d V  t                                            d S # t                                            w xY wN)r   appendpop)fns    Z/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torch/_inductor/hooks.pyintermediate_hookr      sY      b!!!!         s	   = Ac                 P    t           }g a 	 |D ]} || |           	 |a d S # |a w xY wr   )r   )namevalhookshooks       r   run_intermediate_hooksr      sW    E# 	 	DDsOOOO	 #U""""s   ! %)
contextlibcollections.abcr   typingr   torchr   liststr__annotations__contextmanagerr   r        r   <module>r      s         $ $ $ $ $ $              LLL CE D3"7"=>? D D D ! ! !# # # # #r   