CjUdZddlZddlZddlZddlZddlZddlZddlmZm 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.ej0 Z1e e.ej2 Z3e e.ej4 Z5e e.ej6 Z7e e.ej8 Z9e e.ej: Z;e e.ej< Z=e e.ej> Z?e e/ej@ ZAe e/ejB ZCe e/ejD ZEe e/ej@ ZAde)de+de-fd!ZFe eFejG ZHe eFejI ZJe eFejK ZLe eFejM ZNe eFejO ZPe eFejQ ZRe eFejS ZTe eFejU ZVe eFejW ZXe eFejY ZZe eFej[ Z\e eFej] Z^e eFej_ Z`e eFeja Zbe eFejc Zde eFeje Zfe eFejg Zhe eFeji Zje eFejk Zle eFejm Zne eFejo Zpe eFejq Zre eFejs Zte eFeju Zve eFejw Zxe eFejy 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.fdGZidHej[dIejsdJejudKej]dLejcdMej{dNejIdOejkdPejGdQejidRejKdSejmdTej_dUejwdVejadWejydXej}ejejejejMejoejdYZdZe!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 ddddne!doe!dpe-de!f dqZdddne!doe!de!fdrZeZdne!dddoe!dse'de!f dtZdne!dddoe!dse'de!f duZdvdwdxdydzd{d|Zee'e'fed}<d~ddddddZee'e'fed<ddde!de!dse'de!f dZddde!de!dse'de!f dZddd)e!de*e+e-ffdZddd)e!de!fdZ ddZeje jjejfZe*e)dfed<ddde!de!de!fdZddde!de!de*e!e!ffdZdS)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|}|}|tu}|tu}|r.|r,||urtjdntjdS||krtjdSddlm}ddlm}ddlm }ddl m } t|||| |frtjdS | | urtjdSn#t$rYnwxYwdd lm} t#|| jr7t#|| jr"|j|jurtjdSd S) 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 UC:\Projects\DSCons\.venv\Lib\site-packages\torch\_dynamo\variables\object_protocol.pyvt_identity_comparer6.s u}}&t,,,0022H2244I/J>1K k 9$$  #D ) ) )!(// [  &u---)(((((######''''''!!!!!!  , =NO. &u---     U%6%6%8%8 8 8#*511 1 9       tY011. 5)5 6 6. M / /&u--- 4s;D D('D(txrvw op_symbolc t|d|d|d|ddS)Nz unsupported operand type(s) for z: '' and '')rpython_type_name)r7r8r9r:s r5binop_type_errorr?jsZ  m9mm9K9K9M9MmmVWVhVhVjVjmmm)maxsizeobj_typec t|S)z'Get all type slots for a type (cached).)rrCs r5_get_cached_slotsrFvs ( # ##r@slotcJt|\}}}}t||S)z4Check whether obj_type implements the given sq slot.rFr)rCrG seq_slots_s r5type_implements_sq_slotrL|s(*844Iq!Q It $ $$r@cJt|\}}}}t||S)z4Check whether obj_type implements the given mp slot.rI)rCrGrK map_slotss r5type_implements_mp_slotrOs(*844Ay!Q It $ $$r@)rGcJt|\}}}}t||S)z.Check whether obj_type implements the nb slot.rI)rCrGrK number_slotss r5type_implements_nb_slotrRs(-h77Aq, L$ ' ''r@c^t|\}}}}t|tjSN)rFrr TP_ITERrCrK type_slots r5type_implements_tp_iterrX s+*844Aq!Y I{2 3 33r@c^t|\}}}}t|tjSrT)rFrr TP_ITERNEXTrVs r5type_implements_tp_iternextr[s+*844Aq!Y I{6 7 77r@c^t|\}}}}t|tjS)z3Check whether obj_type implements the tp_repr slot.)rFrr TP_REPRrVs r5type_implements_tp_reprr^s+*844Aq!Y I{2 3 33r@c^t|\}}}}t|tjS)z2Check whether obj_type implements the tp_str slot.)rFrr TP_STRrVs r5type_implements_tp_strras+*844Aq!Y I{1 2 22r@c t|SrT)r[rEs r5 pyiter_checkrc"s 'x 0 00r@cNt|trdSt|S)zBImplements PySequence_Check semantics for VariableTracker objects.F) issubclassdicttype_implements_sq_itemrEs r5pysequence_checkrh)s)(D!!u "8 , ,,r@c t|S)z@Implements _PyIndex_Check semantics for VariableTracker objects.)type_implements_nb_indexrEs r5 pyindex_checkrk1s $H - --r@objc |S#t$r&td|ddgtjYdSwxYw)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_typert7s}         4===@"-          s,AAkeycontainer_namect|}|tttfvrSt |s/t t ||d|g||}|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_indexrFs%S))HT5)))'11  $%hhPSPdPdPfPfhh    ## Jr@rc t|}t|r||St|r%t ||dt |d|ddS)Nz is not a mappingzobject of type z has no len())rttype_implements_mp_length mp_lengthtype_implements_sq_lengthrr>r7rlTs r5vt_mapping_sizer]s c""A ##!}}R    ##K 4 4 6 6IIIJJJRP3+?+?+A+APPPQQQQQr@ct|}t|r||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_lenrksC c""A ##!}}R   2s # ##r@cddlm}|rt |jt |S#t $r7}tt||t|gYd}~nd}~wwxYwt|}t|r| |}||S t||}ddlm}t!||r,|||dkS|}|dkrtt$|dg|j|dkS#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_boolrzs+***** A A*#*40F0F0H0H+I+IJJ J A A A $T!WWbAx @ @ @ @ @ @ @ @ @ A%S))Hx((r""  M &R%%++++++ fo . . E"))"foo.?.?!.CDD D..00 >> $B&D%E    '&zA~666 &&&!"%%%%%& #  "4 ( ((s1.A B -BB AEAEE;:E; _repr_runningct|}t|rt|}|tvrDtdt dt jdi}tj | |dSt | | |}t |n#t |wxYwt|}t|tst!|d|jd|St!|d|ddS) zMirrors PyObject_Repr with Py_ReprEnter/Py_ReprLeave cycle detection. https://github.com/python/cpython/blob/v3.13.3/Objects/object.c#L745-L778 Resolution order: tp_repr -> 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 ] " "gtWk6GQH#*8<<%+H+HII I&!!! *]]2&&F  ! !& ) ) ) )M ! !& ) ) ) )+F33 +s++  Mk6JMMM    RQC,@,@,B,BQQQRRRRRs CC#cxt|tur|St|}t|r6t|jt jur||}nt ||}t|}t|tst|d|j d|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((' II o&> > >b!!b#&&'//K k3 ' '   H1E H H H    Mr@ct|}t|r|||St|rct|}t |r&||}t |||St||jd|jt|tr|||St|d|jddS)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 - - 5##B''C&r344 4  T TAR T T   (D!!.$$R---RMX.MMMNNNNNr@indexcFt|}t|rt|tr||}t|t rS|dkrMt |r>||}tj||z}| ||St|r&t|d| dt|d| ddS)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- . . 0022I)S)) i!mm,X66 ]]2..F,3!F$=$=$?$??EE***#H--NL!5!5!7!7LLLMMMRWS1133WWWXXXXXr@siocHt|}t|rt|tr||}t|t rS|dkrMt |r>||}tj||z}| |||St|r&t|d| dt|d| ddS)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) * * Y,,..I)S)) Yi!mm,V44Y[[__F(/ F)r7rruro_typers r5generic_setitemrKs#1 % %F'//7&&r3666"6** (--  " " :##B''C&r1c599 9 Q8L8L8N8NQQQ    O""$$ O O Or@ct|}t|r{t|trO|}|dkr5t |r&||}t|||}|||dSt|r&t|d| dt|d| ddS)Nrr=rz'' object does not support item deletion) rtrr+rrrrvt_addrrrr>)r7rrridxrs r5sequence_delitemrcs#1 % %F"6**/ a) * * .&&((CQww,V44.[[__Fr1f--A!!"a...'//LJ!3!3!5!5JJJKKK M""$$ M M Mr@ct|}t|r|||dSt|}t|r&||}t |||St |r%t|d|t|d|ddS)Nz$sequence index must be integer, not r=z ' does not support item deletion) rtrrrkr~rrrr>)r7rrurr key_values r5generic_delitemr|s#1 % %F'//6&&r3555$S))HX %%b)) Ay111 $V , ,  Os7K7K7M7MOO   RSQ//11SSSTTTTTr@cHddlm}|rVt|t r/|jt |St|}t|r| |St|r| |S|rt|tttfrl |jt |S#t$r/}t!t|t|gYd}~nd}~wwxYwt#|d|ddS)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+*****  FJs/E/E/G/G$M$MF&&s3+A+A+C+C'D'DEEE$S))Hh''#r"""))%  $$$ DJ   3y"9%%D D*#*3s/E/E/G/G+H+HII I D D D $Z3q66( C C C C C C C C C D  <"%"6"6"8"8 < < <s.E E9 %E44E9c<ddlm}|rVt|t r/|jt |St|}t|r| |St|r| |S|rt|ttfrl |jt |S#t$r/}tt|t|gYd}~nd}~wwxYwt!|d|ddS)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+*****  HJs/E/E/G/G$O$OH&&uS-C-C-E-E'F'FGGG$S))H))%  $$$ ))%  $$$ DJs/E/E/G/G#u$V$VD D*#*51G1G1I1I+J+JKK K D D D $Z3q66( C C C C C C C C C D  *$$&& * * *s .D:: E3%E..E3ct|}t|s&t|d|d||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_iternextrs^ c""A &q ) )VT3;O;O;Q;QTTTUUU    # ##r@ct|}t|r||St|d|ddS)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_negrn%S))H"8,,(##B''' C#*>*>*@*@CCCr@ct|}t|r||St|d|ddS)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||St|d|ddS)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%sn%S))H"8,,(##B''' A(<(<(>(>AAAr@c\t|}t|pt|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!!Sr""%c**E""  R9M9M9O9ORRR    !  S333333##I$6$HIIWW r    Q!5!5!7!7QQQRRRRRr@ 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_divmodrc6|tSrT)is_constant_matchNotImplemented)rs r5is_nb_not_implementedrs  # #N 3 33r@c t||S#t$rYdSwxYw)z?Check if w's underlying Python type is a proper subtype of v's.F)rer,r-)r9r8s r5is_python_subtypersI!--//1==??;;; uus36 AAop_slotcJ|d}t|}t|}t|}t||rtt ||dnd}t||rtt ||dnd} || urd} |R| 1t ||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!sh2!!!I!'*K "1 % %F "1 % %F #6; 7 7 QD)))   #6; 7 7 QD)))     "3Aq"9"9 VAr1d++F(00  F2q%(($V,, M 2q$''$V,, M  "> 2 22r@clt||||}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|r@|d}tt ||}||||}t |s|St ||||S)Nr)rtrRrrrrr!) r7r8r9r$rrrrGrs r5 binary_iop1r&s#1 % %Fvx'@AA&&& tAww **aQ$V,, M b!Q ( ((r@cnt|||||}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(sBQ8W 5 5FV$$.Q9--- Mr@ct|||d}t|s|St|}t|r|||St |||ddS)zHImplements addition via nb_add / nb_inplace_add with binary_op dispatch.r+N)r!rrttype_implements_sq_concatsq_concat_implr?)r7r8r9rrs r5rr*syAq( + +F  ( ( a  A ##'A&&&RAs#####r@c t|||dd}t|rkt|}t|r|||St |r|||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;sQ#3X > >FV$$-(++ ,X 6 6 -++B22 2 &x 0 0 -##B** * RAt , , , Mr@seqnct|}t|s&t|d|d||}t |||||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:Zs#1 % %F #F + +   R1;M;M;O;O R R R    OOB  E"2u---   b% ( ((r@ct|}t|s&t|d|d||}t |||||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=qs#1 % %F #F + +   R1;M;M;O;O R R R    OOB  E"2u---  % %b% 0 00r@r9c|}|tj dz ks|tjkrtt|dgdSdS)Nrz,cannot fit 'int' into an index-sized integerrx)rsysrBr OverflowError)r7r9r2s r5r5r5sg   ""ACKr7r8r9rrrs r5generic_multiplyrEsAq- 0 0F  ( ( "1 % %F "1 % %F (()r1a((( (()r1a(((  A    A A)*););)=)= A A Ar@ct|||dd}t|s|St|}t|}t|rt |||St |rt |||St |rt |||St|d|d|ddS)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- H HF  ( ( "1 % %F "1 % %F(001&r1a000 (()r1a(((!(()r1a(((  A    A A)*););)=)= A A Ar@Fselfotherreversec>t|}t|tjr||||St |rt |||St|d|d|ddS)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-*CDDA$$R$@@@ ++0r4///  H  ! ! # # H H,1,B,B,D,D H H Hr@ct|}t|tjr|||St |rt |||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 sp&d++Iy-*KLL8,,R777(338&r4777 Be , ,,r@opc|dkrSt||}|(|rtjdStjtS|dkrO|||d}t |r|Stj| 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__NT__ne__)r6rrr%rrichcompare_implis_richcompare_not_implemented)rIr7rJrUidentity eq_results r5object_richcomparer]Zs X~~&tU33  H$?$?$A$A #*400 0&~666 x))"eX>> )) 4 4  &9+G+G+I+I'IJJJ&~666r@c|r|stjtS|}|} t t ||||}n@#t$r3}tt|t|j Yd}~nd}~wwxYwtj|S)z=Constant-fold comparison for types with as_python_constant().rxN) rrr%rrrrr}rrry)rIr7rJrUself_val other_valrrs r5 python_constant_richcompare_implraxs  " " $ $7E,D,D,F,F7&~666&&((H((**IC,h,,XyAA CCC BT!&\\BBBBBBBBBC  "6 * **s+'B C)C  C__gt____lt____ge____le__rWrX)rcrbrerdrWrX _REFLECTED_OPz>=)rcrerWrXrbrd_OP_STRc ^t|} |}n#t$rd}YnwxYw |}n#t$rd}YnwxYwd}|@|>||ur:t||r*d}||||}t |s|S||||}t |s|S|s(||||}t |s|S|dvrlt ||} | /|dkr'tj| S| Stdd|d|d|d d gtj  dSt|d t|d |d|d dS)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_richcomparerps^b!I  O    & vv & & !##B955-f55 M  Ar * *F )& 1 1  ##B955-f55 M !!!&q!,,  X~~'.83N3N3P3P/PQQQO@:1::::R:::S2%12          E  E E""$$ E E-.-?-?-A-A E E E     s$ 33A AAc|dvrtd|t||}|,|rtj|dkSt ||||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 NrW)AssertionErrorr6rrr%rp)r7r8r9rUr[s r5generic_richcompare_boolrssw  %%%VRTVVWWW"1a((H ; ; = =&rX~666 r1a , ,,r@c,||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|||jS|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~ sjJIIIIIIIII"2s++JAw:~am&89999 "  "1 % %%r@itemct|}t|r|||St||}t j|t j|||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)r7rlrrits r5generic_containsrsx c""A"1%%,r4(((R % %$  3  -T2J + + ,r@.(_CONSTANT_FOLD_SUBCLASSCHECK_METACLASSESderivedclscb|}|}|tus |tur3tdd|d|dddgtjtjt |}|turc tj t||S#t$r3}tt|t|jYd }~nd }~wwxYwtjh}t"jd kr|t(j||vr)t)j|}t/j||}d d lm} t7|| ra|jD]E} t;||| } t7| tr| jrtj d cSFtj dSt7|trtt|t>rc tj t||S#t$r3}tt|t|jYd }~nd }~wwxYwt7|tstA|d|!|d|gi} tE|| S)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<s&5577J  - - / /F^##v'?'?=3'33S333T5"-#. %S))H4 G#*   G G G $Yaf F F F F F F F F F G?#K '!!  %   ;v&&#B--%$$$$$#}%%.I 5 5D"2w55A!-.. 517 5'.t44444&u---&$ GJ V >%% G G#*   G G G $Yaf F F F F F F F F F G fd # #   P   __R!4wi D DF F # ##s0!B)) C&3)C!!C&!H## I -)II iter_ null_or_idxcddlm}t||s,t|tur|||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_nextrss&#""""" {L 1 17 !+ . .# 5 5%%b+666 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@r5rs9  ((((((((******** ,+++++++11111111&&&&&&=<<<<<<9 9 99999x #          3$$sCc/A)B$$$$ %d%#%$%%%% %d%#%$%%%%"'"9@WXXX#G/";$G/";%,G/"C%%%!&g/"=&g/"=$G/";%,G/"C%%%! $G.": 'w."=   $+7."A$$$ $G.": (d(#($(((( !!8}?STTT%g-";&g-"; 'w-"<   #7#:AWXXX%g-";&g-";&g-";"'"9 @UVVV#G-"9$G-"9$G-"9!!8}?STTT !8}?STTT 7m>QRRR !8}?STTT"7#:AWXXX!(-">"""'.g-"C'''#'.g-"C'''#(/w-"D((($$+7-"@$$$ %,G-"A%%%!%,G-"A%%%!")-">"""")-">"""!(-"=!!!#*'-"?###")-">"""+2'-"G+++'*1-"F***&#7#:AWXXX%,W-"B&&&".5W-"J...* 4d4t4444 8$848888 4d4t4444 3T3d3333 141D1111-t-----.D.T....    4     #  . R# R*; R R R R R $# $*; $ $ $ $ $')#')*9')')')')')T#%% s3xS#S*9SSSSS@#*;<(O#(O (O (O (O(O(O(OVY#Y Y Y YYYYB#   4#     0# 2U#UU U UUUU.'#'*9'''''T&#&*9&&&&&R$#$*9$$$$$&#*9,#*9,#*9,==D==== !S#!S*9!S!S!S!S!SR(88(   ]  ]0 =,=< m"m2=,=< m"m2 m" m2!"}4#$ -D#2+B!.)>(/64/4d4444_"F3#F3F3F3 F3  F3F3F3F3R#     @)#))) )  )  ))))&#     ($#$$$ $$$$"# >)#) )) )))).1#1 11 1111*  #          # <# R #     ,-#- - - ----V"77 7#7 7 7  7777<+ +#+ + +  ++++( !! tCH~ c3hE #E E E  E  E E E E P-#--- -  ----0 # *9  39    &#&*9&&&&&$,#,*;,CT,,,,,4K HM>(%c *:_$#_$ _$ _$ _$_$_$_$D!#! !!! ?O +, !!!!!!r@