h dZddlZddlmZmZmZmZmZmZddl m Z edZ e dZ e ddee e fd e d ee e ffd Ze ddedd e d eee e fgee e fffd Z ddeee e fd e d eee e feee e fgee e ffffd ZdS)za APIs related to torch.compile which lazily import torch._dynamo to avoid circular dependencies. N)CallableLiteralOptionaloverloadTypeVarUnion) ParamSpec_T_PTfn recursivereturncdSNr r s V/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/torch/_compile.py_disable_dynamors scdSrrrs rrrs 69SrcBtjdtjdtjdt ffd }|StjtS)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. Nargskwargsrctdd}|(ddl}|jd}|_||i|S)N__dynamo_disablerF)wrapping)getattr torch._dynamo_dynamodisabler)rr disable_fntorchr r s rinnerz_disable_dynamo..inner)sd!%7>>J!$$$$#]222y52QQ &0#:t.v.. .r)r ) functoolswrapsr rrr partialr)r r r#s`` rrrs{ ~    / /BI /" / / / / / /   /  IFFFFr)T)NT)__doc__r$typingrrrrrrtyping_extensionsr r r boolrrrrr+s HHHHHHHHHHHHHHHH'''''' WT]]Yt__ ,0R%) b"f   0499 9)-9 xB (2r6"223999 9 >B G G"b&!" G6: G 8BF XxB'7&8(2r6:J&JK KL G G G G G Gr