o i @sdZddlZddlmZddlmZmZddlmZedZ edZ e dd ee e fd e d ee e ffd d Z e dd dd e d eee e fgee e fffdd Z dd ee e fdBd e d ee e feee e fgee e ffBfdd Z dS)za APIs related to torch.compile which lazily import torch._dynamo to avoid circular dependencies. N)Callable)overloadTypeVar) ParamSpec_T_PTfn recursivereturncCdSNrr r r P/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/_compile.py_disable_dynamorcCr r r rr r rrrcsDdurtdtjdtjdtffdd }|StjtdS)au This API should be only used inside torch, external users should still use torch._dynamo.disable. The main goal of this API is to avoid circular imports issues that is common while using _dynamo.disable inside torch itself. This API avoids it by lazily importing torch._dynamo from the import time to the invocation of the decorated function. Nargskwargsr csBtdd}|durddl}|jjdd}|_||i|S)N__dynamo_disablerF)wrapping)getattr torch._dynamo_dynamodisabler)rr disable_fntorchrr rinner*s z_disable_dynamo..inner)r ) functoolswrapsrrrrpartialr)rr rr rrrs  " )T)NT) __doc__rcollections.abcrtypingrrtyping_extensionsrrrboolrr r r rsB      (