ȅi"SrSSKrSSKJrJrJr SSKJrJrJ r J r SSK J r SSK r SSKJr SSKJrJrJr SSKJr S S KJr S S KJr S S KJrJrJr S SKJr S SK J!r! S SK"J#r# S SK$J%r% S SK&J'r'J(r( \ (aSSK)J*r* \ "S5r+\ "S5r,S\-S\\\+\,4/\\+\,444Sjr."SS\(5r/"SS\'5r0g)a This module implements variable tracking for TorchScript objects during Dynamo tracing. The TorchScriptObjectVariable class provides specialized handling for TorchScript objects with strong safety guarantees by: - Enforcing method-call-only access to prevent unsafe attribute manipulation - Converting graph breaks into hard errors via _raise_hard_error_if_graph_break - Proper proxy and source tracking for TorchScript method calls - Integration with higher-order operators for method call handling Key safety features: - Strict validation that only method calls are allowed (no direct attribute access) - Immediate error reporting for potentially unsafe operations - Proper source tracking for debugging and guard installation - Safe handling of TorchScript object method calls through torchbind The module ensures that TorchScript objects are handled safely during tracing by limiting operations to known-safe patterns and failing fast for unsafe usage. N)CallableIterableSequence)AnyOptional TYPE_CHECKINGTypeVar) ParamSpec)Source)is_opaque_reference_typeis_opaque_typeis_opaque_value_type)Proxy)graph_break_hints) skip_code) unimplementedUnsafeScriptObjectError Unsupported)VariableTracker)ConstantVariable)ConstDictVariable) TupleVariable)UserDefinedObjectVariableUserDefinedVariable)InstructionTranslator_P_Treasonreturnc\S[[[4S[[[44SjnU$)Nfnr!c^[R"T5S[RS[RS[ 4U4Sjj5nU$)Nargskwargsr!cf>T"U0UD6$![an[UR5UeSnAff=fN)rrmsg)r%r&er#s _/home/james-whalen/.local/lib/python3.13/site-packages/torch/_dynamo/variables/script_object.pygraph_break_as_hard_errorQ_raise_hard_error_if_graph_break..deco..graph_break_as_hard_error8s9 <4*6** <-aee4!; .deco7sC   [TU]"S0UD6 Xlg)N)super__init__r7)selfr7r& __class__s r+r;"OpaqueObjectClassVariable.__init__Ls "6" r2cUR$r(r7r<s r+as_python_constant,OpaqueObjectClassVariable.as_python_constantP zzr2c>[[UR55$r()rtyper7rAs r+is_python_hashable,OpaqueObjectClassVariable.is_python_hashableSs#D$455r2cUR$r(r@rAs r+as_proxy"OpaqueObjectClassVariable.as_proxyVrDr2cNURRSURS3$)N())r=__name__r7rAs r+__repr__"OpaqueObjectClassVariable.__repr__Ys$..))*!DJJ^\rSrSr%0r\\S4\S'\S\ S\ 4Sj5r \ S\ S\S\SS4S j5rSS\ S\S \\S \SS 4 U4S jjjrS\ 4Sjr\"S5SSS\S\4U4Sjj5r\"S5SSS\S\\S \\\4S\4 Sj5rS\4U4SjjrSrU=r$)r]_fake_script_object_cacheuser_clsr!cZ[U[R5=(d [U5$r() issubclasstorch ScriptObjectr )clsrss r+is_matching_cls)TorchScriptObjectVariable.is_matching_clss(E$6$67S>(;SSr2proxyr7optionsc [X40UD6$r()r])r{r7r|s r+r^ TorchScriptObjectVariable.creates(AAAr2Nsourcer&c >[TU]"U40UD6 Xl[UR[R R 5(a"X RRRS'X0l g)N example_value) r:r;r{ isinstancervfxrnodemetar)r<r{r7rr&r=s r+r;"TorchScriptObjectVariable.__init__sP )&) djj%((.. 1 149JJOO  1 r2cUR$r()r{rAs r+rJ"TorchScriptObjectVariable.as_proxyrDr2zSSKJn SSKJn SSKJn [ [UR55(a[TU])X5nU$[URS5(aA[URR5(a[SS URS U3S S /S 9 [URUS5nUcO[SS URSU3SURSUS3SUSURS3/[ R"QS 9 [%U5(d[SS URSU3SS/S 9 URceUR'UU"URU5UUS9$)Nr)call_torchbindr) AttrSourcer) TorchHigherOrderOperatorVariablescript_class_namez9Attempted to access attributes/methods on an OpaqueObjectvalue=z, attr=z:Attribute/method access of OpaqueObjects is not supported.z?Use custom operators instead of direct attribute/method access.rUz.FakeScriptObject missing method implementation , method=zTorchScript object z doesn't define the method rTzEnsure the method z is implemented in z@Attempted to access non-callable attribute of TorchScript objectzWAttribute accesses of TorchScript objects to non-callable attributes are not supported.z-Use method calls instead of attribute access.)rscript_obj_var method_name)!torch._higher_order_ops.torchbindrrrhigher_order_opsrrrFr7r: var_getattrhasattrr rrgetattrr USER_ERRORcallablemake) r<rRrrrrresmethodr=s r+r%TorchScriptObjectVariable.var_getattrs E'F TZZ 0 1 1'%b/CJ 4::2 3 3 JJ ( (9 9  S  GD6:XU  T40 > H  IdV<1$**=XY]X^^_`(.A$**QO&11   Z  IdV<uC  {{&&&/44 dkk40 5  r2r%cF[SSURSU3SUS3S/S9 g)Nz'Weird method call on TorchScript objectrrzThis particular method call (zn) is not supported (e.g. calling `__setattr__`). Most method calls to TorchScript objects should be supported.zAvoid calling this method.rU)rr7)r<rRrr%r&s r+ call_method%TorchScriptObjectVariable.call_methods@ =TZZL $8/v6PP- r2c~>[[UR55(a UR$[TU]5$r()rrFr7r:rB)r<r=s r+rB,TorchScriptObjectVariable.as_python_constants/ TZZ 0 1 1:: w)++r2)r{rr()rOrfrgrhrrrlint__annotations__ classmethodrFrjry staticmethodrrr^rr r;rJr3rkrrrrrBrmrnros@r+r]r]sZHJtC)D$DEJTtTTTBeBCBCBrs(8888'  ! "EE!&$ HDt_ T]    xB (2r6"223  :H 3:Hzk, 9k,r2