9jx, dZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZmZmZmZmZddlmZd d lmZmZd d lmZerdd lmZd ededdfdZeddedeeeeeffdZ dede!fdZ"dede!fdZ#dede!fdZ$dede!fdZ%dede!fdZ&dede!fdZ'dedefdZ( d(d Z) d(d!Z*dddedefd"Z+ddded#edefd$Z,dddedefd%Z-dddedefd&Z.y'))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. ) lru_cache) TYPE_CHECKING)get_type_slotshas_slotPyMappingSlots PyNumberSlotsPySequenceSlots)graph_break_hints)handle_observed_exceptionObservedTypeErrorraise_observed_exceptionraise_type_error unimplemented)istype)NO_SUCH_SUBOBJVariableTrackerConstantVariable)InstructionTranslatorleftrightreturnzVariableTracker | Nonec||urtjdS|j}|j}|tu}|tu}|r0|r.||urtjdStjdS||k7rtjdSddlm}ddlm}ddlm }t||||frtjdS |j|jurtjdS ddl m} 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) SetVariabler ) variablesN)rcreateget_real_python_backed_valuerdictsrlistsrsetsr isinstance python_typeNotImplementedErrorrrExceptionVariableexc_type) rrleft_val right_val left_known right_knownrrrrs g/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/_dynamo/variables/object_protocol.pyvt_identity_comparer0&sb u}&&t,,002H224I/J>1Kk9$  # #D ) "((/ [ &&u--)#!$*L+FG&&u--    U%6%6%8 8#**51 1 9  tY001 5)55 6 MM /&&u--     s 4E## E/.E/)maxsizeobj_typect|S)z'Get all type slots for a type (cached).)r)r3s r/_get_cached_slotsr5_s ( ##cVt|\}}}}t|tjS)z>Check whether obj_type implements __len__ as sequence protocol)r5rr SQ_LENGTH)r3 seq_slots_s r/type_implements_sq_lengthr;es(*84Iq!Q I88 99r6cVt|\}}}}t|tjS)z=Check whether obj_type implements __len__ as mapping protocol)r5rr MP_LENGTH)r3r: map_slotss r/type_implements_mp_lengthr?ks(*84Ay!Q I~77 88r6cVt|\}}}}t|tjS)zRCheck whether obj_type implements the nb_bool slot (i.e. has __bool__ or __len__).)r5rrNB_BOOLr3r: number_slotss r/type_implements_nb_boolrDqs(-h7Aq, L-"7"7 88r6cVt|\}}}}t|tjS)z2Check whether obj_type implements the nb_int slot.)r5rrNB_INTrBs r/type_implements_nb_intrGws(-h7Aq, L-"6"6 77r6cVt|\}}}}t|tjS)z4Check whether obj_type implements the nb_index slot.)r5rrNB_INDEXrBs r/type_implements_nb_indexrJ}(-h7Aq, L-"8"8 99r6cVt|\}}}}t|tjS)z4Check whether obj_type implements the nb_float slot.)r5rrNB_FLOATrBs r/type_implements_nb_floatrNrKr6objc |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)rOs r/maybe_get_python_typerVsR      4e<=@"--    s+AAtxrct|}t|r|j|St|rt ||j dt |d|j dy)Nz is not a mappingzobject of type z has no len())rVr? mp_lengthr;rpython_type_namerWrOTs r/vt_mapping_sizer]sf c"A #}}R   # 4 4 677HIJR?3+?+?+A*B-PQr6cht|}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. )rVr; sq_lengthr]r[s r/ generic_lenr`s2 c"A #}}R  2s ##r6cddlm}|jr)|jt |j St |}t|r|j|}||S t||}ddl m }t||r#|j||jdkDS|j|j dkDS#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. rr)SymNodeVariablerT)constantris_python_constantr boolas_python_constantrVrD bool_implr`tensorrbr%as_proxyr r )rWrOrr3resultlengthrbs r/ generic_boolrls+ &&&tC,B,B,D'EFF$S)Hx(r"  M&R%+ fo ."))"foo.?!.CD D&&&v'@'@'BQ'FGG &!"%& #  " "4 ((s,AC-"CC'&C'keyc&|j||S)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). TODO(follow-up): use has_slot(map_slots, PyMappingSlots.MP_SUBSCRIPT) to gate Branch 1 and has_slot(seq_slots, PySequenceSlots.SQ_ITEM) to gate Branch 2, matching CPython's dispatch order. TODO(follow-up): Branch 2 (sq_item) for C extension types that only have tp_as_sequence (e.g. deque — Modules/_collectionsmodule.c:1888). Branch 3 is handled by TypingVariable.mp_subscript_impl for typing module types and by BuiltinVariable for builtin types like list[int]. Types that work via constant fold fallback (no dedicated mp_subscript_impl): TODO(follow-up): str (unicode_subscript, Objects/unicodeobject.c:13809) TODO(follow-up): bytes (bytes_subscript, Objects/bytesobject.c) )mp_subscript_impl)rWrOrms r/ vt_getitemrps6  S ))r6cddlm}|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. rrargsNzLint() argument must be a string, a bytes-like object or a real number, not '')rcrrdr%rfintr rVrG nb_int_implrJ nb_index_implstrbytes bytearray ValueErrorrrrZrWrOrr3es r/ generic_intr~s&+  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. rrrrNz9float() argument must be a string or a real number, not 'rt)rcrrdr%rffloatr rVrN nb_float_implrJrwrxryr{rrrZr|s r/ 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 N)rWrrOrrr)/__doc__ functoolsrtypingrtorch._C._dynamorrrrr r(r excr r rrrutilsrbaserrrcrsymbolic_convertrr0typetuplerur5rer;r?rDrGrJrNrVr]r`rlrpr~rr6r/rs !1&86 6 66r 3$$sCc/A)B$$ ::: 999 9d9t9 8T8d8 :t:: :t::    4   R R&7 R R $ $&7 $ $),)?))@** * * *<%+%/%o%P$-$O$$r6