o Zh@s\UddlZddlmZmZerddlZgaeeedgdfed<ej ddZ ddZ dS) N)Callable TYPE_CHECKINGz torch.TensorINTERMEDIATE_HOOKSccs,t|z dVWtdStwN)rappendpop)fnr D/var/www/auris/lib/python3.10/site-packages/torch/_inductor/hooks.pyintermediate_hook s  r cCs.t}gaz|D]}|||qW|adS|awr)r)namevalhookshookr r r run_intermediate_hookss r) contextlibtypingrrZtorchrliststr__annotations__contextmanagerr rr r r r s