+ jB:a0t$Rt^RIHt^RIt^RIt^RIt^RIt^RIt^RI t ^RI H t H t H t Ht^RIHt^RIHtHt^RIHtHtHtHtHt^RIHtHtHtHtHtH t ^RI!H"t"H#t#^R I$H$t$^R I%H&t&H't'H(t(H)t)H*t*^R I+H,t,H-t-^R I.H/t/H0t0^R I1H2t2H3t3]4!0ER-m4t5]'d+^RI6H7t7^RI8H9t9^RI:H;t;^RIH?t?^RI@HAtA^RIBHCtC]P!]E4tF]P!RR7!RR44tH]!RRR 7tIR!]JR&!R"R#]4tK!R$R%4tL!R&R']4tM!R(R)]L4tN!R*R+]L4tO!R,R-]L4tP!R.R/]P4tQ!R0R1]P4tRR2R3ltSR4R5ltTR6R7ltU!R8R94tV!R:R;]4tWR<R=ltXER.wtYtZt[t\t]]PR>R?l4t_R@RAlt`]P!RRB7!RCRD44ta]P!RRB7!RERF44tb]P!RRB7!RGRH44tcRIRJltdRKRLlteRMRNltf!RORP]g4thRQRRltiRSRTltjRURVltkRWRXltlRYRZltmR[R\ltnR]R^ltoR_R`ltpRaRbltqRcRdltrReRfltsRgRhlttRiRjltu]u!R4tv]u!R4tw]u!R4tx]u!R4ty]u!R4tz]u!R4t{RkRllt|RmRnlt}RoRplt~RqRrltRsRtltRuRvltRwRxltRyRzlt!R{R|]4t]P!RR7!R}R~44t!RR]4t!RR]4t!RR]4t!RR]4t];QJd.R]EP4FNK 5M!R]EP44t]EP]EPUu/uFp]!]VEP44VbK up]EP ]EPUu/uFp]!]VEP44VbK up]EP"]EPUu/uFp]!]VEP44VbK up]EP$]Uu/uFp]!] VEP44VbK up/tR]JR&]EP]EPUu/uFqV R2bK up]EP ]EPUu/uFqV R2bK up]EP"]EPUu/uFqV R2bK up]EP$/]Uu/uFqV R2bK upCRRRR/C/tR]JR&]EP+4FEwtt]!]4]!]],EP3448wgK1]!]EP6 R24h RRlt]PRRl4t] RR] ER/,]R,]]R3,.R3,tR]JR&]P!RR7!RR44tRRltRRltRRltRRlt.]!RR] EPN]i4N]!RR] EPP]|4N]!RR] EPR]i4N]!RR] EPT]4N]!RR] EPT]4N]!RR] EPV]v4N]!RR] EPV]w4N]!RR] EPV]x4N]!RR] EPV]y4N]!RR] EPV]z4N]!RR] EPV]{4N]!RR] EPX]i4N]!RR] EPZ]i4N]!RR] EP\]4N]!RR] EP^]4N]!RR] EP^]4N]!RR] EP`]}4N]!RR]EPb]j4N]!RR]EPb]k4N]!RR]EPd]j4N]!RR]EPd]k4N]!RR]EPf]j4N]!RR]EPf]k4N]!RR]EPh]j4N]!RR]EPh]k4N]!RR]EPj]j4N]!RR]EPj]k4N]!RR]EPl]l4N]!RR]EPl]m4N]!RR]EPn]i4N]!RR]EPp]i4N]!RR]EPr]i4N]!RR]EPt]i4N]!RR]EPv]i4N]!RR]EPx]j4N]!RR]EPx]k4N]!RR]EPz]j4N]!RR]EPz]k4N]!RR]EP|]j4N]!RR]EP|]k4N]!RR]EP~]j4N]!RR]EP~]k4N]!RR]EP]j4N]!RR]EP]k4N]!RR]EP]i4N]!RR]EP]i4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]j4N]!RR]EP]k4N]!RR]EP]j4N]!ERR]EP]k4N]!ERER]EP]j4N]!ERER]EP]j4N]!ERER]EP]i4N]!ERER]EP]j4N]!ER ER]EP]k4N]!ER ER ]EP]j4N]!ER ER ]EP]i4N]!ERER]EP]j4N]!ERER]EP]q4N]!ERER]EP]r4N]!ER ER]EP]i4N]!RER]EP]j4N]!RER]EP]p4N]!RER]EP]p4N]!ERER]EP]o4N]!ERER]EP]n4N]!ERER]EP]s4N]!ERER]EP]t4N]!RER]EP]j4N]!RER]EP]p4NtER]JER&]PERERl4t!ER ER!]4t!ER"ER#]4t!ER$R]ER%7tER&ER'lt]EPtER(tER)ER*lt^ER+I!Ht^ER,IHtR#uupiuupiuupiuupiuupiuupiuupiuupi(0a Core variable tracking functionality for Dynamo. This module defines the fundamental classes and systems used to track and manage variables during Dynamo's operation. The module provides: 1. VariableTracker - The base class for tracking variables during compilation 2. MutationType system - Classes for tracking and managing mutations to variables 3. Source type management - Utilities for tracking variable origins and scope 4. Variable state management - Tools for managing variable state and transformations These components form the foundation of Dynamo's variable handling system, enabling accurate tracking and transformation of Python code into optimized computations. ) annotationsN)Callable ItemsViewKeysView ValuesView) ContextVar)EnumIntFlag)Any NamedTupleNoReturn TYPE_CHECKING TypeAlias)get_type_slotshas_slotPyMappingSlots PyNumberSlotsPySequenceSlots PyTypeSlots)graph_break_hints variables)current_scope_id)ObservedAttributeErrorraise_observed_exceptionraise_type_error unimplemented Unsupported) GuardBuilder install_guard) AttrSourceSource)format_source_rangeistype__eq____ne____lt____le____gt____ge__)Guard)Node) PyCodegen) SideEffects)InstructionTranslatorBase)ConstantVariable)UserFunctionVariableT)frozencf]tRt^Gt$RtR]R&R]R&RtR]R&RtR]R &RtR]R &R R lt R t R#)SourceLocationzMSource position of the bytecode instruction that generated a VariableTracker.strfilenameintlinenoN int | None end_lineno col_offsetend_col_offsetcV^8dQhRR/#returnr3)formats"]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/variables/base.py __annotate__SourceLocation.__annotate__Qs   c :RVP RVP R2p\VPVPVPVPVP 4pV'd5V\ P!VPR4R4R,, pV#)z File "z", line  z ) r4r6r!r8r9r:textwrapindentrstrip)selfresultsources& rAr@SourceLocation.formatQszDMM?(4;;-rB$ MM KK OO OO       hoofmmD&96BTI IF rDr?) __name__ __module__ __qualname____firstlineno____doc____annotations__r8r9r:r@__static_attributes__r?rDrAr2r2Gs7WM K!J !!J !!%NJ%  rDr2_vt_active_calls)defaultz'ContextVar[set[tuple[int, int]] | None]c"]tRt^jtRt^t^tRtR#) SourceTypeav This Enum divides VariableTracker into 2 cases, depending on the variable it represents: - already existed that Dynamo began tracking while introspection (Existing) - is a new variable that is created during Dynamo introspection (New) In general, we have these invariants: 1. for `VariableTracker` associated with `Existing`, its `source` field must not be None. 2. for `VariableTracker` associated with `New`, most of the time its `source` field is None, except for cases like side effect codegen for `AttributeMutationNew`, during which we generate a `LocalSource('tmp...')` for such variable, to facilitate codegen. r?N)rNrOrPrQrRExistingNewrTr?rDrArXrXjs H CrDrXc&]tRt^}tRtRRltRtR#) MutationTypez Base class for Variable.mutation_type. It encodes information about 1. The type of mutation Dynamo allows on the variable. 2. Whether the value represented by this variable already existed before Dynamo tracing. c V^8dQhRRRR/#)r=typrXr>Noner?)r@s"rArBMutationType.__annotate__s%%J%4%rDc V\PJd ^VnR#V\PJd\ 4VnR#\ RRV RV 2RV R2R.\ POR7R#) zUnsupported SourceTypezMutationType.__init__  z"Dynamo does not support the type ``z,This branch is not supposed to be reachable.gb_typecontext explanationhintsN)rXrYscoperZrrr DYNAMO_BUG)rJr^s&&rA__init__MutationType.__init__sl0 *%% %DJ JNN ")+DJ 00au=@QGB&11  rD)rjN)rNrOrPrQrRrlrTr?rDrAr\r\}s%%rDr\c]tRt^t^t^tRtR#)AttrMutationKindr?N)rNrOrPrQGENERIC_SETATTR INSTANCE_DICTrTr?rDrArorosOMrDrocJa]tRt^tRtRV3RlltRRltRRltRtV;t #) ValueMutationNewa This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value itself (rather than its attributes). 2. The value is created by the bytecode Dynamo is tracing through. For instance, Dynamo could model a newly created list with this marker, indicating that while we need to model mutations to this list, we don't have to emit bytecode for these mutations if the list doesn't escape into the Python world. cV^8dQhRR/#r=r>r_r?)r@s"rArBValueMutationNew.__annotate__s))$)rDc B<\SV`\P4R#N)superrlrXrZrJ __class__s&rArlValueMutationNew.__init__s (rDcV^8dQhRR/#)r=r>r5r?)r@s"rArBrvs#rDc \V4#rx)idrJs&rA__hash__ValueMutationNew.__hash__s $xrDc V^8dQhRRRR/#)r=otherobjectr>boolr?)r@s"rArBrvsFtrDc WJ#rxr?)rJrs&&rAr#ValueMutationNew.__eq__s }rDr?) rNrOrPrQrRrlrr#rT __classcell__r{s@rArsrss! ))rDrscBa]tRt^t$RtR]R&RRV3RllltRtV;t#)ValueMutationExistinga This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value itself (rather than its attributes). 2. The value exists before Dynamo tracing started. For instance, Dynamo could model a pre-existing list with this marker, indicating that if we encounter mutations to this list, we need to buffer and re-apply those mutations after the graph runs, since the list might be used afterwards in Python. r is_modifiedc V^8dQhRRRR/#)r=rrr>r_r?)r@s"rArB"ValueMutationExisting.__annotate__s''D'T'rDc N<\SV`\P4WnR#rx)ryrlrXrYr)rJrr{s&&rArlValueMutationExisting.__init__s ,,-&rD)rF) rNrOrPrQrRrSrlrTrrs@rArrs '''rDrc]tRt^tRtRtR#)AttributeMutationzt This case of VariableTracker.mutation_type marker indicates that Dynamo allows mutation on the value's attributes. r?NrNrOrPrQrRrTr?rDrArrsrDrc2a]tRt^tRtRV3RlltRtV;t#)AttributeMutationExistinga This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value's attributes. 2. The value exists before Dynamo tracing started. For instance, Dynamo could model a pre-existing object with this marker, indicating that if we encounter mutations to this object, we need to buffer then re-apply those mutations after the graph runs, since the object might be used afterwards in Python. cV^8dQhRR/#rur?)r@s"rArB&AttributeMutationExisting.__annotate__s..$.rDc B<\SV`\P4R#rx)ryrlrXrYrzs&rArl"AttributeMutationExisting.__init__s ,,-rDr?rNrOrPrQrRrlrTrrs@rArrs ..rDrc6a]tRt^tRtRRV3RllltRtV;t#)AttributeMutationNewa This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value's attributes. 2. The value is created by the bytecode Dynamo is tracing through. For instance, Dynamo could model a newly created object with this marker, indicating that while we need to model mutations to this object, we don't have to emit bytecode for these mutations if the object doesn't escape into the Python world. c V^8dQhRRRR/#)r= cls_source Source | Noner>r_r?)r@s"rArB!AttributeMutationNew.__annotate__s%%=%D%rDc N<\SV`\P4WnR#rx)ryrlrXrZr)rJrr{s&&rArlAttributeMutationNew.__init__s ($rD)rrxrrs@rArrs %%%rDrc V^8dQhRRRR/#)r=scope_idr5r>rr?)r@s"rArBrB s#$rDc V^8H#)r?)rs&rA_is_top_level_scoper s q=rDc V^8dQhRRRR/#)r=mr\r>rr?)r@s"rArBrBs  < D rDcZ\4p\V4'dR#VPV8H#)T)rrrj)rrs& rAis_side_effect_safers)!H 8$$ 77h rDc V^8dQhRRRR/#)r=objVariableTrackerr>typer?)r@s"rArBrBs     4  rDcVP4# \d'\RT R2R.\POR7R#i;i)zUnsupported python_type() callz! does not implement python_type()zhThis VariableTracker does not implement python_type(), which is required for object protocol operations.reN) python_typeNotImplementedErrorrrrk)rs&rAmaybe_get_python_typersW      4e<=@"--     s-AAc]tRtRtRtRtR#)NO_SUCH_SUBOBJi)z;Sentinel indicating no concrete Python object is available.r?Nrr?rDrArr)sErDrc*]tRtRtRt^t^t^t^tRt R#) MethodFlagsi-zCalling convention for a `Method` handler, mirroring CPython's PyMethodDef ml_flags. `call_method` enforces arity from these before invoking the handler, centralizing the raise_args_mismatch boilerplate.r?N) rNrOrPrQrRVARARGSKEYWORDSNOARGSOrTr?rDrArr-sOGH F ArDrc4V^8dQhRRRRRRRRR R R R R R /#)r=vtrtxr-namer3flagsrargsr kwargsr>r_r?)r@s"rArBrB8sWYYY!Y Y  Y  Y  Y YrDc\V4pVP4 RV 2pV'd,V\P,'g\ W R24V\P \P ,,'drV\P ,'dV'd\ W RV R24V\P ,'dV^8wd\ W RV R24R#R#R#R#).z() takes no keyword argumentsz() takes no arguments (z given)z() takes exactly one argument (N)lenpython_type_namerrrrr)rrrrrrnqualnames&&&&&& rA_check_method_arityr8s D A%%'($0H u{3333z)FGH ""[]]233 ;%% % %$ R:-DQCw!O P ;== Q!V R:-LQCw!W X&, 4rDc$V^8dQhRRRRRR/#)r=rrrr3r>rr?)r@s"rArBrBTs!d#+rDcv^RIp\P\P,p\ WR4pVfRM.VP P PPV4pVfV#\^4pV\,'dV\P,pV\,'dV\P,pV\\,,'dV\P,pV\,'dV\P,pT;'gT#)zArity convention for `python_type.name`, read from its CPython PyMethodDef.ml_flags. Falls back to VARARGS|KEYWORDS when python_type has no such attribute or it carries no ml_flags (slot wrappers, pure-Python functions).N)torchrrrgetattr_C_dynamo eval_frameget_method_ml_flags _METH_NOARGSr_METH_Or _METH_VARARGS_METH_FASTCALL_METH_KEYWORDS)rrrrVfnmlrs&& rA_flags_from_ml_flagsrSs !!K$8$88G D )B!1!1!rr?)r@s"rArBrBls!>>_>C>K>rDc6\VP4V4#)zuResolve the arity flags for method `name` from the VT's arity-reference type (see VariableTracker.method_flags_type).)rmethod_flags_type)rrs&&rA_derive_method_flagsrls  4 4 6 ==rD)slotsc2]tRtRt$RtR]R&RRltRtR#) MethodiraDeclarative entry in a VariableTracker's `tp_methods` table, analogous to CPython's PyMethodDef. Handlers have signature `(self, tx, args, kwargs)` and return the result VariableTracker, or None to decline the call (the equivalent of the old `super().call_method` fall-through) so `call_method` continues to the object-protocol dispatch below. The method name is the tp_methods key this entry is stored under; its arity convention is derived on demand from that method's ml_flags (see _derive_method_flags).%Callable[..., VariableTracker | None]handlerc 0V^8dQhRRRRRRRRR RR R /#) r=rrrr-rr3rr rr>VariableTracker | Noner?)r@s"rArBMethod.__annotate__sF 2 2  2 & 2 2  2  2  2rDc V\W4p\WW6WE4VPWWE4#rx)rrr)rJrrrrrrs&&&&&& rA__call__Method.__call__s+%R.BD>||BD11rDr?NrNrOrPrQrRrSrrTr?rDrArrrsI32 2 2rDrc4]tRtRt$RtR]R&RtR]R&RtR#) GetSetiz`tp_getset` entry, analogous to CPython's PyGetSetDef. `getter` `(self, tx) -> VT | None` (None declines); `setter` `(self, tx, value) -> VT | None`, None for read-only.rgetterN,Callable[..., VariableTracker | None] | Nonesetterr?rNrOrPrQrRrSrrTr?rDrArrs= 21;?F 8?rDrc4]tRtRt$RtR]R&RtR]R&RtR#) Memberiz`tp_members` entry, analogous to CPython's PyMemberDef. Same shape as GetSet; a distinct type so members and getsets never share a class.rrNrrr?rr?rDrArrsK 21;?F 8?rDrc V^8dQhRRRR/#)r=accessorz Callable[[Any], VariableTracker]r>Callable[..., VariableTracker]r?)r@s"rArBrBs++.+#+rDcaV3Rl#)z>Getter for a GetSet/Member whose value is an already-built VT.c<S!V4#rxr?rJrrs&&rAgetset_read..s HTNrDr?rsfrA getset_readrs  +*rDc V^8dQhRRRR/#)r=rzCallable[[Any], Any]r>rr?)r@s"rArBrBs FF"F#FrDcaV3Rl#)zBGetter that builds a VT from the raw value returned by `accessor`.c<<\PVS!V44#rx)rbuildrs&&rArgetset_build..sO11"htnErDr?rsfrA getset_buildrs  FErDc V^8dQhRRRR/#)r=rr3r>rr?)r@s"rArBrBs  3 #H rDcaRV3RllpV#)c$V^8dQhRRRRRR/#)r=rrrr-r>rr?)r@s"rArB&unsupported_attr..__annotate__s$   !:  rDc<\RRV RS 2\V4P RS R2.\POR7R#)zUnsupported attributezattr_unsupported rcz does not support attribute ''reNrrrNrrk)rrrs&&rA graph_break%unsupported_attr..graph_breaksL +'t1TF38,,--J4&PQR1%001  rDr?)rr sf rAunsupported_attrr s   rDc>a]tRtRt$R]R&RRV3RllltRtV;t#)#AsPythonConstantNotImplementedErrorirrc$V^8dQhRRRRRR/#)r=rrmsgz str | Noner>r_r?)r@s"rArB0AsPythonConstantNotImplementedError.__annotate__s!?trDc H<VfV R2MTp\SV`V4WnR#)Nz is not a constant)ryrlr)rJrrr{s&&&rArl,AsPythonConstantNotImplementedError.__init__s(+.;&'C rD)rrx)rNrOrPrQrSrlrTrrs@rArrsrDrc 0V^8dQhRRRRRRRRR R R R/# r=rJrrr-funcrrlist[VariableTracker]rdict[str, VariableTracker]r>r?)r@s"rArBrBsD    !  )   '   rDcV'd \VR4\V4^8wd\VR\V4 24V!W4#)&this method takes no keyword argumentszexpected 0 arguments, got rrrJrrrrs&&&&&rA_wrap_unaryfuncrs=EF 4yA~9#d)EF >rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBsD ! !  !! ! ) ! ! ' !  !rDcV'd \VR4\V4^8wd\VR\V4 24V^,pV!WV4#rexpected 1 argument, got rrJrrrrrs&&&&& rA_wrap_binaryfuncr$sIEF 4yA~8T DE GE %  rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBsD / /  /! / ) / / ' /  /rDcV'd \VR4\V4^8wd\VR\V4 24V^,pV!WVRR7#)rr"Treverserr#s&&&&& rA_wrap_binaryfunc_rr)sKEF 4yA~8T DE GE % ..rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBsD & &  &! & ) & & ' &  &rDcV'd \VR4\V4R9d\VR\V4 24\V4^8Xd V^,MRpV!WV^,V4#)rexpected 1 or 2 arguments, got Nrr=rrJrrrrzs&&&&& rA_wrap_ternaryfuncr0s\EF 4y>s4ykJKt9>QtA $q'1 %%rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBsD 4 4  4! 4 ) 4 4 ' 4  4rDcV'd \VR4\V4R9d\VR\V4 24\V4^8Xd V^,MRpV!WV^,VRR7#)rr,NTr'r-rr.s&&&&& rA_wrap_ternaryfunc_rr3s^EF 4y>s4ykJKt9>QtA $q'1d 33rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrB$sD$$ $!$ )$ $ ' $  $rDcV'd \VR4^RIHp\V4^8wd\VR\V4 24VwrgV!WV4pV!WW4#)rgetindexexpected 2 arguments, got robject_protocolr7r) rJrrrrr7argvalueis &&&&& rA_wrap_sq_setitemr>$sVEF) 4yA~9#d)EFLS3A ! ##rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrB6D ! )  '   rDcV'd \VR4^RIHp\V4^8wd\VR\V4 24VwpV!WV4pV!WV4#)rr6r"r9rJrrrrr7r;r=s&&&&& rA _wrap_sq_itemrC6sVEF) 4yA~8T DE ES3A ! rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBIr@rDcV'd \VR4^RIHp\V4^8wd\VR\V4 24VwpV!W\R7pV!WV4#)r)pynumber_as_ssize_tr")err)rr:rFr OverflowError)rJrrrrrFor=s&&&&& rA_wrap_indexargfuncrJIsVEF4 4yA~8T DE CQB}5A ! rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrB\sD & &  &! & ) & & ' &  &rDcV'd \VR4\V4^8wd\VR\V4 24VwrVV!WWV4#rr8r)rJrrrrkeyr<s&&&&& rA_wrap_objobjargprocrO\sGEF 4yA~9#d)EFLS # %%rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBksD % %  %! % ) % % ' %  %rDcV'd \VR4\V4^8wd\VR\V4 24VwpV!WVR4#rr"Nr)rJrrrrrNs&&&&& rA _wrap_delitemrSksIEF 4yA~8T DE ES #t $$rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBzsD## #!# )# # ' #  #rDcV'd \VR4^RIHp\V4^8wd\VR\V4 24VwpV!WV4pV!WVR4#)rr6r"Nr9rBs&&&&& rA_wrap_sq_delitemrVzsXEF) 4yA~8T DE ES3A !T ""rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBsD ! !  !! ! ) ! ! ' !  !rDcV'd \VR4\V4^8wd\VR\V4 24VwpV!WV4#r!rr#s&&&&& rA_wrap_objobjprocrYsGEF 4yA~8T DEGU %  rDc V^8dQhRRRR/#)r=opr3r>zCallable[[VariableTracker, InstructionTranslatorBase, Callable[..., VariableTracker], list[VariableTracker], dict[str, VariableTracker]], VariableTracker]r?)r@s"rArBrBs  rDcaRV3RllpV#)c 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArB#_make_richcmp..__annotate__sD + + + % +- +$ + + +  +rDc<V'd \VR4\V4^8wd\VR\V4 24V!WV^,S4#r!r)rJrrrrr[s&&&&&rAwrapper_make_richcmp..wrappersG  R!I J t9> R#s4ykJK q'C4y1}DG#*>*>r;*OE # %%rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrBrjrDcV'd \VR4\V4^8wd\VR\V4 24V!WV^,V^,4#rMrrs&&&&&rA_wrap_descr_setr|rmrDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrB%sD ) )  )! ) ) ) ) ' )  )rDcV'd \VR4\V4^8wd\VR\V4 24V!WV^,R4#rRrrs&&&&&rA_wrap_descr_deleter%sFEF 4yA~8T DE $q'4 ((rDc*]tRtRtRt^t^t^t^tRt R#) SlotGroupi5zVA CPython slot group. The value is the group's index into get_type_slots()'s 4-tuple.r?N) rNrOrPrQrRSEQUENCEMAPPINGNUMBERTYPErTr?rDrArr5s#HG F DrDrc2]tRtRt$RtR]R&RRltRtR#) Sloti?z\A CPython type slot: the name of the VariableTracker method implementing the slot function. r3implc ,V^8dQhRRRRRRRRRR/#) r=rrrr-rr rr>r?)r@s"rArBSlot.__annotate__GsCEE E &E E  E  ErDc T\\V4VP4!W.VO5/VB#rx)rrr)rJrrrrs&&&*,rAr Slot.__call__Gs(tBx+BDTDVDDrDr?Nrr?rDrArr?s IEErDrcv]tRtRt$R]R&R]R&R]R&R]R&R]R&R]R&R]R &R]R &R]R &R]R &R]R &R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R&R]R &R]R!&R]R"&R]R#&R]R$&R]R%&R&tR'#)(PyNumberMethodsiQ Slot | Nonenb_boolnb_indexnb_intnb_float nb_negative nb_positive nb_absolute nb_invertnb_addnb_inplace_add nb_subtractnb_inplace_subtract nb_multiplynb_inplace_multiply nb_remaindernb_inplace_remaindernb_powernb_inplace_power nb_lshiftnb_inplace_lshift nb_rshiftnb_inplace_rshiftnb_andnb_inplace_andnb_xornb_inplace_xornb_or nb_inplace_ornb_floor_dividenb_inplace_floor_dividenb_true_dividenb_inplace_true_divide nb_divmodnb_matrix_multiplynb_inplace_matrix_multiplyr?NrNrOrPrQrSrTr?rDrArrQs   $$$$%%!!""""     ((''## ++rDrch]tRtRt$R]R&R]R&R]R&R]R&R]R&R]R&R]R &R]R &R tR #) PySequenceMethodsiwr sq_length sq_concat sq_repeatsq_item sq_ass_item sq_containssq_inplace_concatsq_inplace_repeatr?Nrr?rDrArrws2 """"rDrc6]tRtRt$R]R&R]R&R]R&RtR#)PyMappingMethodsir mp_length mp_subscriptmp_ass_subscriptr?Nrr?rDrArrs!!rDrc]tRtRt$R]R&R]R&R]R&R]R&R]R&R]R&R]R &R]R &R]R &R]R &R]R &R]R&R]R&R]R&R]R&RtR#) PyTypeObjectirtp_reprtp_hashtp_calltp_str tp_getattro tp_setattrotp_richcomparetp_iter tp_iternext tp_descr_get tp_descr_settp_initr tp_as_numberrtp_as_sequencer tp_as_mappingr?Nrr?rDrArrs\      !!%%##rDrc#V"TFqPR4'dKVxK! R#5i)tp_as_N) startswith).0fs& rA rsS 41LLdict[str, Any]r?)r@s"rArBrBs"#<rDc WP,p\V,p\V,P4UUu/uF*wrET\ W$4'd\ W5,4MRbK, upp#uuppirx)r<rritemsrr)rrmaskfn_mapbitfields&& rA_fillrsf  D e_F%U+113 3JC ht&9&9tFM"tC3  s0A2c V^8dQhRRRR/#)r=obj_typerr>rr?)r@s"rArBrBs  t  rDc0\V4p\RR\R/\\P V4BR\ R/\\PV4BR\R/\\PV4B/\\PV4B#)rrrr?) rrrrrrrrrrr)rrs& rA_tp_typers 8 $E  $FuY-=-=u'EF (L51C1CU+KL 'Iy/@/@%)HI    &  rDrr-r WrapperTypecZ]tRtRt$RtR]R&R]R&R]R&R]R &R ]R &R R ltRtR#)SlotDefia;One slotdefs entry: dunder name -> (impl method, call wrapper). ``impl`` is the VariableTracker method implementing the behavior; ``wrapper`` adapts a generic ``(self, tx, args)`` call to it (arity + operand order). A reflected op reuses the forward impl with ``_wrap_binaryfunc_r``; an in-place op is a plain binary call into its own ``nb_inplace_*`` impl -- so no reverse/inplace flags are needed. ``group`` is the CPython slot group (``SlotGroup``) the entry lives in and ``slot`` is the bit within it; together they decide whether a real type implements the slot. r3rrrrr5slotrr`c ,V^8dQhRRRRRRRRR R/#) r=rrrr-rrrrr>r?)r@s"rArBSlotDef.__annotate__s< 8 8  8 & 8$ 8 + 8  8rDc f\\V4VP4pVPWWSV4#)z:Call the impl via the wrapper, passing self, tx, and args.)rrrr`)rJrrrrrs&&&&& rArSlotDef.__call__s+tBx+||BD77rDr?Nrr?rDrArrs.  I I  I  8 8rDrc ,V^8dQhRRRRRRRRRR /# r=rr3rrr5r`rr>rr?)r@s"rArBrBs6BBBCBsB[BWBrDc8\W\PW#4#rx)rrrrrrr`s&&&&rASQSLOTrs 4y114 AArDc ,V^8dQhRRRRRRRRRR /#rr?)r@s"rArBrBs6AAACAsA[AWArDc8\W\PW#4#rx)rrrrs&&&&rAMPSLOTrs 4y00$ @@rDc ,V^8dQhRRRRRRRRRR /#rr?)r@s"rArBrB s6@@@C@s@[@W@rDc8\W\PW#4#rx)rrrrs&&&&rANBSLOTr s 4y// ??rDc ,V^8dQhRRRRRRRRRR /#rr?)r@s"rArBrBs/>>>C>s>[>W>rDc8\W\PW#4#rx)rrrrs&&&&rATPSLOTrs 4y~~t ==rD__repr__ tp_repr_implr__str__ tp_str_impl__getattribute__rw __getattr__tp_richcompare_impl__iter__ tp_iter_impl__next__tp_iternext_impl__get__tp_descr_get_impl__set__tp_descr_set_impl __delete__rl tp_init_impl__add__ nb_add_impl__radd____sub__nb_subtract_impl__rsub____mul__nb_multiply_impl__rmul____mod__nb_remainder_impl__rmod__ __divmod__nb_divmod_impl __rdivmod____pow__ nb_power_impl__rpow____neg__nb_negative_impl__pos__nb_positive_impl__abs__nb_absolute_impl__bool__ nb_bool_impl __invert__nb_invert_impl __lshift__nb_lshift_impl __rlshift__ __rshift__nb_rshift_impl __rrshift____and__ nb_and_impl__rand____xor__ nb_xor_impl__rxor____or__ nb_or_impl__ror____int__ nb_int_impl __float__ nb_float_impl__iadd__nb_inplace_add_impl__isub__nb_inplace_subtract_impl__imul__nb_inplace_multiply_impl__imod__nb_inplace_remainder_impl__ipow__nb_inplace_power_impl __ilshift__nb_inplace_lshift_impl __irshift__nb_inplace_rshift_impl__iand__nb_inplace_and_impl__ixor__nb_inplace_xor_impl__ior__nb_inplace_or_impl __floordiv__nb_floor_divide_impl __rfloordiv__ __truediv__nb_true_divide_impl __rtruediv__ __ifloordiv__nb_inplace_floor_divide_impl __itruediv__nb_inplace_true_divide_impl __index__ nb_index_impl __matmul__nb_matrix_multiply_impl __rmatmul__ __imatmul__nb_inplace_matrix_multiply_impl__len__mp_length_impl __getitem__mp_subscript_impl __setitem__mp_ass_subscript_impl __delitem__sq_length_implsq_concat_implsq_repeat_impl sq_item_implsq_ass_item_impl __contains__sq_contains_implsq_inplace_concat_implsq_inplace_repeat_implz list[SlotDef] _SLOTDEFSc V^8dQhRRRR/#)r=rrr>zdict[str, SlotDef]r?)r@s"rArBrBs   ); rDc\V4p/p\F\pVPV9dK\WPP ,VP 4'gKNW2VP&K^ V#rx)rryrrrr<r)rrdsds& rA_compute_slotdefsr~sW 8 $EA 77a<  E((..)277 3 3bggJ  HrDcPa]tRtRt$.tR]R&RV3RlltRV3RlltRtV;t #) VariableTrackerMetai z list[type]all_subclassesc ,V^8dQhRRRRRRRRR R/#) r=mcsrrr3basestuple[type, ...]attrsrr>r?)r@s"rArB VariableTrackerMeta.__annotate__ s4 %5>L rDc <VPRR4;'g5\;QJdRV4F 'gK RM RM !RV44pV'gVR8Xd\SV` \WV4#\SV` \ WV4#)_no_implicit_realizeFc3<"TFp\VRR4xK R#5i)rFN)r)rbases& rAr.VariableTrackerMeta.__new__..s!N EJTGD0% 8 8UsTr)getanyry__new__r$ImplicitRealizingVariableTrackerMeta)rrrrno_implicit_realizer{s&&&& rArVariableTrackerMeta.__new__ s$ii(>F  ##N EJN ###N EJN K  $*;";7?#6UK K7?4d5 rDc ,V^8dQhRRRRRRRRR R /#) r=clsrrr3rrrrr>r_r?)r@s"rArBrs477 77%57>L7 7rDc f<\SV`WV4\PP V4R#rx)ryrlrrappend)rrrrr{s&&&&rArlVariableTrackerMeta.__init__s( e,**11#6rDr?) rNrOrPrQrrSrrlrTrrs@rArr s!!#NJ#&77rDrc"]tRtRtRRltRtR#)ri&c V^8dQhRRRR/#)r=instancerr>rr?)r@s"rArB1ImplicitRealizingVariableTrackerMeta.__annotate__'s--&-T-rDc f\\V4'dVPV4#\W4#)z-Make isinstance work with LazyVariableTracker) instancecheckLazyVariableTrackerlazy_isinstance)rJrs&&rA__instancecheck__6ImplicitRealizingVariableTrackerMeta.__instancecheck__'s, ,h 7 7++D1 1T,,rDr?N)rNrOrPrQrrTr?rDrArr&s --rDrcLa]tRtRt$RtRtR]R&/tR]R&/tR]R &/t R ]R &R R lt RRlt RRlt RRlt 0ER mtRtR]R&RRlt]ER RRll4tRRltRR ltR!R"ltR#R$ltR%R&ltR'R(ltR)R*ltR+R,ltR-R.ltR/R0ltR1R2ltR3R4ltR5R6ltR7R8lt R9R:lt!R;R<lt"R=R>lt#R?R@lt$RARBlt%RCRDlt&RERFlt'RGRHlt(RIRJlt)RKRLlt*RMRNlt+RORPlt,RQRRlt-RSRTlt.RURVlt/RWRXlt0RYRZlt1R[R\lt2R]R^lt3R_R`lt4RaRblt5RcRdlt6ReRflt7RgRhlt8RiRjlt9RkRllt:RmRnlt;RoRpltRuRvlt?RwRxlt@RyRzltAR{R|ltBR}R~ltCRRltDRRltERRltFRRltGRRltHRRltIRRltJRRltKRRltLRRltMRRltNRRltORRltP]QER RRll4tRRRltSRRltTRRltURRltVRRltWRRltXRRltYRRltZER RRllt[ER RRllt\RRlt]ER RRllt^RRlt_ER RRllt`RRltaER RRlltbRRltcER RRlltdRRlteER RRlltfRRltgER RRllthRRltiER RRlltjER RRlltkRRltlRRltmER RRlltnRRltoER RRlltpRRltqRRltrRRltsER RRllttRRltuRRltvRRltwER RRlltxRRltyER RRlltzRRlt{RRlt|RRlt}RRRRRRRR/RV3Rlllt~]RRl4t]RRl4t]RRl4t]RRl4t]RRl4t]RRl4t]RERl4t]ERERl4t]ER4tERV3ERllt]QERERl4tERtV;t#( ri.z Base class for tracked locals and stack values VariableTracker instances are immutable and should be copied in order to change them. Prefer the factory function VariableTracker.build() over VariableTracker.__init__(). Nztype | tuple[type, ...] | None _cpython_typezdict[str, Method] tp_methodszdict[str, GetSet] tp_getsetzdict[str, Member] tp_membersc$V^8dQhRRRRRR/#)r=rr3 table_attrsr>r r?)r@s"rArBVariableTracker.__annotate__Is!SrDc \V4PFApVPpVF,pVPV4pVfKW9gK"Wa,uu# KC R#)aResolve *name* in the given declarative tables across the class MRO. Mirrors CPython method/getset/member inheritance: each class contributes its own table entries into its own tp_dict, and lookup walks the MRO with the most-derived class winning. A subclass table therefore *adds* to (rather than replaces) the entries it inherits. Within a class the tables are consulted in the order given. N)r__mro____dict__r)rJrrklassr|attrtables&&* rA_lookup_tp_table VariableTracker._lookup_tp_tableIsO$Z''EA#d $ ;&$( rDc V^8dQhRRRR/#)r=rr3r>zGetSet | Member | Noner?)r@s"rArBrZsFFCF4JFrDc (VPVRR4#)rrrrJrs&&rAlookup_tp_getset_member'VariableTracker.lookup_tp_getset_memberZs$$T; EErDc V^8dQhRRRR/#)r=rr3r>z Method | Noner?)r@s"rArBr]s99S9]9rDc &VPVR4#)rrrs&&rAlookup_tp_method VariableTracker.lookup_tp_method]s$$T<88rDcV^8dQhRR/#r=r>rr?)r@s"rArBr`s++4+rDc \V4#)a2Type whose CPython ml_flags define this VT's tp_methods arities (see _derive_method_flags). Defaults to python_type(); a VT whose python_type() is a pure-Python stand-in without C ml_flags (e.g. OrderedSet) overrides this to the builtin whose method arities it mirrors, so MethodFlags still enforces arity.)rrs&rAr!VariableTracker.method_flags_type`s %T**rDrLsource_location mutation_typedict_vt#variables.DunderDictVariable | Nonec V^8dQhRRRR/#)r=rr r>rr?)r@s"rArBrws&&c&o&rDc r\VP4pVPV4VP!R/VB#)z)Shallow copy with some (optional) changesr?)dictrupdater{)rJrrs&, rAcloneVariableTracker.clonews.DMM" F~~%%%rDc ,V^8dQhRRRRRRRRR R /#) r=rz!Callable[[VariableTracker], None]r<r cachezdict[int, Any] | None side_effectszSideEffects | Noner>r_r?)r@s"rArBr~s<8080 -8080% 80 ) 80  80rDc Vf/pV.pV'EdtVP4p\V4pWs9dK,WcV&\V\4'dVP 4pV!V4VP 4pVP pVP P4U U u.uFwrW9gK V NK p p p VeAWdP9d1V PVPV,P44Mk\V\\34'd \V4p MC\V\\P 34'd\VP44p MEK_VP\#V 44EK|R#uup p i)a} Walk value and call fn on all the VariableTracker instances. When side_effects is provided, also walks attributes stored in store_attr_mutations (e.g. dataclass fields set during tracing that aren't in the VT's __dict__). Implemented with an explicit worklist rather than recursion so that deeply chained structures (e.g. the ~N-node buffer built by itertools.tee over a long iterable) do not overflow the Python stack. Children are pushed in reverse so they are popped in their original order, matching the pre-order DFS the previous recursive version produced (callers rely on the visitation order, e.g. for graph output ordering). N)popr isinstancerunwrap_nonvar_fieldsrrstore_attr_mutationsextendvaluesr"listtupler collections OrderedDictreversed) rrr<rrworklistcuridxnonvarsrNsubvaluechildrens &&&&& rAvisitVariableTracker.visit}sB. =E7h,,.CS'C|#J#//jjl3jjl,,*-););)=)= )H)= !,@@@OOL$E$Ec$J$Q$Q$STdE]++9dK$;$;<== - OOHX. /=s ! F0FcV^8dQhRR/#r<r?)r@s"rArBrs..#.rDc 4VPP R2#)z())r{rNrs&rArVariableTracker.__repr__s..))*"--rDcV^8dQhRR/#r<r?)r@s"rArBrsCrDc n\VP44# \d\T4u#i;irx)reprrsrrs&rA debug_reprVariableTracker.debug_reprs4 //12 2" :  s 44cV^8dQhRR/#rr?)r@s"rArBrsGGTGrDc r\VP44# \d\T R24Rhi;i)a Abstract method to be implemented by subclasses of VariableTracker. This method should return the type represented by the instance of the subclass. The purpose is to provide a standardized way to retrieve the Python type information of the variable being tracked. Returns: type: The Python type (such as int, str, list, etc.) of the variable tracked by the subclass. If the type cannot be determined or is not relevant, leaving it undefined or invoking super() is always sound. Note: This is an abstract method and may be overridden in subclasses. Example: class SetVariable(VariableTracker): def python_type(self): return set Raises: NotImplementedError: If the method is not implemented in a subclass. z has no typeN)rrsrrs&rArVariableTracker.python_typesA0 G//12 2" G%l&;<$ F Gs6cV^8dQhRR/#r<r?)r@s"rArBrs $ $# $rDc \VP4P# \dR#i;i)aq Return the type name for the Python type this VariableTracker represents. Mirrors CPython's tp_name slot (PyTypeObject.tp_name). In Python 3.10+, type.__name__ matches CPython's tp_name exactly (e.g., "list", "NoneType"). Note: There are no external callers outside torch._dynamo that rely on this. Internal uses should prefer this over hardcoded type name strings. )rrNrrs&rAr VariableTracker.python_type_names. $##%.. ." $# $s  ++cV^8dQhRR/#r<r?)r@s"rArBrssrDc VP4pTPpTPpTR9dT RT 2#T# \dR#i;i)z~ Equivalent to _PyType_GetFullyQualifiedName See https://github.com/python/cpython/blob/v3.15.0b4/Objects/typeobject.c#L1658 rr)__main__builtins)rrrOrP)rJtype_modqns& rApython_qualified_name%VariableTracker.python_qualified_names_  $$$&E     . .U!B4= I# $# $s: A A cV^8dQhRR/#r=r>r r?)r@s"rArBrs88C8rDc \V4h)z For constantsrrs&rArs"VariableTracker.as_python_constants 1$77rDcV^8dQhRR/#rr?)r@s"rArBrs  # rDc tVP4# \d\RRT 2RT R2.R7R#i;i)zcSimilar to as_python_constant(), but add ID_MATCH guards to try to force things to become constantszNot a Python constantzguard_as_python_constant zFailed to convert z into a Python constant.reN)rsrrrs&rAguard_as_python_constant(VariableTracker.guard_as_python_constantsO **, ,"  /3D6:06NO    s !77cV^8dQhRR/#r=r>rr?)r@s"rArBrsDrDc LVP4R# \dR#i;iTF)rsrrs&rAis_python_constant"VariableTracker.is_python_constants(   # # %"    ##c V^8dQhRRRR/#)r=rr-r>rr?)r@s"rArBrs8=SrDc R#rxr?rJrs&&rAr.VariableTracker.nb_bool_impls rDcV^8dQhRR/#rr?)r@s"rArBr s  T rDc R#)uWhether the underlying Python object is hashable. Defaults to True — in CPython, all objects are hashable unless the type explicitly sets __hash__ = None (PyObject_HashNotImplemented). https://github.com/python/cpython/blob/e76aa128fe/Objects/typeobject.c#L7288 Subclasses override to return False for unhashable types (list, dict, set, etc.). This is a fast, side-effect-free check used by HashableTracker and __contains__ before attempting to hash. Tr?rs&rA is_hashableVariableTracker.is_hashable srDc V^8dQhRRRR/#)r=rr-r>ztuple[int, bool]r?)r@s"rArBr-s!!5!:J!rDc  VP'dVPPVP4p\V4P\ PJde\ RRV 2\V4P R\V4P R\V4P R2.\POR7\V4R3#\V4R3#)aeDefault tp_hash: object.__hash__ = PyObject_GenericHash (identity hash). PyObject_GenericHash: https://github.com/python/cpython/blob/e76aa128fe/Python/pyhash.c#L143-L147 Assigned on object: https://github.com/python/cpython/blob/e76aa128fe/Objects/typeobject.c#L7288 Returns (hash_value, is_fake). is_fake=True means the hash depends on a sourceless object's identity and must not escape into output bytecode. VT subclasses override this for types with their own C tp_hash. UserDefinedObjectVariable.hash_impl handles the MRO walk for custom Python __hash__, builtin-inherited C hash, and unknown C tp_hash. zMissing hash_impl overridez hash_impl z wraps z> which has a non-default tp_hash. Add a hash_impl override to rreT) rLoutputresolve_source_valuerrrrrNrrkhashr)rJrreal_vals&& rArVariableTracker.hash_impl-s ;;;yy55dkkBHH~&&foo=8(/#':#6#6"7wH~../0<Dz**+1!.:-889>4' '$x~rDc(V^8dQhRRRRRRRR/#)r=rr-rrr[r3r>r?)r@s"rArBrPs2  %       rDc  \RRV RV 2\V4P R\V4P R2.\POR7R#)aAPer-VT tp_richcompare slot. Subclasses must override. Analogous to CPython's tp_richcompare function pointer on PyTypeObject. Returns ConstantVariable(NotImplemented) when the type does not handle the comparison (signaling do_richcompare to try the other operand). Called from two paths: - call_method("__eq__") calls tp_richcompare_impl directly (like CPython's a.__eq__(b) calling tp_richcompare without do_richcompare). - generic_richcompare calls tp_richcompare_impl as part of the 4-step do_richcompare algorithm (subclass priority, forward, reflected, fallback). z$Missing tp_richcompare_impl overrideztp_richcompare_impl rczO does not implement tp_richcompare_impl. Add a tp_richcompare_impl override to rreNr )rJrrr[s&&&&rAr #VariableTracker.tp_richcompare_implPs]& :*4&"6:../0JDz""#1&2%001  rDc V^8dQhRRRR/#)r=rr r>rr?)r@s"rArBrls    rDc R#)a> Check if this variable is a python constant matching one of the given values. Examples: var.is_constant_match(None) # True if var is constant None var.is_constant_match(True, False) # True if var is constant True or False var.is_constant_match(NotImplemented) # True if var is constant NotImplemented Fr?)rJrs&*rAis_constant_match!VariableTracker.is_constant_matchlsrDcV^8dQhRR/#rr?)r@s"rArBrws$rDc R#)z0Check if this variable is a constant None value.Fr?rs&rAis_constant_none VariableTracker.is_constant_nonewrDc V^8dQhRRRR/#)r=rzCallable[..., Any]r>r)r?)r@s"rArBr{s""/"E"rDc hVP'dVPPV4#\hrx)rL make_guardr)rJrs&&rAr'VariableTracker.make_guard{s& ;;;;;))"- -!!rDc$V^8dQhRRRRRR/#)r=rr-rr3r>r r?)r@s"rArBrs""" 9"""rDc \h)z/getattr(self, name) returning a python constantrrJrrs&&&rA const_getattrVariableTracker.const_getattrs!!rDcV^8dQhRR/#rr?)r@s"rArBrsrDc R#)zAReturn True for values that can participate in SymNode operationsFr?rs&rAis_symnode_likeVariableTracker.is_symnode_liker$rDcV^8dQhRR/#rr?)r@s"rArBrs4rDc R#)z(Return True for TensorVariable instancesFr?rs&rA is_tensorVariableTracker.is_tensorr$rDcV^8dQhRR/#)r=r>type[VariableTracker]r?)r@s"rArBrs  /D rDc \V4#)aReturn the VariableTracker type to use for builtin handler dispatch. This is used by BuiltinVariable to look up the appropriate handler for a given set of arguments. Most VariableTrackers just return their own type, but LazyConstantVariable returns ConstantVariable since it can be treated as a constant for most builtin operations. Subclasses that override this should install appropriate guards to ensure the dispatched handler remains valid. )rrs&rAget_handler_type_for_dispatch-VariableTracker.get_handler_type_for_dispatchsDzrDcV^8dQhRR/#)r=r>z object | Noner?)r@s"rArBrs}rDc R#)zReturn the wrapped Python object for generic STORE_ATTR mutation, or None to decline. Only override for VTs with __dict__ and standard __setattr__.Nr?rs&rAget_value_for_setattr%VariableTracker.get_value_for_setattrsrDc$V^8dQhRRRRRR/#r=rr-rr3r>rr?)r@s"rArBr$  + 36   rDc R#)aLook up *name* in the instance __dict__ (tp_dictoffset equivalent). Returns a VT if the attribute exists in the instance dict, None if the attribute is absent or the VT has no instance dict at all. UDOV overrides to check self.value.__dict__ + side effects. Nr?r,s&&&rAlookup_instance_dict$VariableTracker.lookup_instance_dictrDc$V^8dQhRRRRRR/#rAr?)r@s"rArBrrBrDc R#)zCall __getattr__ fallback (step 6 of GenericGetAttr). Returns a VT if the type defines __getattr__ and it succeeds, None otherwise. The base returns None (most types have no __getattr__). UDOV overrides to walk the MRO for __getattr__. Nr?r,s&&&rAcall_getattr_fallback%VariableTracker.call_getattr_fallbackrFrDc$V^8dQhRRRRRR/#)r=rr-rr3r>rr?)r@s"rArBrs),G,G+,G36,G ,GrDc VPV4pVeVPW4pVeV#VR8Xd$\PWP 44#VP 4pVe^RIHpHpV!WV4#VPW4p\PPV4'g\ hVP;'d\VPV4p V 'd?VP4'g)\!V P#\$P&44\PP)WR7# \ dRpLi;i TdLi;i)zDefault attribute access via object_generic_getattr. Tries the MRO-based descriptor protocol first (PyObject_GenericGetAttr), falls back to const_getattr for VTs without a known python_type or when the descriptor protocol hits an unhandled type. Nr{)_UnhandledDescriptorErrorobject_generic_getattr)rL)rrrrrrr:rMrNr-rr. is_literalrLrrrr'rCONSTANT_MATCHcreate) rJrrgetsetrKpy_typerMrNr<rLs &&& rArw VariableTracker.tp_getattro_implsA--d3  ]]4,F!  ; "((-=-=-?@ @ &&(G     -b==""2,))44U;;% %>>DKK!> $1133 &++L,G,GH I))000FF+# G -  s$E 5E EEE'&E'c V^8dQhRRRR/#)r=rr-r>zvariables.DunderDictVariabler?)r@s"rArBrs  +  % rDc ~VPf%\PPW4VnVP#rx)rrDunderDictVariablerQr s&&rA get_dict_vtVariableTracker.get_dict_vts0 << $77>>rHDL||rDcV^8dQhRR/#rr?)r@s"rArBrs$rDc LVP4R# \dR#i;ir)as_proxyrrs&rAis_proxyVariableTracker.is_proxys%  MMO"  r cV^8dQhRR/#rr?)r@s"rArBrs--#-rDc *\\V44hrx)rr3rs&rAr\VariableTracker.as_proxys!#d),,rDcV^8dQhRR/#)r=r>z Node | Noner?)r@s"rArBrs  { rDc VP4p^RIp\WPP4'd VP #R# \ dR#i;i)rbN)r\torch.fxrfxProxynoder)rJproxyrs& rA maybe_fx_nodeVariableTracker.maybe_fx_nodesG MMOE %00zz!"  sAA AAcV^8dQhRR/#rr?)r@s"rArBrs$rDc aaRoRVV3RllpSPP4F-wr#VSP9gK\P W4K/ S#)zBCheck if this variable references itself (directly or indirectly).Fc V^8dQhRRRR/#)r=rrr>r_r?)r@s"rArB>VariableTracker._contains_self_reference..__annotate__s " "o "$ "rDc<VSJdRoR#R#)TNr?)r found_selfrJs&rAcheck7VariableTracker._contains_self_reference..checksTz! rD)rrrrr)rJrqrNrrpsf @rA_contains_self_reference(VariableTracker._contains_self_referencesQ  " " "]]002MC$---%%e63rDc V^8dQhRRRR/#)r=codegenr+r>r_r?)r@s"rArBrs""9""rDc \hrxr+rJrvs&&rA reconstructVariableTracker.reconstructs!!rDc V^8dQhRRRR/#)r=rvr+r>r3r?)r@s"rArBrsTT)TTrDc \RV 24h)z'reconstruct_pycode not implemented for r+rxs&&rAreconstruct_pycode"VariableTracker.reconstruct_pycodes!$KD6"RSSrDc V^8dQhRRRR/#)r=rr-r>rr?)r@s"rArBrs""+" "rDc \hrxr+r s&&rAunpack_var_sequence#VariableTracker.unpack_var_sequences "!rDc$V^8dQhRRRRRR/#)r=rr-rr3r>zConstantVariable | Noner?)r@s"rArBrs$  + 36   rDc "VPPPW4'ddVPPPWRR7p\P P \V\P4'*4#R#)zGIf *name* has a pending mutation, return the hasattr result; else None.T) deleted_okN) rrhas_pending_mutation_of_attr load_attrrr.rQrDeletedVariable)rJrrr<s&&& rA_hasattr_check_side_effects+VariableTracker._hasattr_check_side_effectsso 99 ! ! > >t J JII**44TD4QE--44ui&?&?@@ rDc$V^8dQhRRRRRR/#)r=rr-rr3r>r.r?)r@s"rArBr)s$  + 36  rDc  VPW4pVeV#VPW4\PP R4# \ d=TP P4\PP R4u#\\3dMi;i\RRT RT 2RTP4 R2R TPP R T R 2.\POR 7R#) a9Dynamo's hasattr(): try tp_getattro_impl, catch AttributeError. Mirrors CPython's PyObject_HasAttr (via PyObject_GetOptionalAttr): call tp_getattro, suppress AttributeError via PyErr_Clear, return True/False. https://github.com/python/cpython/blob/848cb25624ab44c9fef2966c777419376b65af1b/Objects/object.c#L1346 NTFzUnsupported hasattr callzcall_obj_hasattr rc0Dynamo does not know how to trace the function `rdzAvoid calling `hasattr(, z)` in your code.re)rrwrr.rQr exn_vt_stackclear_current_exceptionrrrrr{rNr SUPPORTABLE)rJrrrKs&&& rAcall_obj_hasattr VariableTracker.call_obj_hasattr)s11";  M   ! !" +--44T: :% < OO 3 3 5--44U; ;#[1    .'vQtf5J4??K\J]]^_)$..*A*A)B"TFJZ["..  s/A AB"B"B"!B"c V^8dQhRRRR/#r=rr-r>rr?)r@s"rArBrJs   8  _  rDc \RRVP4 2RVP4 R2.\POR7R#)zf Implements PyObject_GetIter semantics (tp_iter slot). Subclasses override this to support iteration. zmissing tp_iterz!tp_iter_impl not implemented for z*Dynamo does not know how to iterate over ``.reNrrrrrr s&&rAr VariableTracker.tp_iter_implJsG %78M8M8O7PQDT__EVDWWYZ2%112  rDc(V^8dQhRRRRRRRR/# r=rr-rrrrr>rr?)r@s"rArBrYs2   %  $  +    rDc \RRVP4 2RVP4 R2.\POR7R#)z>tp_init slot (__init__). VTs override to initialize instances.zmissing tp_initz!tp_init_impl not implemented for z/Dynamo does not know how to trace __init__ on `rreN)rrrrrk)rJrrrs&&&&rArVariableTracker.tp_init_implYsG %78M8M8O7PQI$//J[I\\^_1%001  rDc(V^8dQhRRRRRRRR/#rr?)r@s"rArBrgs2  % $ +    rDc  6^RIHpVP4pVe-V!V4'g\ VRVP 4 R24\ RRV RV RV 2RVP4 R 2R VP4 R 2R .R 7R# \dRpLi;i)r)pycallable_checkNr z' object is not callablezUnsupported function callzcall_function rcrrdAvoid calling `` in your code."Please report an issue to PyTorch.re)r:rrrrrrr)rJrrrrrs&&&& rArVariableTracker.call_functiongs 6 '')H  (8(B(B R1T%:%:%<$==U!V W/$TF!D66(;J4??K\J]]^_!$//"3!4OD4  # H sB BBc V^8dQhRRRR/#rr?)r@s"rArBr  !:  rDc P\\VRVP4 R2.R7R#)z)Called when sq_length is not implemented.object of type '' has no len()rNr TypeErrorrr s&&rArpVariableTracker.sq_length_impl)  $T%:%:%<$=^LM rDc V^8dQhRRRR/#rr?)r@s"rArBrrrDc P\\VRVP4 R2.R7R#)z)Called when mp_length is not implemented.rrrNrr s&&rArjVariableTracker.mp_length_implrrDc$V^8dQhRRRRRR/#r=rr-rNrr>r?)r@s"rArBrs(  %     rDc \RR\V4P 2RVP4 R2.\P OR7R#)missing_mp_subscriptz"mp_subscript_impl not defined for z*Dynamo does not yet support subscripting ''.reNrrrNrrrrJrrNs&&&rArl!VariableTracker.mp_subscript_implsJ *8d9L9L8MNDTEZEZE\D]]_`2%112  rDc$V^8dQhRRRRRR/#rr?)r@s"rArBrs(   %       rDc 6\RRV RV 2RV 2.R7R#)z!unsupported __getitem__ (sq_item)z sq_item_impl rcz.Dynamo does not know how to handle sq_item on reNrrs&&&rArsVariableTracker.sq_item_impls, 7#D63%0HO  rDc$V^8dQhRRRRRR/#)r=rr-itemrr>r?)r@s"rArBrs$   +  3B    rDc  \RRVP4 2RVP4 RVP4 R2.\POR7R#)z+Called when sq_contains is not implemented.zmissing sq_containsz%sq_contains_impl not implemented for z&Dynamo does not know how to check if `z ` is in `rreNr)rJrrs&&&rArv VariableTracker.sq_contains_implsY );Dr?)r@s"rArBr2   %    &    rDc \RRVP4 2RVP4 R2.\POR7R#)missing_mp_ass_subscriptz&mp_ass_subscript_impl not defined for z0Dynamo does not yet support item assignment on 'rreNrrrrrJrrNr<s&&&&rArn%VariableTracker.mp_ass_subscript_implsI .r?)r@s"rArBr(   %       rDc  \RR\V4P 2RVP4 RVP4 R2.\P OR7R#)z7Sequence concatenation via + operator (sq_concat slot).zmissing sq_concatzsq_concat not defined for z+Dynamo does not yet support concatenating '' and 'r reNrrJrrs&&&rArqVariableTracker.sq_concat_implsc '0d1D1D0EFEdF[F[F]E^^efkf|f|f~e@AB2%112  rDc$V^8dQhRRRRRR/#rr?)r@s"rArBrrrDc  \RR\V4P 2RVP4 RVP4 R2.\P OR7R#)zIIn-place sequence concatenation via += operator (sq_inplace_concat slot).zmissing sq_inplace_concatz"sq_inplace_concat not defined for z/Dynamo does not yet support inplace concat of 'rr reNrrs&&&rArw&VariableTracker.sq_inplace_concat_implss /8d9L9L8MNI$J_J_JaIbbijokAkAkCjDDEF2%112  rDc ,V^8dQhRRRRRRRRR R /#) r=rr-rr3rrrrr>rr?)r@s"rArBrsCQ Q %Q Q $ Q + Q  Q rDc  VPV4pVeV!WW#V4pVeV#VPPV4pVe V!WW44#VR8XdV'gV'g^RIHpV!W4#VR8Xd\ V4^8XdV'gVP 4;p 'drV^,P 4;p 'dS\P!V RR4\PJd)\PP\W44#VR8Xdx\ V4^8XdhV'g`VP4'dJV^,P 4p \ P#WP 4P%V 44#RVP'4 RV R2R.p \)V\P*4'dfVR9d_\)VP,\.\0\234'dV P5R 4V P5R 4V P5R 4\7R R V RV RV RV 2RV RVP'4 R2V R7R#)Nr) generic_hash__subclasscheck__ __reduce_ex__rrrrzConsider moving the creation of dict view object (e.g. `dict.keys()`, `dict.items()`,) to the compiled region, instead of passing it as an input to the compiled region.a Dynamo does not fully support tracing builtin iterators (e.g. `map`, `zip`, `enumerate`) passed in from uncompiled to compiled regions (e.g. `torch.compile(fn)(enumerate(...))`). This can happen unintentionally if a previous graph break happens with a builtin iterator in the local scope.aFList/dict comprehensions in Python <= 3.11 result in implicit function calls, which Dynamo cannot trace as a top level frame. Possible workarounds are (1) use a loop instead of a comprehension, (2) fix any graph breaks in the function above the comprehension, (3) wrap the comprehension in a function, or (4) use Python 3.12+.zUnsupported method callz call_method rcz*Dynamo does not know how to trace method `z ` of class `rdre)r r )r _slotdefsrr:rrrsinspectgetattr_staticrrrr.rQ issubclassrrrrrrUserDefinedObjectVariabler<rrrrr) rJrrrrmethodrKslotdefrself_py derived_pyprotocolris &&&&& rA call_methodVariableTracker.call_methodsE&&t,  Dd&9F! ..$$T*  4T2 2 : d6 5) ) & &3t9>&22444"1g88:: :**74GN--.%55<<":7 O #D Q''))Aw113H"((++-;;HE  d3356av_ M 0  dI?? @ @TN F $**xJ&GHH h LL&  LL5  -"4&$qax@DTF,W[WlWlWnVoopq  rDc 0V^8dQhRRRRRRRRR R R R/# r=rr- tree_map_fnr/map_fnrrestrtree_map_kwargsrr>r?)r@s"rArBrD sF  % *   $  4    rDc PVPR4pVeXVP4'gBVPW./4pVP4pT'dTPY.TO/4#VP VVVVV4# \dTP TTTTT4u#i;i)zLPerformance optimization to implement optree.tree_map faster than tracing itis_leaf)rr"rrsr_tree_map_fallbackcall_tree_map_branch) rJrrrrr is_leaf_var pred_result leaf_decisions &&&&&& rA call_tree_mapVariableTracker.call_tree_mapD s&)))4  ";+G+G+I+I%33BCK  + > > @ ++B  rBB((        ' ..#  sB B%$B%c 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBre sF  % *   $  4    rDc ,VPVVVVV4#)zKEmulate optree.tree_map without is_leaf/none_is_leaf checks (handled above))r)rJrrrrrs&&&&&&rAr$VariableTracker.call_tree_map_branche s&&&        rDc 0V^8dQhRRRRRRRRR R R R/#rr?)r@s"rArBrv sF  % *   $  4    rDc VP4pRVn\PRVVV4VP VW0.VOV4#)cR#rxr?rrs*,rAr4VariableTracker._tree_map_fallback.. QUrDz@tree_map fastpath fallback triggered for %s (rest=%s, kwargs=%s)r_maybe_call_tree_map_fastpathlogdebugr)rJrrrrrtree_map_fn_copys&&&&&& rAr"VariableTracker._tree_map_fallbackv sZ',,.9U6 N      --   !D !   rDc4V^8dQhRRRRRRRRR R R R R R/#r=rr-rr/rrrrrrkeypathztuple[Any, ...]r>r?)r@s"rArBr sP% % %% *%  % $ % 4 % !%  % rDc  VPR4pVeVP4'g|VPW./4pVP4p T 'dP\ P!TU u.uFp \PY4NK up 4p TPYT.TO/4#VPVVVVVV4# \dTP TTTTTT4u#i;iuup i)zOPerformance optimization to implement tree_map_with_path faster than tracing itr) rr"rrsr_tree_map_with_path_fallbackr TupleVariablerrcall_tree_map_with_path_branch) rJrrrrrrrrrk keypath_vars &&&&&&& rAcall_tree_map_with_path'VariableTracker.call_tree_map_with_path s&)))4  ";+G+G+I+I%33BCK  + > > @ '55;BC7a_**217C ++Bd0JT0JBOO22         ' 88#  DsB=,C$=!C! C!c4V^8dQhRRRRRRRRR R R R R R/#rr?)r@s"rArBr sY H H % H* H H $ H 4 H! H  HrDc  \P!VUu.uFp\PW4NK up4pVP WV.VO/4#uupi)z;Handle tree_map_with_path for leaf nodes (default behavior))rr rrr) rJrrrrrrr rs &&&&&&& rAr .VariableTracker.call_tree_map_with_path_branch sU --3: ;7a_ " "2 )7 ; ##Bd(BT(BBGG VariableTracker._tree_map_with_path_fallback.. rrDzVtree_map_with_path fastpath fallback triggered for %s (rest=%s, kwargs=%s, keypath=%s)r)rJrrrrrrrs&&&&&&& rAr ,VariableTracker._tree_map_with_path_fallback s]',,.9U6 d       --   !D !   rDc V^8dQhRRRR/#)r=rr3r>r_r?)r@s"rArBr s  # $ rDc R#rxr?rs&&rA set_name_hintVariableTracker.set_name_hint s rDc V^8dQhRRRR/#)r=rr2r>r_r?)r@s"rArBr s//>/d/rDc WnR#rx)r)rJrs&&rAset_source_location#VariableTracker.set_source_location s.rDcV^8dQhRR/#r=r>rr?)r@s"rArBr srDc V#)z=Used by LazyVariableTracker to build the real VariableTrackerr?rs&rArealizeVariableTracker.realize  rDcV^8dQhRR/#r"r?)r@s"rArBr srDc V#)zSUsed by LazyVariableTracker to return the real VariableTracker if it already existsr?rs&rArVariableTracker.unwrap r&rDcV^8dQhRR/#rr?)r@s"rArBr sTrDc R#)z:Used by LazyVariableTracker to indicate an unrealized nodeTr?rs&rA is_realizedVariableTracker.is_realized srDc V^8dQhRRRR/#rr?)r@s"rArBr s  #<  rDc \RRVP4 R2RVP4 R2.\POR7R#)u Implements tp_iternext slot semantics. Subclasses override this to support next(). Reaching this base is a bug — it means tp_iternext is missing for that VariableTracker subclass. zMissing tp_iternextznext()z"Dynamo does not support next() on z8. Add tp_iternext_impl to this VariableTracker subclass.reNrr s&&rAr VariableTracker.tp_iternext_impl sT )D1134A64T5J5J5L4MNJJ3%112 rDc V^8dQhRRRR/#rr?)r@s"rArBr s)) 9)o)rDc $VPV4#rx)rr s&&rA next_variableVariableTracker.next_variable s$$R((rDc V^8dQhRRRR/#)r=rr-r>rr?)r@s"rArBr sGG!:GtGrDc ^\VP;'dVPV44#rx)rstrict_checks_fnr s&&rAis_strict_modeVariableTracker.is_strict_mode s&B''EEB,?,?,EFFrDcV^8dQhRR/#rr?)r@s"rArBr s''D'rDc ,VP4'*#)z0Whether Dynamo allows mutation on this variable.) is_immutablers&rA is_mutableVariableTracker.is_mutable s$$&&&rDcV^8dQhRR/#rr?)r@s"rArBr s**d*rDc VPRJ#)z.Whether Dynamo bans mutation on this variable.N)rrs&rAr=VariableTracker.is_immutable s!!T))rDc ,V^8dQhRRRRRRRRR R/#) r=rr-r<r rLrr$rr>r?)r@s"rArBr sANN %NNN N NrDc `Vf \PPW4#V'd\P!W4!V4#\ V4\ P P9d \ P PW4#\ PPWVR7#)z=Create a new VariableTracker from a value and optional Source)r) builderSourcelessBuilderrQVariableBuilderrrLazyConstantVariablesupported_typesr)rr<rLr$s&&&&rArVariableTracker.build s >,,33B> > **26u= = %[I::JJ J1188G G0077"7M MrDc V^8dQhRRRR/#)r=rr-r>r7r?)r@s"rArBr s  2 z rDc VP'd/\VPPVP44#R#)utReturn id() of the underlying Python object, or None if unavailable. The base implementation uses source resolution for sourceful VTs. Subclasses override for special cases (e.g. NNModuleVariable uses get_submodule, ConstantVariable handles singletons). Returns None for sourceless VTs — callers use FakeIdVariable in that case (see generic_id in object_protocol.py). N)rLrrrr s&&rAget_idVariableTracker.get_id s. ;;;bii44T[[AB BrDcV^8dQhRR/#)r=r>zCallable[..., Any] | Noner?)r@s"rArBr, s#<rDc JVP'd\P#R#rx)rLrID_MATCHrs&rAget_id_guard_type!VariableTracker.get_id_guard_type, s ;;;(( (rDcV^8dQhRR/#)r=r>rr?)r@s"rArBr1 s " "f "rDc RVP4# \d \u#i;i)zReturn the Python object this VT wraps, for `is` comparison. Returns NO_SUCH_SUBOBJ if no concrete Python object is available. )rsrrrs&rAget_real_python_backed_value,VariableTracker.get_real_python_backed_value1 s+  "**, ," "! ! "s &&c V^8dQhRRRR/#rr?)r@s"rArBr<   %   rDc P\\VRVP4 R2.R7R#)aMirrors CPython's PyNumber_Index / nb_index slot. https://github.com/python/cpython/blob/c09ccd9c429/Objects/abstract.c#L1411-L1450 The base implementation raises TypeError, matching CPython's behavior when tp_as_number->nb_index is NULL (_PyIndex_Check fails). r z,' object cannot be interpreted as an integerrNrr s&&rArcVariableTracker.nb_index_impl< s. !  D))+,,XY rDc V^8dQhRRRR/#rr?)r@s"rArBrO rYrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_repr slot. https://github.com/python/cpython/blob/v3.13.3/Objects/object.c#L745-L778 Called when type_implements_tp_repr returns True for this type. Subclasses override to provide the actual repr implementation. ztp_repr_impl not implementedz. has tp_repr slot but no tp_repr_impl override The type z[ has a tp_repr C slot but the corresponding VariableTracker doesn't implement tp_repr_impl.reNrr s&&rArVariableTracker.tp_repr_implO sQ 2Dz**++YZ#D$9$9$;#<=PP2%112  rDc V^8dQhRRRR/#rr?)r@s"rArBrb s  %   rDc  \R\V4P RVP4 2RVP4 R\V4P R2.\P OR7R#)zDynamo hook for VariableTrackers with dedicated str behavior. Subclasses override this for the per-type str result; generic_str() handles dispatch and repr fallback. ztp_str_impl not implementedz! has no tp_str_impl override for z&Dynamo does not implement __str__ for z in rreNrr s&&rArVariableTracker.tp_str_implb sp 1Dz**++LTMbMbMdLef@AVAVAX@YZt*%%&a)2%112  rDc V^8dQhRRRR/#rr?)r@s"rArBrr   %   rDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_int slot. Called when type_implements_nb_int returns True for this type. Subclasses override to provide the actual conversion. znb_int_impl not implementedz, has nb_int slot but no nb_int_impl overrider^zZ has an nb_int C slot but the corresponding VariableTracker doesn't implement nb_int_impl.reNrr s&&rArAVariableTracker.nb_int_implr sQ 1Dz**++WX#D$9$9$;#<=OO2%112  rDc V^8dQhRRRR/#rr?)r@s"rArBr rdrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_float slot. Called when type_implements_nb_float returns True for this type. Subclasses override to provide the actual conversion. znb_float_impl not implementedz0 has nb_float slot but no nb_float_impl overrider^z^ has an nb_float C slot but the corresponding VariableTracker doesn't implement nb_float_impl.reNrr s&&rArCVariableTracker.nb_float_impl sQ 3Dz**++[\#D$9$9$;#<=QQ2%112  rDc$V^8dQhRRRRRR/#)r= tp_slot_namer3rrr(rr?)r@s"rArBr s(         rDc  \RV RVP4 RVP4 RV R2RVP4 RV 2.\POR7R#) z!tp_as_number slot not implemented(rz , reverse=r0r^z does not implement the reNr)rJrkrr(s&&&&rA_nb_slot_not_implemented(VariableTracker._nb_slot_not_implemented so 7#nAd&;&;&=%>bAWAWAY@ZZdeldmmno#D$9$9$;#<r?)r@s"rArBr 2 : : % : : :  :rDc 6\P!\4#)ztp_as_number->nb_lshift slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rlshift__`` instead of ``__lshift__``). rr.NotImplementedrJrrr(s&&&&rAr2VariableTracker.nb_lshift_impl )).99rDc$V^8dQhRRRRRR/#rr?)r@s"rArBr (:: %::  :rDc 6\P!\4#)zFtp_as_number->nb_inplace_lshift slot. Default: returns NotImplemented.rtrs&&&rArO&VariableTracker.nb_inplace_lshift_impl  )).99rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_rshift slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rrshift__`` instead of ``__rshift__``). rtrvs&&&&rAr5VariableTracker.nb_rshift_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zFtp_as_number->nb_inplace_rshift slot. Default: returns NotImplemented.rtrs&&&rArQ&VariableTracker.nb_inplace_rshift_impl r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_and slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rand__`` instead of ``__and__``). rtrvs&&&&rAr8VariableTracker.nb_and_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zCtp_as_number->nb_inplace_and slot. Default: returns NotImplemented.rtrs&&&rArS#VariableTracker.nb_inplace_and_impl r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_xor slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rxor__`` instead of ``__xor__``). rtrvs&&&&rAr;VariableTracker.nb_xor_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zCtp_as_number->nb_inplace_xor slot. Default: returns NotImplemented.rtrs&&&rArU#VariableTracker.nb_inplace_xor_impl r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_floor_divide slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rfloordiv__`` instead of ``__floordiv__``). rtrvs&&&&rArY$VariableTracker.nb_floor_divide_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zLtp_as_number->nb_inplace_floor_divide slot. Default: returns NotImplemented.rtrs&&&rAr_,VariableTracker.nb_inplace_floor_divide_impl r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_true_divide slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rtruediv__`` instead of ``__truediv__``). rtrvs&&&&rAr\#VariableTracker.nb_true_divide_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zKtp_as_number->nb_inplace_true_divide slot. Default: returns NotImplemented.rtrs&&&rAra+VariableTracker.nb_inplace_true_divide_impl r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_remainder slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rmod__`` instead of ``__mod__``). rtrvs&&&&rAr!VariableTracker.nb_remainder_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr, rzrDc 6\P!\4#)zItp_as_number->nb_inplace_remainder slot. Default: returns NotImplemented.rtrs&&&rArK)VariableTracker.nb_inplace_remainder_impl, r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr4 2 : : % : : :  :rDc 6\P!\4#)ztp_as_number->nb_divmod slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rdivmod__`` instead of ``__divmod__``). divmod has no in-place form. rtrvs&&&&rAr"VariableTracker.nb_divmod_impl4 )).99rDc ,V^8dQhRRRRRRRRR R/#) r=rr-rrr/rr(rr>r?)r@s"rArBrB s<:: %:: " :  :  :rDc 6\P!\4#rxrt)rJrrr/r(s&&&&&rAr%VariableTracker.nb_power_implB )).99rDc(V^8dQhRRRRRRRR/#)r=rr-vrwr>r?)r@s"rArBrK s2 : : % :  :  :  :rDc 6\P!\4#)ztp_as_number->nb_power slot when self is the modulus (z) operand. ``v`` is the base, ``w`` is the exponent, ``self`` is the modulus. rt)rJrrrs&&&&rAnb_power_z_implVariableTracker.nb_power_z_implK s)).99rDc(V^8dQhRRRRRRRR/#)r=rr-rrr/rr>r?)r@s"rArBrW s2:: %:: " :  :rDc 6\P!\4#rxrt)rJrrr/s&&&&rArM%VariableTracker.nb_inplace_power_implW r}rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr_ s8 Y Y % Y Y Y  YrDc (VPRW#R7#)a>tp_as_number->nb_multiply slot. Default: graph-breaks. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rmul__`` instead of ``__mul__``). The base default graph-breaks so unmigrated VTs surface loudly; ``binary_op1`` only invokes this on types whose CPython type advertises the slot. rr'rnrvs&&&&rAr VariableTracker.nb_multiply_impl_ s,,-?,XXrDc$V^8dQhRRRRRR/#rr?)r@s"rArBrn s-PP %PP  PrDc &VPRV4#)z>tp_as_number->nb_inplace_multiply slot. Default: graph-breaks.rIrrs&&&rArI(VariableTracker.nb_inplace_multiply_impln s ,,-GOOrDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBrv s2   %        rDc (VPRW#R7#)ztp_as_number->nb_matrix_multiply slot. Default: graph-breaks. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rmatmul__`` instead of ``__matmul__``). rer'rrvs&&&&rAre'VariableTracker.nb_matrix_multiply_implv s!,, %u-  rDc$V^8dQhRRRRRR/#rr?)r@s"rArBr s-WW %WW  WrDc &VPRV4#)zEtp_as_number->nb_inplace_matrix_multiply slot. Default: graph-breaks.rhrrs&&&rArh/VariableTracker.nb_inplace_matrix_multiply_impl s ,,-NPUVVrDc$V^8dQhRRRRRR/#r=rr-countrr>r?)r@s"rArBr s(  %     rDc \R\V4P R2RVP4 R2.\P OR7R#)ztp_as_sequence->sq_repeat slot. Implements sequence repetition (e.g. ``[1, 2] * 3``). ``count`` is an int VariableTracker already produced by ``nb_index_impl``. zsq_repeat_impl not implementedz2 has sq_repeat slot but no sq_repeat_impl overrider^z` has an sq_repeat C slot but the corresponding VariableTracker doesn't implement sq_repeat_impl.reNrrJrrs&&&rArrVariableTracker.sq_repeat_impl sQ 4Dz**++]^#D$9$9$;#<=RR2%112  rDc$V^8dQhRRRRRR/#rr?)r@s"rArBr s(   %       rDc \R\V4P R2RVP4 R2.\P OR7R#)z'tp_as_sequence->sq_inplace_repeat slot.z&sq_inplace_repeat_impl not implementedzB has sq_inplace_repeat slot but no sq_inplace_repeat_impl overrider^zp has an sq_inplace_repeat C slot but the corresponding VariableTracker doesn't implement sq_inplace_repeat_impl.reNrrs&&&rArx&VariableTracker.sq_inplace_repeat_impl sQ <Dz**++mn#D$9$9$;#<=ZZ2%112  rDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_or slot. Default: returns NotImplemented. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__ror__`` instead of ``__or__``). rtrvs&&&&rAr>VariableTracker.nb_or_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)zBtp_as_number->nb_inplace_or slot. Default: returns NotImplemented.rtrs&&&rArW"VariableTracker.nb_inplace_or_impl r}rDc V^8dQhRRRR/#rr?)r@s"rArBr rdrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_negative slot. Called when type_implements_nb_negative returns True for this type. Subclasses override to provide the actual negation. z nb_negative_impl not implementedz6 has nb_negative slot but no nb_negative_impl overrider^zd has an nb_negative C slot but the corresponding VariableTracker doesn't implement nb_negative_impl.reNrr s&&rAr( VariableTracker.nb_negative_impl Q 6Dz**++ab#D$9$9$;#<=TT2%112  rDc V^8dQhRRRR/#rr?)r@s"rArBr rdrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_positive slot. Called when type_implements_nb_positive returns True for this type. Subclasses override to provide the actual positive. z nb_positive_impl not implementedz6 has nb_positive slot but no nb_positive_impl overrider^zd has an nb_positive C slot but the corresponding VariableTracker doesn't implement nb_positive_impl.reNrr s&&rAr* VariableTracker.nb_positive_impl rrDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrDc 6\P!\4#)ztp_as_number->nb_add slot. Default: graph break. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__radd__`` instead of ``__add__``). rtrvs&&&&rArVariableTracker.nb_add_impl rrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr s(:: %::  :rDc 6\P!\4#)z8tp_as_number->nb_inplace_add slot. Default: graph break.rtrs&&&rArE#VariableTracker.nb_inplace_add_impl rrDc(V^8dQhRRRRRRRR/#rqr?)r@s"rArBr rrrDc 6\P!\4#)ztp_as_number->nb_subtract slot. Default: graph break. ``reverse=True`` means self is the right-hand operand (CPython would look up ``__rsub__`` instead of ``__sub__``). rtrvs&&&&rAr VariableTracker.nb_subtract_impl rxrDc$V^8dQhRRRRRR/#rr?)r@s"rArBr rzrDc 6\P!\4#)z=tp_as_number->nb_inplace_subtract slot. Default: graph break.rtrs&&&rArG(VariableTracker.nb_inplace_subtract_impl r}rDc V^8dQhRRRR/#rr?)r@s"rArBr rdrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_absolute slot. Called when type_implements_nb_absolute returns True for this type. Subclasses override to provide the actual absolute value. z nb_absolute_impl not implementedz6 has nb_absolute slot but no nb_absolute_impl overrider^zd has an nb_absolute C slot but the corresponding VariableTracker doesn't implement nb_absolute_impl.reNrr s&&rAr, VariableTracker.nb_absolute_impl rrDc V^8dQhRRRR/#rr?)r@s"rArBr! rdrDc \R\V4P R2RVP4 R2.\P OR7R#)zMirrors CPython's tp_as_number->nb_invert slot. Called when type_implements_nb_invert returns True for this type. Subclasses override to provide the actual bitwise inversion. znb_invert_impl not implementedz2 has nb_invert slot but no nb_invert_impl overrider^z` has an nb_invert C slot but the corresponding VariableTracker doesn't implement nb_invert_impl.reNrr s&&rAr0VariableTracker.nb_invert_impl! sQ 4Dz**++]^#D$9$9$;#<=RR2%112  rDc ,V^8dQhRRRRRRRRR R /#) r=rLrrzMutationType | NonerzSourceLocation | Nonerrr>r_r?)r@s"rArBr2 s>(((+ ( / ( 5 ( (rDc ><\SV`4WnW0nW nW@nVer\ V\\34'dVe \R4hR#\ V\\34'g\R\V4 24hVf \R4hR#R#)Nz=source must be None for ValueMutationNew/AttributeMutationNewzAExpected ValueMutationExisting or AttributeMutationExisting, got zKsource must not be None for ValueMutationExisting/AttributeMutationExisting) ryrlrLrrrrrsrAssertionErrorrrr)rJrLrrrr{s&$$$$rArlVariableTracker.__init__2 s  .*   $-*:(e") %rDcV^8dQhRR/#)r=r>rr?)r@s"rArBr] s555rDc *\\V44#rx)rrrs&rAtp_typeVariableTracker.tp_type\ s-d344rDcV^8dQhRR/#)r=r>rr?)r@s"rArBra sBBoBrDc >\\V44P#rx)rrrrs&rArVariableTracker.tp_as_number` s-d34AAArDcV^8dQhRR/#)r=r>rr?)r@s"rArBre sDD 1DrDc >\\V44P#rx)rrrrs&rArVariableTracker.tp_as_sequenced s-d34CCCrDcV^8dQhRR/#)r=r>rr?)r@s"rArBri sCC/CrDc >\\V44P#rx)rrrrs&rArVariableTracker.tp_as_mappingh s-d34BBBrDcV^8dQhRR/#r=r>rr?)r@s"rArBrm ===rDc >\\V44P#rx)rrrrs&rArVariableTracker.tp_iterl -d34<<\\V44P#rx)rrrrs&rArVariableTracker.tp_iternextp s-d34@@@rDcV^8dQhRR/#rr?)r@s"rArBru s<< \\V44P#rx)rrrrs&rArVariableTracker.tp_strt s-d34;;;rDcV^8dQhRR/#rr?)r@s"rArBry rrDc >\\V44P#rx)rrrrs&rArVariableTracker.tp_reprx rrDc *\\V44#rx)r~rrs&rArVariableTracker._slotdefs| s !6t!<==rDc V^8dQhRRRR/#)r=rr r>r_r?)r@s"rArBr s"V"V"V"VrDc <\SV`!R/VBRRlp\PVRV4RRlp\PVRV4R#) a Wraps all subclasses' `as_python_constant` and `reconstruct` so that it cannot be called twice in the same call chain - i.e. self-referential objects. For `as_python_constant` - self-referential objects are NOT treated as constants. For `reconstruct` - we will graph break. The graph break can be avoided if the VT subclass can generate and cache itself before recursively `reconstruct`ing - see ListVariable for an example. cV^8dQhRR/#r=r>r r?)r@s"rArB7VariableTracker.__init_subclass__..__annotate__ s   rDc \W R2R7h)z is self-referential)rrrs&rAas_python_constant_failureEVariableTracker.__init_subclass__..as_python_constant_failure s5F"67 rDrscV^8dQhRR/#rr?)r@s"rArBr s   rDcf\R\V4RV R2RR.\PORNR7R#)z)Reconstruction failure (self-referential)z0Dynamo tried to reconstruct sourceless variable z~, but it is self-referential. Dynamo must manually implement reconstruction rules for self-referentiable sourceless variables.zIf Dynamo is attempting to trace a return statement and your code is attempting to return a variable that Dynamo cannot reconstruct, then remove it from the return statement.zhRemove the self-reference in the variable. A self-referring list, for example, is `l = []; l.append(l)`.zPReport an issue to PyTorch if you need self-referential reconstrtuction support.reN)rr3rCAUSED_BY_EARLIER_GRAPH_BREAKrs&rAreconstruct_failure>VariableTracker.__init_subclass__..reconstruct_failure sQ CD NtfUss`~'DD  g  rDryNr?)ry__init_subclass__r_add_call_once_guard)rrrrr{s&, rAr!VariableTracker.__init_subclass__ sM !+F+  ,, %'A   ,,S-ATUrDc(V^8dQhRRRRRRRR/#) r=rr8rr3callbackz Callable[[VariableTracker], Any]r>r_r?)r@s"rArBr s0-- "--3-  -rDc aa\W4oS\\V4Jg\SR4'dR#\P!S4RVV3Rll4pRVn\ WV4R#)_call_once_guardedNc$V^8dQhRRRRRR/#)r=rr rr>rr?)r@s"rArB:VariableTracker._add_call_once_guard..__annotate__ s! $ $ $s $ $rDcD<\P4pVf \4p\PV4\V4\S43pWC9d S!V4VP V4S!V.VO5/VBVP V4# TP T4i;irx)rUrsetradddiscard)rJrractiverNroriginal_methods&*, rAguarded_method.guarded_method s%))+F~ $$V,d8R01C} JJsO $&t=d=f=s#s#s / B BT)rrhasattr functoolswrapsrsetattr)rrrr'r&s&&f @rAr$VariableTracker._add_call_once_guard sk "#. gov> >' 1C C    ) $ $ * $ -1)^,rD)rrrLr>r<guardsrLrrparents_trackerruser_code_variable_name)NN)NFr)rNrOrPrQrRrrSrrrrrrrrrr classmethodrrrrrrrsrrr.rrr rr"r'r-r1r5r:r>rDrIrwrXr]r\rirsryr}rrrr rrrprjrlrsrvrnrtrqrwrrrrrr r rrr$rr,rr4r9r>r= staticmethodrrMrRrVrcrrrArCrnr2rOr5rQr8rSr;rUrYr_r\rarrKr"r%rrMrrIrerhrrrxr>rWr(r*rrErrGr,r0rlpropertyrrrrrrrrrrrrTrrs@rArr.s 59M18%'J!&$&I %$&J!&"F9+ N48G 07& 8080t.G: $&8  !F 8 " "   ,G\ -  "T"  B     8              Q f B " ,% N H 0 / )G'*NN&  " & &  " "    : :: : :: : :: : :: : :: : :: : :: : ::: :: Y YP   W $   : :: " " : :: : :: " "(!%(.2 ( 26 ( 8< ((T55BBDDCC==AA<<==>>"V"VH--rD) metaclassc V^8dQhRRRR/#)r=objsrr>r3r?)r@s"rArBrB s,,6,c,rDc\V4^8Xd;Vwp\V\4'd \V4#\ V4P #RP \\V44#)rrc) rrrr3rrNjoinmaptypestr)r6rs* rAr:r: sP 4yA~ c? + +s8O9%% %xxGT*++rDz/https://github.com/python/cpython/blob/v3.13.0/cV^8dQhRR/#rur?)r@s"rArBrB s''T'rDc ^RIHpHpHpHpHpHpHpHpH pH p H p H p H p Hp HpHpHp/p\$P&FipVP(P+R4pVfK$\-V\.4'dTMV3pVF$pVP1V.4P3V4K& Kk \5VP74RR7FAwppVF5p\9VP: RVP<R RVP> 24K7 KC R#) zPrint the mapping from CPython types to Dynamo VariableTracker classes. Reads _cpython_type from each VT subclass (own attribute only, not inherited). )builtinconstant ctx_managerdicts distributed functionshigher_order_opsiterlazylistsmisc nn_modulestreamstensorrtorch_function user_definedrNc(V^,P#)rb)rP)xs&rAr-print_cpython_to_vt_mapping.. sQqTEVEVrD)rNr30sz -> ) r=r>r?r@rArBrCrDrErFrGrHrIrJrrKrLrrrrrr setdefaultrsortedrprintrOrPrN)r=r>r?r@rArBrCrDrErFrGrHrIrJrrKrLmappingvt_cls cpython_typetypestrS vt_classess rAprint_cpython_to_vt_mappingr[ s (')G%44**?;    *< ? ? l_A   q" % , ,V 4 5 &gmmo;VW F %%&a(<(r#r(r'r&r%r$)rr=).r)__conditional_annotations__rR __future__rr dataclassesr*rloggingrGcollections.abcrrrr contextvarsrenumrr typingr r r r rtorch._C._dynamorrrrrrrQrrrexcrrrrrr.rrrLrr utilsr!r" frozenset_RICHCOMPARE_OPS torch._guardsr)torch.fx.proxyr*rvr+rr,symbolic_convertr-r>r.rBr/ getLoggerrNr dataclassr2rUrSrXr\rorsrrrrrrrrrrrrrrrrrrrrrrrr rrrr$r)r0r3r>rCrJrOrSrVrYrb richcmp_lt richcmp_le richcmp_eq richcmp_ne richcmp_gt richcmp_gererhrlrprtryr|rrrrrrrr_fields _TYPE_FIELDSrrupperrrrrrr_group_fnsr"rrrrrrrr3rrrrrrTP_REPRTP_CALLTP_STR TP_GETATTROTP_RICHCOMPARETP_ITER TP_ITERNEXT TP_DESCR_GET TP_DESCR_SETTP_INITNB_ADD NB_SUBTRACT NB_MULTIPLY NB_REMAINDER NB_DIVMODNB_POWER NB_NEGATIVE NB_POSITIVE NB_ABSOLUTENB_BOOL NB_INVERT NB_LSHIFT NB_RSHIFTNB_ANDNB_XORNB_ORNB_INTNB_FLOATNB_INPLACE_ADDNB_INPLACE_SUBTRACTNB_INPLACE_MULTIPLYNB_INPLACE_REMAINDERNB_INPLACE_POWERNB_INPLACE_LSHIFTNB_INPLACE_RSHIFTNB_INPLACE_ANDNB_INPLACE_XOR NB_INPLACE_ORNB_FLOOR_DIVIDENB_TRUE_DIVIDENB_INPLACE_FLOOR_DIVIDENB_INPLACE_TRUE_DIVIDENB_INDEXNB_MATRIX_MULTIPLYNB_INPLACE_MATRIX_MULTIPLY MP_LENGTH MP_SUBSCRIPTMP_ASS_SUBSCRIPT SQ_LENGTH SQ_CONCAT SQ_REPEATSQ_ITEM SQ_ASS_ITEM SQ_CONTAINSSQ_INPLACE_CONCATSQ_INPLACE_REPEATryr~rrrrr:rr_CPYTHON_BASE_URLr[rErEr)rr_s0@rArs #EE"FF,/1'/@ ###*<*/! d#$:=G=9   &--`t|,'L',  . 1 . %,%"   FF ' Y0HXD ~|Wn 0>  T"22#24 T"@@#@ T"@@#@+F "*=0  ! / & 4 $$&& & %#$ ! :8 $ 8 $ 8 $ 8 $ 8 $ 8 $ ( ( , # 8&" , )    d#EE$E"#,j#,L# #"z" $:$&uS 4 4SuuS 4 4SS 6E6M6M6M qwwy)1,6M8I8Q8Q8Q1+Q.8Q7G7O7O7O! *A-7ONNNAW[!'')4a7N / +  /F/FG/F!QCu+~/FG 1B1J1JK1JAqcK1JK 0@0H0HI0H1aS;0HI NN#/ 0< 0?; - ) NN$LFD 4yC 6*11344 },NOPP%    "#'(  S# #$   Y  d#88$8@BA@>c  :~{':':OL c : (;(;YGc 9m[%7%7Ic  c  cD "" EcP "" Qc\ "" ]ch "" ict "" uc@ "" AcL :~{':':OLMcN  OcZ    [cf    gcr    sc~ :~{':':JGcN  OcZ  [cf !! gcr !! sc~ !! cJ !! KcV "" Wcb "" ccn  ocz  {cF  GcR  Sc^ !! _cj !! kcv !! wcB  CcN  OcZ  [cf  gcr  sc~  cJ  KcV  Wcb  ccn  ocz  {cF  GcR  Sc^  _cj $$ kcv ")) wcB  ")) C cN  #** O cZ  && [ cf   '' g cr   '' s c~  $$  cJ  $$ K cV  ## W cb  %% c cn  %% o cz  $$ { cF  $$ G cR  &-- S c^  %,, _ cj   k cv  !(( w cB  !(( C cN  )00 O c\     ] ch  ## i ct  '' u c@  '' A cN  !! O cZ  !! [ cf  !! g cr  !! s c~    cJ ## KcV ## Wcb ## ccn  )) ocz  )) {c =cL   7$7:-+>-U- 3U-p4,&& F'T%%GKO HKI 0s0"u5"u: "u?"v v 5v vv