VhD 6dZddlZddlmZmZmZmZmZmZddl m Z edZ e dZ e d dee e fde d ee e ffd Ze ddedde d eee e fgee e fffd Z ddeee e fde d eee e feee e fgee e ffffd Zy)za APIs related to torch.compile which lazily import torch._dynamo to avoid circular dependencies. N)CallableLiteralOptionaloverloadTypeVarUnion) ParamSpec_T_Pfn recursivereturncyNr r s >/home/dcms/DCMS/lib/python3.12/site-packages/torch/_compile.py_disable_dynamorscyrrrs rrrs69rcGtjdtjdtjdt ffd }|Stj tS)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. argskwargsrctdd}|'ddl}|jj}|_||i|S)N__dynamo_disabler)getattr torch._dynamo_dynamodisabler)rr disable_fntorchr r s rinnerz_disable_dynamo..inner)sK!%7>J!$"]]222yA &0#t.v. .r)r ) functoolswrapsr rrr partialr)r r r"s`` rrrs^ ~   / /BII /" /  /   IFFr)T)NT)__doc__r#typingrrrrrrtyping_extensionsr r r boolrrrrr*s& HH' T]t_ ,0R%) b"f   049 9)-9 xB (2r6"2239 9 >BG"b&!"G6:G 8BF XxB'7&8(2r6:J&JK KLGr