^jUdZddlZddlZddlZddlZddlZddlZddlmZm Z ddlm Z m Z ddl Z ddl mZmZmZmZmZmZddlmZmZddlmZmZmZmZmZdd lmZd d lm Z m!Z!d d l"m#Z#e rdd l$m%Z%de!de!ddfdZ&ddde!de!de'de f dZ(edde)de*e+e+e+e+ffdZ,de)de+de-fdZ.de)de+de-fdZ/e e.ej` Z1e e.ejd Z3e e.ejh Z5e e.ejl Z7e e.ejp Z9e e.ejt Z;e e.ejx Z=e e.ej| Z?e e/ej ZAe e/ej ZCe e/ej ZEe e/ej ZAde)de+de-fd!ZFe eFej ZHe eFej ZJe eFej ZLe eFej ZNe eFej ZPe eFej ZRe eFej ZTe eFej ZVe eFej ZXe eFej ZZe eFej Z\e eFej Z^e eFej Z`e eFej Zbe eFej Zde eFej Zfe eFej Zhe eFej Zje eFej Zle eFej Zne eFej Zpe eFej Zre eFej Zte eFej Zve eFej Zxe eFej Zze eFej Z|e eFej Z~e eFej Ze eFej Ze eFej Ze eFej Ze eFej Ze eFej Zde)de-fd"Zde)de-fd#Zde)de-fd$Zde)de-fd%Zde)de-fd&Zde)de-fd'Zde)de-fd(Zd)e!de)fd*Zddd+e!d,e'de!fd-Z dd/Z dd0Zddd)e!de!fd1ZeZee+ed2<ddd)e!de!fd3Z dd4Zddd)e!d+e!de!fd5Zddd)e!d6e!de!fd7Zddd8e!d9e!d:e!de!f d;Zddd:e!d+e!dZddd:e!d+e!de!fd?Zddd)e!de!fd@Zddd)e!de!fdAZddd)e!dd.fdBZddd)e!de!fdCZddd)e!de!fdDZddd)e!de!fdEZd)e!de-fdFZddd)e!dd.fdGZidHejdIejdJejdKejdLejdMejdNejdOejdPejdQejdRejdSejdTejdUejdVejdWejdXejejejejejejejTdYZdZe!de-fd[Zde!de!de-fd\Zddde!de!d]e'de!f d^Zddde!de!d]e'de'de!f d_Zddde!de!d`e'd]e'de!f daZddde!de!d`e'd]e'de'de!fdbZddde!de!de!fdcZddde!de!de!fddZdddee!dfe!de!fdgZdddee!dfe!de!fdhZdddie!ddfdjZddde!de!de!fdkZddde!de!de!fdlZ ddddme!dne!doe-de!f dpZdddme!dne!de!fdqZeZdme!dddne!dre'de!f dsZdme!dddne!dre'de!f dtZdudvdwdxdydzd{Zee'e'fed|<d}d~dddddZee'e'fed<ddde!de!dre'de!f dZddde!de!dre'de!f dZddd)e!de*e+e-ffdZddd)e!de!fdZ ddZeje jjejfZe*e)dfed<ddde!de!de!fdZddde!de!de*e!e!ffdZy)a3 Dynamo implementations of CPython's PyObject_* default slot algorithms. Analogous to CPython's Objects/object.c, this module holds the general dispatch machinery that is independent of any specific type. Per-type hook implementations (bool_impl, richcompare_impl, etc.) live in their respective VT files. N) lru_cachepartial)NoReturn TYPE_CHECKING)get_type_slotshas_slotPyMappingSlots PyNumberSlotsPySequenceSlots PyTypeSlots)graph_break_hints polyfills)handle_observed_exceptionObservedTypeErrorraise_observed_exceptionraise_type_error unimplemented)istype)NO_SUCH_SUBOBJVariableTrackerConstantVariable)InstructionTranslatorBaseleftrightreturnzVariableTracker | Nonec||urtjdS|j}|j}|tu}|tu}|r0|r.||urtjdStjdS||k7rtjdSddlm}ddlm}ddlm }ddl m } t|||| |frtjdS |j|jurtjdS dd lm} t#|| j$rCt#|| j$r-|j&|j&urtjdSy #t$rYkwxYw) zTry to determine Python identity (left is right) at trace time. Returns ConstantVariable(True/False) if determinable, else None. Mirrors the logic in BuiltinVariable's handle_is handler. TFr)ConstDictVariable) ListVariable)TracebackVariable) SetVariabler ) variablesN)rcreateget_real_python_backed_valuerdictsr listsr!miscr"setsr# isinstance python_typeNotImplementedErrorr$rExceptionVariableexc_type) rrleft_val right_val left_known right_knownr r!r"r#r$s r/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_dynamo/variables/object_protocol.pyvt_identity_comparer6.sm u}&&t,,002H224I/J>1Kk9$  # #D ) "((/ [ &&u--)#'!  , =NO &&u--    U%6%6%8 8#**51 1 9  tY001 5)55 6 MM /&&u--     s4E** E65E6txrvw op_symbolc ht|d|d|jd|jdy)Nz unsupported operand type(s) for z: '' and '')rpython_type_name)r7r8r9r:s r5binop_type_errorr?js=  *9+S9K9K9M8NgVWVhVhVjUkklm)maxsizeobj_typect|S)z'Get all type slots for a type (cached).)rrCs r5_get_cached_slotsrFvs ( ##r@slotc:t|\}}}}t||S)z4Check whether obj_type implements the given sq slot.rFr)rCrG seq_slots_s r5type_implements_sq_slotrL|s"*84Iq!Q It $$r@c:t|\}}}}t||S)z4Check whether obj_type implements the given mp slot.rI)rCrGrK map_slotss r5type_implements_mp_slotrOs"*84Ay!Q It $$r@)rGc:t|\}}}}t||S)z.Check whether obj_type implements the nb slot.rI)rCrGrK number_slotss r5type_implements_nb_slotrRs"-h7Aq, L$ ''r@cVt|\}}}}t|tjSN)rFrr TP_ITERrCrK type_slots r5type_implements_tp_iterrX s(*84Aq!Y I{22 33r@cVt|\}}}}t|tjSrT)rFrr TP_ITERNEXTrVs r5type_implements_tp_iternextr[s(*84Aq!Y I{66 77r@cVt|\}}}}t|tjS)z3Check whether obj_type implements the tp_repr slot.)rFrr TP_REPRrVs r5type_implements_tp_reprr^s(*84Aq!Y I{22 33r@cVt|\}}}}t|tjS)z2Check whether obj_type implements the tp_str slot.)rFrr TP_STRrVs r5type_implements_tp_strras(*84Aq!Y I{11 22r@ct|SrT)r[rEs r5 pyiter_checkrc"s 'x 00r@c:t|tryt|S)zBImplements PySequence_Check semantics for VariableTracker objects.F) issubclassdicttype_implements_sq_itemrEs r5pysequence_checkrh)s(D! "8 ,,r@ct|S)z@Implements _PyIndex_Check semantics for VariableTracker objects.)type_implements_nb_indexrEs r5 pyindex_checkrk1s $H --r@objc |jS#t$r%td|ddgtjYywxYw)NzUnsupported python_type() callz! does not implement python_type()zhThis VariableTracker does not implement python_type(), which is required for object protocol operations.gb_typecontext explanationhints)r,r-rr DYNAMO_BUG)rls r5maybe_get_python_typert7sR      4e<=@"--    s+AAkeycontainer_namect|}|tttfvrBt |s&t t ||d|jg|j|}|S)u_PyIndex_Check → nb_index path used by list/tuple/range/str/bytes subscript. ref: https://github.com/python/cpython/blob/v3.13.3/Include/internal/pycore_abstract.h (_PyIndex_Check) z) indices must be integers or slices, not args) rtintboolslicerjr TypeErrorr> nb_index_impl)r7rurvkey_types r5validate_sequence_indexrFsj%S)HT5))'1 $%&&OPSPdPdPfOgh # Jr@rct|}t|r|j|St|rt ||j dt |d|j dy)Nz is not a mappingzobject of type z has no len())rttype_implements_mp_length mp_lengthtype_implements_sq_lengthrr>r7rlTs r5vt_mapping_sizer]sf c"A #}}R   # 4 4 677HIJR?3+?+?+A*B-PQr@cht|}t|r|j|St||S)z Implements PyObject_Size/PyObject_Length semantics for VariableTracker objects. Dispatches to sq_length (sequences) or mp_length (mappings) depending on the VT type. )rtr sq_lengthrrs r5 generic_lenrks2 c"A #}}R  2s ##r@cddlm}|jr* |jt |j St|}t|r|j|}||S t||}ddlm}t!||r#|j||j#dkDS|j }|dkrtt$|dg|j|dkDS#t $r+}tt||t|gYd}~d}~wwxYw#t&$rt)|YnwxYw|jdS) uMirrors PyObject_IsTrue. https://github.com/python/cpython/blob/c09ccd9c429/Objects/object.c#L2135-L2158 Resolution order: constants → nb_bool → mp_length/sq_length → truthy. rrrxN)SymNodeVariablerz__len__() should return >= 0T)constantris_python_constantr%r{as_python_constant Exceptionrtypestrrttype_implements_nb_bool bool_implrtensorrr+as_proxy ValueErrorrr) r7rlrerCresultlengthr length_vals r5 generic_boolrzsB+  A*#**40F0F0H+IJ J%S)Hx(r"  M &R%+ fo ."))"foo.?!.CD D..0 > $B&D%E '&&zA~66+ A $T!WbAx @ @ A, &!"%& #  " "4 ((s0(C+-AD". TypeError if the result is not str. z[...]z{...}z...z#__repr__ returned non-string (type )zobject of type 'z ' has no reprN)rtr^idrlistrf collectionsdequerr%getadd repr_impldiscardrerr__name__r>)r7rlrCobj_idsentinelr result_types r5 generic_reprrs%S)Hx(C ] "gtWk6G6GQH#**8<<%+HI I&! *]]2&F  ! !& )+F3 +s+ 5k6J6J5K1M  R+C,@,@,B+C=QR  ! !& )s C??DcHt|tur|St|}t|r7t|jt jur|j |}n t ||}t|}t|tst|d|jd|S)zMirrors PyObject_Str semantics in Dynamo. https://github.com/python/cpython/blob/v3.13.3/Objects/object.c#L781-L829 Resolution order: str identity check -> tp_str (str_impl) -> tp_repr fallback. z"__str__ returned non-string (type r) rtrrarstr_implrrrerr)r7rlrCrrs r5 generic_strrsS!S( $S)Hx( I  o&>&> >b!b#&'/K k3 ' 01E1E0Fa H  Mr@ct|}t|r|j||St|rYt|}t |r|j |}t |||St||jd|jt|tr|j||St|d|jdy)uCPython's PyObject_GetItem — dispatch to the type's mp_subscript/sq_item. PyObject_GetItem: https://github.com/python/cpython/blob/62a6e898e01/Objects/abstract.c#L155-L206 CPython checks three branches in order: 1. tp_as_mapping->mp_subscript (L161-166) 2. tp_as_sequence->sq_item (L168-181) — only if key passes _PyIndex_Check 3. PyType_Check(o) (L183-203) — type[int] → GenericAlias/__class_getitem__ Branch 1 is the common path (list, tuple, dict, range all have mp_subscript). Branch 2 fires for types with only sq_item (e.g. deque). Branch 3 delegates to mp_subscript_impl for type objects (__class_getitem__). z indices must be integers, not r=z' object is not subscriptableN) rttype_implements_mp_subscriptmp_subscript_implrgrjr~vt_sequence_getitemrrrer)r7rlrurCrs r5 vt_getitemrs$%S)H#H-$$R--x((- #H -##B'C&r34 4   !!@ARAR@S T (D!$$R--R1X..//LMNr@indexct|}t|rt|trg|j }t|t rG|dkrBt |r7|j|}tj||j z}|j||St|rt|d|jdt|d|jdy)uMCPython's PySequence_GetItem — always sq_item, never mp_subscript. ref: https://github.com/python/cpython/blob/v3.13.3/Objects/abstract.c#L1874-L1902 Called by PyObject_GetItem branch 2, reversed() fallback, and the old iteration protocol. Wraps negative indices via sq_length before dispatching to sq_item. rr=' is not a sequencez"' object does not support indexingN) rtrgr+rrrzrrr% sq_item_implrrr>)r7rlrrC index_valrs r5rrs%S)Hx( e- .002I)S)i!m,X6 ]]2.F,33!F$=$=$??EE**#H-q!5!5!7 88KLMR1S11344VWXr@sioct|}t|rt|trg|j }t|t rG|dkrBt |r7|j|}tj||j z}|j|||St|rt|d|jdt|d|jdy)Nrr=r)' object does not support item assignment) rttype_implements_sq_ass_itemr+rrrzrrr%sq_ass_item_impl type_implements_mp_ass_subscriptrr>)r7rrrs_typerrs r5vt_sequence_setitemr1s#1 %F"6* a) *,,.I)S)i!m,V4[[_F(// F)r7rruro_typers r5generic_setitemrKs#1 %F'/&&r366"6*(-  "##B'C&r1c59 9 78L8L8N7OqQ  a""$%%N Or@ct|}t|rat|tr>|j }|dkr)t |r|j |}t|||}|j||dSt|rt|d|jdt|d|jdy)Nrr=rz'' object does not support item deletion) rtrr+rrrrvt_addrrrr>)r7rrridxrs r5sequence_delitemrcs#1 %F"6* a) *&&(CQw,V4[[_Fr1f-A!!"a..'/q!3!3!5 66IJK a""$%%L Mr@cJt|}t|r|j||dSt|}t|r|j |}t |||St |rt|d|jt|d|jdy)Nz$sequence index must be integer, not r=z ' does not support item deletion) rtrrrkr~rrrr>)r7rrurr key_values r5generic_delitemr|s#1 %F'/&&r355$S)HX%%b) Ay11 $V , 6s7K7K7M6NO R1Q//122RSTr@cddlm}|jrGt|j t r)|j t |j St|}t|r|j|St|r|j|S|jrSt|j tttfr* |j t |j St#|d|j%dy#t$r&}t!t|t|gYd}~Id}~wwxYw)uMirrors PyNumber_Long (int(x) dispatch). https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1520-L1632 Resolution: nb_int → nb_index → str/bytes/bytearray parsing → TypeError. rrrxNzLint() argument must be a string, a bytes-like object or a real number, not 'r=)rrrr+rrzr%rttype_implements_nb_int nb_int_implrjr~rbytes bytearrayrrrr>r7rlrrCrs r5 generic_intrs&+  Js/E/E/G$M&&&s3+A+A+C'DEE$S)Hh'r"")  $$ J  3y"9% D*#**3s/E/E/G+HI I  ""%"6"6"8!9 < D $Z3q6( C C Ds(D## E,E  Ecddlm}|jrGt|j t r)|j t |j St|}t|r|j|St|r|j|S|jrNt|j ttfr* |j t |j St!|d|j#dy#t$r&}tt|t|gYd}~Id}~wwxYw)uMirrors PyNumber_Float (float(x) dispatch). https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1635-L1692 Resolution: nb_float → nb_index → str parsing → TypeError. rrrxNz9float() argument must be a string or a real number, not 'r=)rrrr+rfloatr%rttype_implements_nb_float nb_float_implrjr~rrrrrr>rs r5 generic_floatrs$+  Js/E/E/G$O&&&uS-C-C-E'FGG$S)H)  $$ )  $$ Js/E/E/G#u$V D*#**51G1G1I+JK K  $$&'q * D $Z3q6( C C Ds(D E 'EE ct|}t|st|d|jd|j |S)a7 Implements PyIter_Next / tp_iternext semantics for VariableTracker objects. Calls obj.tp_iternext_impl(tx) if the object is an iterator, otherwise raises TypeError. StopIteration propagation is left to the caller (mirrors CPython's iternext contract where NULL return signals exhaustion). zexpected an iterator, got 'r=)rtr[rr>tp_iternext_implrs r5generic_iternextrsF c"A &q ):3;O;O;Q:RRSTU    ##r@ct|}t|r|j|St|d|j dy)zMirrors PyNumber_Negative. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1375-L1392 Algorithm: 1. If type has nb_negative slot, call obj.nb_negative_impl(tx) 2. Otherwise, raise TypeError zbad operand type for unary -: 'r=N)rttype_implements_nb_negativenb_negative_implrr>r7rlrCs r5 generic_negrH%S)H"8,##B'' )#*>*>*@)ACr@ct|}t|r|j|St|d|j dy)zMirrors PyNumber_Positive. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1375-L1393 Algorithm: 1. If type has nb_positive slot, call obj.nb_positive_impl(tx) 2. Otherwise, raise TypeError zbad operand type for unary +: 'r=N)rttype_implements_nb_positivenb_positive_implrr>rs r5 generic_posrrr@ct|}t|r|j|St|d|j dy)zMirrors PyNumber_Absolute. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1375-L1395 Algorithm: 1. If type has nb_absolute slot, call obj.nb_absolute_impl(tx) 2. Otherwise, raise TypeError zbad operand type for abs(): 'r=N)rttype_implements_nb_absolutenb_absolute_implrr>rs r5 generic_absr%sH%S)H"8,##B'' '(<(<(>'?qAr@cHt|}t|xs t|S)zOCheck if the object supports iteration (i.e. has tp_iter or sequence protocol).)rtrXrh)rlrs r5vt_is_iterabler;s"c"A "1 % <)9!)<rh functionsrrbuiltinssequence_iterator call_function)r7rlrresres_Trs r5generic_getiterrAs c"Aq!r"%c*E" 89M9M9O8PPQR  ! 3#I$6$6$H$HIWW r   q!5!5!7 88PQRr@ nb_lshiftnb_inplace_lshiftnb_inplace_rshift nb_rshiftnb_or nb_inplace_or nb_subtractnb_inplace_subtractnb_addnb_inplace_add nb_multiplynb_inplace_multiplynb_andnb_inplace_andnb_xornb_inplace_xornb_floor_divide)nb_inplace_floor_dividenb_true_dividenb_inplace_true_divide nb_remaindernb_inplace_remainder nb_divmodrc,|jtSrT)is_constant_matchNotImplemented)rs r5is_nb_not_implementedrs  # #N 33r@cr t|j|jS#t$rYywxYw)z?Check if w's underlying Python type is a proper subtype of v's.F)rer,r-)r9r8s r5is_python_subtypers2!--/1==?;; s '* 66op_slotc|d}t|}t|}t|}t||rtt ||dnd}t||rtt ||dnd} || urd} |@| &t ||r| |||d} t | s| Sd} ||||d} t | s| S| | |||d} t | s| StjtS)u/CPython's binary_op1: try v's slot, then w's slot with subclass priority. Each VT that participates provides a ``_impl(self, tx, other, reverse)`` method. ``reverse=False`` means "self is left operand" (forward, e.g. ``__or__``), ``reverse=True`` means "self is right operand" (reverse, e.g. ``__ror__``). For built-in types the flag is ignored because their slots check both operands symmetrically. CPython splits the check into two steps: ``Py_TYPE(v)->tp_as_number != NULL`` (does the type have a number-protocol struct), then read the slot pointer (which may itself be NULL). We collapse both into a single :func:`type_implements_nb_slot` query — its bit is set only when the specific slot is non-NULL, which already implies ``tp_as_number`` is non-NULL. Treating a missing slot as "no impl" is what keeps the base ``VariableTracker.nb_*_impl`` graph-break out of the path for types that genuinely lack the slot in C. https://github.com/python/cpython/blob/3.13/Objects/abstract.c#L926-L977 _implNTF) NB_SLOT_MAPPINGrtrRgetattrrrrrr%r) r7r8r9r impl_attr nb_slot_bitv_typew_typev_slotw_slotrs r5 binary_op1r!s2)5!I!'*K "1 %F "1 %F #6; 7 QD)   #6; 7 QD)     "3Aq"9Ar1d+F(0 F2q%($V,M 2q$'$V,M  " "> 22r@cTt||||}t|rt|||||S)zCPython's binary_op: binary_op1 + TypeError fallback. https://github.com/python/cpython/blob/3.13/Objects/abstract.c#L997-L1020 )r!rr?)r7r8r9rr:rs r5 binary_opr#s/Aq' *FV$Q9- Mr@iop_slotct|}t|t|r1|d}tt ||}||||}t |s|St ||||S)Nr)rtrRrrrrr!) r7r8r9r$rrrrGrs r5 binary_iop1r&sd#1 %Fvx'@Aj& tAw *aQ$V,M b!Q ((r@cVt|||||}t|rt|||||S)zCPython's binary_iop: try inplace slot, fallback to binary_op1. Combines binary_iop1 + TypeError fallback from binary_iop. https://github.com/python/cpython/blob/3.13/Objects/abstract.c#L1229-L1270 (binary_iop1, binary_iop) )r&rr?)r7r8r9r$rr:rs r5 binary_iopr(s1Q8W 5FV$Q9- Mr@ct|||d}t|s|St|}t|r|j ||St |||dy)zHImplements addition via nb_add / nb_inplace_add with binary_op dispatch.r+N)r!rrttype_implements_sq_concatsq_concat_implr?)r7r8r9rrs r5rr*sTAq( +F  ( a A #A&&RAs#r@ct|||dd}t|rSt|}t|r|j ||St |r|j ||St|||d|S)zIImplements in-place addition via nb_inplace_add with binary_iop dispatch.rrz+=)r&rrt!type_implements_sq_inplace_concatsq_inplace_concat_implr+r,r?)r7r8r9rrCs r5vt_inplace_addr0;srQ#3X >FV$(+ ,X 6++B2 2 &x 0##B* * RAt , Mr@seqnct|}t|st|d|jd|j |}t |||j ||S)zMirrors CPython's sequence_repeat helper. Validates that ``n`` is index-like, converts it to an int, and dispatches to ``seq.sq_repeat_impl(tx, count)``. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1156-L1174 ,can't multiply sequence by non-int of type 'r=)rtrjrr>r~validate_sequence_repeat_countsq_repeat_implr7r1r2n_typecounts r5sequence_repeatr:Zse#1 %F #F + :1;M;M;O:PPQ R  OOB E"2u-   b% ((r@ct|}t|st|d|jd|j |}t |||j ||S)zsequence_repeat using sq_inplace_repeat. The validation step is identical to ``sequence_repeat``; only the target slot differs. r4r=)rtrjrr>r~r5sq_inplace_repeat_implr7s r5sequence_inplace_repeatr=qse#1 %F #F + :1;M;M;O:PPQ R  OOB E"2u-  % %b% 00r@r9c|j}|tj dz ks|tjkDrtt|dgyy)Nrz,cannot fit 'int' into an index-sized integerrx)rsysrBr OverflowError)r7r9r2s r5r5r5sH   "ACKKr7r8r9rrrs r5generic_multiplyrEsAq- 0F  ( "1 %F "1 %F (r1a(( (r1a((    !););)=(>a Ar@cVt|||dd}t|s|St|}t|}t|r t |||St |r t |||St |r t |||St|d|jd|jdy)aMirrors CPython's PyNumber_InPlaceMultiply. Try nb_inplace_multiply / nb_multiply via binary_iop1; on NotImplemented fall back to sq_inplace_repeat (preferred), then sq_repeat. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L1330-L1357 rrz%unsupported operand type(s) for *=: 'r<r=N) r&rrt!type_implements_sq_inplace_repeatr=rCr:rr>rDs r5generic_inplace_multiplyrHsQ#8- HF  ( "1 %F "1 %F(0&r1a00 (r1a((!(r1a((    !););)=(>a Ar@selfotherreversect|}t|tjr|j |||St |r t |||St|d|jd|jdy)z@``self.__mul__(other)`` / ``self.__rmul__(other)`` slot wrapper.)rKrBr<r=N) rtrRr NB_MULTIPLYnb_multiply_implrCr:rr>)r7rIrJrK self_types r5slot_wrapper_mulrPs&d+Iy-*C*CD$$R$@@ +r4//   ! ! # $GE,B,B,D+EQ Hr@ct|}t|tjr|j ||St |r t |||St|||S)u``self.__imul__(other)`` slot wrapper. When neither ``nb_inplace_multiply`` nor ``sq_inplace_repeat`` is installed, the slotdef machinery doesn't generate ``__imul__`` at all — but the operator-level fallback in ``slot_nb_inplace_multiply`` (typeobject.c) does try the non-inplace slot. We mirror that here so method lookups don't graph-break on (e.g.) tuple, even though tuple has no ``__imul__`` attribute in standard CPython. )rtrRr NB_INPLACE_MULTIPLYnb_inplace_multiply_implrGr=rP)r7rIrJrOs r5slot_wrapper_imulrT sW&d+Iy-*K*KL,,R77(3&r477 Be ,,r@opch|dk(rLt||}|%|jrtjdStjtS|dk(rD|j ||d}t |r|Stj|j StjtS)aobject's tp_richcompare. https://github.com/python/cpython/blob/e76aa128fe/Objects/typeobject.c#L6263-L6305 - __eq__: identity check, else NotImplemented - __ne__: delegates to tp_richcompare(self, other, Py_EQ) and inverts - ordering __eq__T__ne__)r6rrr%rrichcompare_implis_richcompare_not_implemented)rIr7rJrUidentity eq_results r5object_richcomparer]Zs X~&tU3  H$?$?$A#**40 0&&~66 x))"eX> )) 4 &&9+G+G+I'IJJ&&~66r@c|jr|jstjtS|j }|j } t t ||||}tjS#t$r/}tt|t|jYd}~Hd}~wwxYw)z=Constant-fold comparison for types with as_python_constant().rxN) rrr%rrrrr}rrry)rIr7rJrUself_val other_valrrs r5 python_constant_richcompare_implraxs  " " $E,D,D,F&&~66&&(H((*IC,h,XyA  " "6 ** C BT!&&\BBCsB C%B??C__gt____lt____ge____le__rWrX)rcrbrerdrWrX _REFLECTED_OPz>=)rcrerWrXrbrd_OP_STRc t|} |j} |j}d}|4|2||ur.t||r"d}|j |||}t |s|S|j |||}t |s|S|s |j |||}t |s|S|dvrct ||} | +|dk(r$tj| j S| Stdd|d|d|d d gtj yt|d t|d |jd|jd y#t$rd}Y:wxYw#t$rd}Y;wxYw)auDynamo's do_richcompare. https://github.com/python/cpython/blob/e76aa128fe/Objects/object.c#L994-L1039 Implements the 4-step algorithm directly using richcompare_impl slots. Graph breaks inside user comparison methods propagate to COMPARE_OP (which runs eagerly) because UDOV.richcompare_impl disables nested graph breaks on the resolved funcvar. NFTrWrXrXz*richcompare identity fallback undeterminedzgeneric_richcompare(, rz9Cannot determine object identity for comparison fallback.rnr=z&' not supported between instances of 'r<)rfr,r-rerYrZr6rr%rrr SUPPORTABLErrir>) r7r8r9rU reflectedrrchecked_reverserr[s r5generic_richcomparerpsb!IO    & vv &##B95-f5M  Ar *F )& 1  ##B95-f5M !!&q!,  X~'..83N3N3P/PQQO@*1#Rs"RD:S2%112     }""$%WQ-?-?-A,B! E ]  s"EE EE E('E(c|dvrtd|t||}|(|jrtj|dk(St ||||S)aDynamo's PyObject_RichCompareBool for eq/ne. https://github.com/python/cpython/blob/e76aa128fe/Objects/object.c#L1046-L1080 Like generic_richcompare, but with an identity shortcut first: if v and w are the same Python object, eq is True and ne is False. This matters for NaN (nan is nan -> True, but nan == nan -> False). Used by container comparisons (list_richcompare, tuplerichcompare) which call PyObject_RichCompareBool per element in CPython. rkz2generic_richcompare_bool only supports eq/ne, got rW)AssertionErrorr6rrr%rp)r7r8r9rUr[s r5generic_richcompare_boolrsse  %%QRTQUVWW"1a(H ; ; =&&rX~66 r1a ,,r@c$|j|S)zInternal API: compute hash as (value, is_fake). Dispatches to the VT's hash_impl. Called by generic_hash (which wraps the result in a VT), container hash_impls (which propagate is_fake), and HashableTracker (which just needs the int). ) hash_impl)r7rls r5generic_hash_implrvs == r@cddlm}m}m}t ||\}}|r|||j S|j |S)a$User-facing API: mirrors PyObject_Hash, returns a VariableTracker. https://github.com/python/cpython/blob/e76aa128fe/Objects/object.c#L1101-L1115 Wraps the result in ConstantVariable or FakeIdVariable depending on whether the hash depends on a sourceless object's identity. r)rFakeIdVariable FakeValueKind)kind)rrrxryrvHASHr%)r7rlrrxryhis_fakes r5 generic_hashr~ sEJI"2s+JAwam&8&899 "  " "1 %%r@ct|}t|r|j||St||}t j |t jj|||giS)z Implements PySequence_Contains semantics for VariableTracker objects. If the object has sq_contains (i.e., __contains__), calls obj.sq_contains(tx, item). Otherwise falls back to iterating over obj and comparing each element. ) rttype_implements_sq_contains sq_containsrrbuildrimpl_CONTAINS_OP_fallbackr)r7rlitemrits r5generic_containsrse c"A"1%r4((R %$$  33 -T2J + ,r@.(_CONSTANT_FOLD_SUBCLASSCHECK_METACLASSESderivedclsc|j}|j}|tus|tur7tdd|d|dddgtjtj t |}|tur tjt||Stj h}t"j$d kr|j't(j*||vr+t)j,|}t/j0||}d d lm} t7|| rg|j8D]C} t;||| } t7| ts!| j<s.tjd cStjdSt7|tr9tt|t>r tjt||St7|ts tA|d|jC|d|gi} tE|| S#t$r0}tt|t|jYd }~d }~wwxYw#t$r/}tt|t|jYd }~d }~wwxYw)zMirrors CPython's PyObject_IsSubclass / object_issubclass. https://github.com/python/cpython/blob/v3.13.0/Objects/abstract.c#L2766-L2823 This only attempts to replicate object_issubclass, otherwise we delegate to cpython z'issubclass() with unsupported argumentsz issubclass(rlrz:Arguments to issubclass() must be backed by python values.z#Make sure your arguments are types.rnrxN)r) TupleVariableTFzBissubclass() arg 2 must be a class, a tuple of classes, or a union__subclasscheck__)#r&rrr USER_ERRORrmrtrrr%rer}rrrytypes UnionTyper? version_infortyping_UnionGenericAliasget_argsrrr(rr+itemsgeneric_issubclassrrr call_methodr) r7rr derived_pycls_pycls_typer union_typesryrrrrs r5rr<sO557J  - - /F^#v'?=!'"SE3T5"--#.. %S)H4 G#** ??#K '!  % % ;v&##B-%#}%II 5D"2w5A!-.177'..t44 5 &&u--&$J V >% G#**  fd #  P __R!4wi DF F ##y G $Yaff F F GP G $Yaff F F Gs0<H 0I I%II J%I;;Jiter_ null_or_idxcddlm}t||s#t|tur|j ||St ||}||fS)ab Mirrors _PyForIter_VirtualIteratorNext. When ``null_or_idx`` is a tagged int (3.15+ virtual-iter path), dispatch to the iterable's ``_tp_iteritem`` slot via ``tp_iteritem_impl`` and return ``(value, next_index)``. Otherwise fall back to the standard iterator protocol (``tp_iternext``) and return ``(value, NULL)``. Iterator exhaustion is signaled by ``ObservedUserStopIteration`` propagating out of the slot impl, matching the rest of Dynamo's iterator protocol. https://github.com/python/cpython/blob/f31a89bb901067dd105b00cfa90523cf7ffdbbdd/Python/ceval.c#L3733 r) NullVariable)r)rr+rtrztp_iteritem_implr)r7rrrnext_s r5virtual_iterator_nextrsL&# {L 1 !+ .# 5%%b+66 R 'E ,.  r@)r7rrlrrr)F)r7rrlrrrrr)__doc__abcrenumr?rr functoolsrrrrtorchtorch._C._dynamorrr r r r r.rrexcrrrrrutilsrbaserrrrsymbolic_convertrr6rr?rtuplerzrFr{rLrOSQ_ITEMrg SQ_LENGTHr SQ_CONCATr+SQ_INPLACE_CONCATr. SQ_CONTAINSr SQ_ASS_ITEMr SQ_REPEATrCSQ_INPLACE_REPEATrG MP_LENGTHr MP_SUBSCRIPTrMP_ASS_SUBSCRIPTrrRNB_ADDtype_implements_nb_add NB_SUBTRACTtype_implements_nb_subtractrMtype_implements_nb_multiply NB_REMAINDERtype_implements_nb_remainderNB_POWERtype_implements_nb_power NB_NEGATIVEr NB_POSITIVEr NB_ABSOLUTErNB_BOOLr NB_INVERTtype_implements_nb_invert NB_LSHIFTtype_implements_nb_lshift NB_RSHIFTtype_implements_nb_rshiftNB_ANDtype_implements_nb_andNB_XORtype_implements_nb_xorNB_ORtype_implements_nb_orNB_INTrNB_FLOATrNB_INPLACE_ADDtype_implements_nb_inplace_addNB_INPLACE_SUBTRACT#type_implements_nb_inplace_subtractrR#type_implements_nb_inplace_multiplyNB_INPLACE_REMAINDER$type_implements_nb_inplace_remainderNB_INPLACE_POWER type_implements_nb_inplace_powerNB_INPLACE_LSHIFT!type_implements_nb_inplace_lshiftNB_INPLACE_RSHIFT!type_implements_nb_inplace_rshiftNB_INPLACE_ANDtype_implements_nb_inplace_andNB_INPLACE_XORtype_implements_nb_inplace_xor NB_INPLACE_ORtype_implements_nb_inplace_orNB_FLOOR_DIVIDEtype_implements_nb_floor_divideNB_TRUE_DIVIDEtype_implements_nb_true_divideNB_INPLACE_FLOOR_DIVIDE'type_implements_nb_inplace_floor_divideNB_INPLACE_TRUE_DIVIDE&type_implements_nb_inplace_true_divideNB_INDEXrjNB_MATRIX_MULTIPLY"type_implements_nb_matrix_multiplyNB_INPLACE_MATRIX_MULTIPLY*type_implements_nb_inplace_matrix_multiplyrXr[r^rarcrhrkrtrrrrsetr__annotations__rrrrrrrrrrrrrrrr NB_DIVMODrrrr!r#r&r(rr0r:r=r5rErHrPrTrZr]rarfrfrirprsrvr~rABCMeta_C _TensorMetaEnumMetarrrr@r5rs  (* ,1&<9 9 99x #       3$$sCc/A)B$$ %d%#%$% %d%#%$%""9@W@WX#/";";$/";";%,/"C"C%!&/"="=&/"="=$/";";%,/"C"C%! $.":": '."="= $+."A"A$ $.":": (d(#($( !!8}?S?ST%-";";&-";"; '-"<"< ##:AWAWX%-";";&-";";&-";";""9 @U@UV#-"9"9$-"9"9$-"9"9!!8}?S?ST !8}?S?ST 7m>Q>QR !8}?S?ST"#:AWAWX!(-">">"'.-"C"C'#'.-"C"C'#(/-"D"D($$+-"@"@$ %,-"A"A%!%,-"A"A%!")-">">"")-">">"!(-"="=!#*-"?"?#")-">">"+2-"G"G+'*1-"F"F*&##:AWAWX%,-"B"B&".5-"J"J.* 4d4t4 8$848 4d4t4 3T3d3 141D1-t--.D.T.    4  #  . R# R*; R R $# $*; $ $')#')*9')')T% s3xS#S*9SS@#*;<(O#(O (O (O (OVY#Y Y Y YB#   4#     0# 2U#UU U U.'#'*9''T&#&*9&&R$#$*9$$&#*9,#*9,#*9,==D= !S#!S*9!S!SR((8888((   ]  ]00 =,,=<< m""m22=,,=<< m""m22 m"" m22!"}44#$ -DD#22+BB!..)>>((/64/4d4_"F3#F3F3F3 F3  F3R#     @)#))) )  )  )&#     ($#$$$ $"# >)#) )) ).1#1 11 1*  #       # <# R #     ,-#- - - -V"77 7#7 7 7  7<+ +#+ + +  +( ! tCH~ c3hE #E E E  E  E P-#--- -  -0 # *9  39 &#&*9&&$,#,*;,CT,,4KK HHMM>(%c *:_$#_$ _$ _$ _$D!#! !!! ?O +, !r@