tjdZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZmZmZddlmZmZmZmZdd lmZmZmZdd lmZmZdd lmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+d dl,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2d dl3m4Z4d dl5m6Z6m7Z7m8Z8d dl9m:Z:d dl;mZ>ddl?m@Z@d dlAmBZBde2deCfdZDde2deCfdZEGdde2ZFGdde2ZGGddeFZHGd d!e2ZIGd"d#eIZJGd$d%eIZKGd&d'eIZLe6eMzZNGd(d)ejOjPeNe2fZQGd*d+eFZRdS),aI Dictionary-related variable tracking classes for PyTorch Dynamo. This module implements variable tracking for different types of dictionary-like objects: - Regular Python dictionaries (dict) - Ordered dictionaries (collections.OrderedDict) - Default dictionaries (collections.defaultdict) - Dictionary views (keys and values) These classes are responsible for tracking dictionary operations during graph compilation, maintaining proper guards for dictionary mutations and key existence checks. They handle dictionary creation, modification, key/value access, and view operations while ensuring correct behavior in the compiled code through appropriate guard installation. The implementation uses a special HashableTracker wrapper to handle dictionary keys while preserving proper aliasing semantics. Set-related classes live in sets.py. N)CallableIterator)Any TYPE_CHECKINGUnion) MappingKey)graph_break_hints polyfills variables)create_call_functioncreate_call_methodcreate_dup_topcreate_instruction)raise_observed_exceptionraise_type_error unimplemented) GuardBuilder install_guard) AttrSourceDictGetItemSourceis_constant_sourceis_from_local_source) _item_debug_reprcmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatch tracked_reprunpack_iterable)AttributeMutationExistingAttributeMutationNewAttrMutationKindNO_SUCH_SUBOBJValueMutationNewVariableTracker)ConstantVariable)HashableTracker is_hashableraise_unhashable) vt_getitem) SetVariable) PyCodegen)InstructionTranslatorBase)UserFunctionVariableobjreturncPt|tSN) issubclass python_typedict)r3s [/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/_dynamo/variables/dicts.py pydict_checkr;Qs coo'' . ..c |}n#t$rYdSwxYwt|ttt t fS)z3PyAnySet_Check(other) || PyDictViewSet_Check(other)F)r8NotImplementedErrorr7set frozensetrr)r3ts r:_is_set_or_dictviewrBVsS OO   uu a#y)Z@ A AAs  %%ceZdZeZejZejZ dhe j Z efdee e fde de ddffd Zde de fdZdee e ffdZdefd Zdee e ffd Zd d de fd Zdee e ffdZde fdZde defdZd d ddde dee deee fde f dZd d ddde dee deee fdee dfde fdZdefdZdefdZde dzde defdZ dBd"Z!dBd#Z"defd$Z#d d d%e de fd&Z$d d d%e de fd'Z%d%e de dzfd(Z&d%e ddfd)Z'dCd*Z(d d d+ee ddfd,Z)d d d-e de fd.Z*d d d/e de fd0Z+d d de fd1Z,d d d2ed+ee deee fde f fd3 Z-d d dee fd4Z. dDd e de d6ede fd7Z/d e de de fd8Z0d d d-e de dzde ffd9 Z1d d de fd:Z2d d d2ede3fd;Z4de de ffd< Z5defd=Z6d d deeeffd>Z7d d de d?ede fd@Z8d d d2effdA Z9xZ:S)EConstDictVariableuser_clsitemskwargsr4Nc ld|vr|dd|vr|dtjd i|tt fd|Dst ddttdfddffd | |}|fd |D|_|j rt|j  nd |_ | |_||_dS) Noriginal_itemsshould_reconstruct_allc3vK|]3\}}t|tfot|tV4dSr6) isinstancer)).0xvHashables r: z-ConstDictVariable.__init__..}s^  1 q?H5 6 6 /1o..      r<zZAll keys must be VariableTracker or HashableTracker and all values must be VariableTrackerkeyr+r4c>t|r|n |Sr6)rL)rRrPs r: make_hashablez1ConstDictVariable.__init__..make_hashables&%S(33F33# Fr<c.i|]\}}||SrV)rMrNrOrTs r: z.ConstDictVariable.__init__..s)MMMtq!}}Q//MMMr<TrV)popsuper__init__r+allrFAssertionErrorrr)_get_dict_cls_from_user_clssourcerrJcopyrIrE)selfrFrErGdict_clsrPrT __class__s @@r:rZzConstDictVariable.__init__ks v % % JJ' ( ( ( #v - - JJ/ 0 0 0""6""""            !9   G(99: G  G G G G G G 33H==XMMMMu{{}}MMMNN 6:[ J$T[11 1 1d #$jjll  r<cttjtjf|vr|}n t fd|jD}|vrt d||tjurt}|S)Nc3$K|] }|v|V dSr6rV)rMbaseaccepted_dict_typess r:rQz@ConstDictVariable._get_dict_cls_from_user_cls..s7T=P5P5P5P5P5P5Pr<zUnexpected dict_cls: )r9 collections OrderedDict defaultdictnext__mro__r\)r`rErarfs @r:r]z-ConstDictVariable._get_dict_cls_from_user_clss#[%UV * * *HH!)!1H . . . !C!C!CDD D {. . .Hr<cHd|jDS)Ncli|]1\}}|j|2SrV)vtas_proxyrMkrOs r:rWz.ConstDictVariable.as_proxy..s0MMM$!Q MMMr<rFr`s r:rozConstDictVariable.as_proxys$MM$*:J:J:L:LMMMMr<cg}|jD]B\}}t|j}t|}||d|Cdd|zdzS)N: {, })rFrrnappendjoinr`rFrqrOkey_strval_strs r: debug_reprzConstDictVariable.debug_reprsJ$$&& 2 2DAq&qt,,G&q))G LLG00w00 1 1 1 1TYYu%%%++r<cHd|jDS)Ncli|]1\}}|j|2SrVrnas_python_constantrps r:rWz8ConstDictVariable.as_python_constant..sH   1 D # # % %q';';'='=   r<rrrss r:rz$ConstDictVariable.as_python_constants1   ((**    r<txr1cfd|jD}tjdd|zdzS)Ncfg|]-\}}t|jdt|.S)rur!rnrMrRvaluers r: z/ConstDictVariable.repr_impl..sQ   UB'' D D<E+B+B D D   r<rvrwrx)rFr)buildrzr`rrFs ` r: repr_implzConstDictVariable.repr_implsd    "j..00   $Rtyy/?/?)?#)EFFFr<cp|d|jDS)NcHi|]\}}|j| SrVrrps r:rWz=ConstDictVariable.keys_as_python_constant..s,LLLA''))1LLLr<install_dict_keys_match_guardrFrss r:keys_as_python_constantz)ConstDictVariable.keys_as_python_constants6 **,,,LL9I9I9K9KLLLLr<c|jSr6rErss r:r8zConstDictVariable.python_types }r<rnct|tstdt|t |sdSt |}||jvo%t|j|tj S)NzExpected VariableTracker, got F) rLr)r\typer,r+rFr DeletedVariable)r`rnrRs r: __contains__zConstDictVariable.__contains__s"o.. N !L$r((!L!LMM M2 5b!!dj   JsOY6* * &  r< tree_map_fnr2map_fnresttree_map_kwargsc  g}|D]}|}t|tr*t|jt|jkr||||||cS||t|j |jD]v\}} g} |D]P} | |j|$#t$r ||||||cYccSwxYw| |||| | |<w fd D} | | ddtS)Nc,i|]}|j|SrVrnrM key_trackernew_items_hasheds r:rWz:ConstDictVariable.call_tree_map_branch..3" " "  N,[9" " " r<TrFrIrJr^ mutation_type) realizerLrDlenrF_tree_map_fallbackryrKeyError call_tree_mapcloner() r`rrrrr other_dicts candidaterrsibling_leavesupdated_original_itemsrs @r:call_tree_map_branchz&ConstDictVariable.call_tree_map_branchs02  * *I!))++Ii):;; s@@TZ@!@!.. VT?   y ) ) ) )+4 ++--"&*"2"2"4"4   K46N(   ")))/+*FGGGG22K-2,?,? -- [ ) )" " " " /" " "  zz"1#'*,,    s C..#DDkeypath.c 6g}|D]}|}t|tr*t|jt|jkr|||||||cS||t|j|jD]\} } g} |D]Q} | |j| $#t$r!|||||||cYccSwxYw| j } |t| fz} | |||| || | <fdD}| |ddtS)Nc,i|]}|j|SrVrrs r:rWzDConstDictVariable.call_tree_map_with_path_branch..1rr<Tr)rrLrDrrF_tree_map_with_path_fallbackryrrrnrrcall_tree_map_with_pathrr()r`rrrrrrrrrrr key_const child_keypathrrs @r:call_tree_map_with_path_branchz0ConstDictVariable.call_tree_map_with_path_branchs502  * *I!))++Ii):;; s@@TZ@!@!88 VT?G   y ) ) ) )+4 ++--"&*"2"2"4"4   K46N(   ")))/+*FGGGG<<K$99;;I#z)'<'<&>>M,1,I,I -- [ ) )" " " " /" " "  zz"1#'*,,    s C//$DDcbtd|jDS)Nc3LK|]}t|tj V dSr6)rLr rrMrNs r:rQz(ConstDictVariable.len..?sC  => 1i788 8      r<)sumrFvaluesrss r:rzConstDictVariable.len>s>  BF*BSBSBUBU      r<cvjp1tfdjDS)Nc3K|]:\}}j|j|V;dSr6) is_new_itemrIgetrn)rMrRrr`s r:rQz2ConstDictVariable.has_new_items..Ds^2 2 U   T044SV<U&&(( >U->->-@-@ >emmoo&&"U]]__*=*== =%yyBuII%%r<codegenr0cHd}|jD]a\}}|j|j}|||s|jr ||j|||dz }b|td|dS)Nrr# BUILD_MAParg)rFrIrrnrrJ append_outputr)r`rnum_argsrRritems r:reconstruct_kvs_into_new_dictz/ConstDictVariable.reconstruct_kvs_into_new_dictQs***,,  JC&**3622De,, 0K A 0(KKKLLLLLr<c|jtjur.fd|rȉgt ddt| t d| gtdtddS| t dddS|rtddtg||  td ddS| dS) NctdgS)Nrh) extend_outputcreate_load_python_modulergcreate_load_attrrsr:z/ConstDictVariable.reconstruct..as>--99+FF00??r<rFupdater#POP_TOPrr DICT_UPDATE)rErgrh add_push_null_contains_self_referencerr r add_cacher load_methodrrrr`rs `r: reconstructzConstDictVariable.reconstruct]s% =K3 3 3  ! !   ,,.. F%%-a77&(( !!$'''%%n&6&6777##H---227;;;%%+A..*955227;;;%%&:1e&D&DEEEEE,,.. <%%*;A>>>&(( !!$'''227;;;%%&8A&N&N&NOOOOO227;;;;;r<ct|jtjs|rt ddfd|jDdS)Nrvrwc3K|]9\}}|jd|V:dS)ruN)rnreconstruct_pycode)rMrqrOrs r:rQz7ConstDictVariable.reconstruct_pycode..soEEimijlm!8!8!A!AddQEYEYZaEbEbddEEEEEEr<rx)r7rErgrhrr>rzrFrs `r:rz$ConstDictVariable.reconstruct_pycodes t}k&= > > &,,.. && %IDIIEEEEquq{rBrBrDrDEEEEEIII Ir<rc|t|}||jvrtt||g|j|S)Nargs)r+rFrr)r`rrrRs r:'getitem_const_raise_exception_if_absentz9ConstDictVariable.getitem_const_raise_exception_if_absentsAc"" dj $Xr > > > >z#r<ct|}||jvr3d|jd}tdd|j|dgtj|j|S)NzDictionary key z not found during tracingzkey not found in dictKey z>Check if the key exists in the dictionary before accessing it.gb_typecontext explanationhints)r+rFrrr USER_ERROR)r`rrrRmsgs r: getitem_constzConstDictVariable.getitem_consts{c"" dj HCIHHHC /*sy**T&1     z#r<cPt|}||jvrdS|j|Sr6)r+rF)r`rrRs r:maybe_getitem_constz%ConstDictVariable.maybe_getitem_consts-c"" dj 4z#r<c|||vrtd|dt|}t|j|}t|j|}t|tj r| dSdS)Nrz not found in dict) r\r+tuplerFkeysindexrIrLr LazyVariableTrackerr)r`rrRroriginal_key_vts r:realize_key_vtz ConstDictVariable.realize_key_vts d?? !?!?!?!?@@ @c""djoo''((..s33 3 8 8 : :;;EB oy'D E E &  # # % % % % % & &r<cp|jr.t|tjdSdSr6)r^r make_guardrDICT_KEYS_MATCHrss r:rz/ConstDictVariable.install_dict_keys_match_guards< ; I $//,*FGG H H H H H I Ir<rc 4|jsdS|jj|rdS|d|v}|dj|drz|rt |jnt |j}t| tj ||d dS|djr5|r| |ddS|dSdS)Nr)rR)r^output side_effects is_modifiedis_python_constantrCONTAINS_GUARDNOT_CONTAINS_GUARDrr functoolspartialrrr)r`rrcontainsguard_fns r:install_dict_contains_guardz-ConstDictVariable.install_dict_contains_guards7"{  F 9 ! - -d 3 3  F7d? 7> !d1g&@&@&B&B !-5XT ))4::;X  %  G6688     !W^ 5 5##DG,,,,,2244444  5 5r<rRc.|||Sr6)rr`rrRs r:mp_subscript_implz#ConstDictVariable.mp_subscript_impls;;BDDDr<rct|s&t|d|d|||g||v}t j||S)Nunhashable type: '')r,rpython_type_namerr)r)r`rrrs r: sq_containszConstDictVariable.sq_containssq4   R R!Pd6K6K6M6M!P!P!P Q Q Q ((dV4444<$R222r<c(ddlm}|jrdt|jsPt |t j|jj |j||j S)Nr#) DictIterator) iterrr^rrrrrrguard_on_key_orderaddrFr)r`rrs r: tp_iter_implzConstDictVariable.tp_iter_impls&&&&&& ; :1$+>> : $//,*FGG H H H I ( , ,T[ 9 9 9|DJOO--...r<namec Rddlm}t}|dkri|j|tg|Ri|}|jj||j |jtj dS|dkr|s|r2t||dt|dt|d||jr$|jj|jt%|S|dkrt|r"t||d t|d ||jr$|jj|jt'|S|d kr|s|r2t||dt|dt|d||jr$|jj|jd t)t+|jD}t/j|t3 S|dkr|s|r2t||dt|dt|d||jr$|jj|j|s|rt5t6|t9|S|dkr||s|r2t||dt|dt|d||jt3dS|dkrt|dvr"t||dt|d |d|vrE|||t|dkrtj dS|dS| ||dS|dkr|!rt|dvr"t||dt|d |d|vrF|||t|dkrt5tD||dSd|_#|jj||j$||dS|dkr|!rt|rt|||jst5tD|dg|j%\} } d|_#|jj|t/j&| j'| gS|dkr|s|r2t||dt|dt|dd|_#|jj||j(tj dS|dkrR|!r=|t|dkr"t||dt|d |s|r|jj||rz|d} tS| tTr5| |j | jn(tS| t.j+tXfrv| -|d.rN| /|dgi} ta|| D]$} tc|| | |j|| <%nteta|| D]s\}}ta||}t|dkr,t5tf|d|dt|d g|d|j||d<t|D]+\} } | |j|tij5|| <,tj dS|d!kr<|!r't|dvr2t||d"t|dt|d||st|dkr2t||d#t|dt|d|6|d}||St|dkrtj d}n|d}|jj|||j||d<|Sto/||||S)$Nr#)DictBuiltinVariablerZrF0 args and 0 kwargs args and  kwargsrz0 argsz args __reversed__cg|] }|j SrVr)rMrqs r:rz1ConstDictVariable.call_method..?sMMMaQTMMMr<rrr_)rFrr^r)r#r z 1 or 2 argsrrXTpopitemzpopitem(): dictionary is emptyrclearrzat most 1 argsr z$dictionary update sequence element #z has length z; 2 is required setdefaultz1 or 2 args and 0 kwargszat most 2 args and 0 kwargs)8rr+call_custom_dictr9rrmutationrFrr*creater rrr^rrDictItemsVariableDictKeysVariablereversedlistrr ListIteratorVariabler(r TypeErrorDictValuesVariablerr_rr is_mutablerrJrXr  TupleVariablernr!rLrDUserDefinedObjectVariableMappingProxyVariablecall_obj_hasattrr call_methodr"r. enumerate ValueErrorr)rrrY)r`rrrrGrrP temp_dict_vt reversed_keysrqrOrrrRidxrpairrrNrbs r:r3zConstDictVariable.call_methods  *)))))" :  ?.?D#)L I " + +D 1 1 1 J  l0 1 1 1#*400 0 W__ v #)4yy@@CKK@@@   . . 0 0 0{ > ,00===$T** * V^^4yy M#Bh3t998K8K8KLLL  . . 0 0 0{ > ,00===#D)) ) ^ # # v #)4yy@@CKK@@@   . . 0 0 0{ > ,00===MM8D9J9J4K4K+L+LMMMM1-=-?-? X   v #)4yy@@CKK@@@   . . 0 0 0{ > ,00=== 8v 8(B777%d++ + V^^  . . 0 0 0 v #)4yy@@CKK@@@  ::joo''7G7I7IRV U]]4yy&&#BmD =P=P=PQQQAwd""00T:::t99>>+24888Aw%%b$q'22 2 U]]t00]4yy&&#BmD =P=P=PQQQAwd""00T:::t99>>,Xr:::Aw*.D ' I " + +D 1 1 1:>>((47"3"344 4 Y  4??#4#4 4yy .#B---: (8:%%''DAq*.D ' I " + +D 1 1 1*AD!955 5 W__ v #)4yy@@CKK@@@  +/D ' I " + +D 1 1 1 J     #*400 0 X  $//"3"3   . . 0 0 04yy1}}#B.>3t99@S@S@STTT 6v 6 &//555 @Qe%677@77999J%%ek2222"<>RS@ ..r6::MMOO @!,,RR@@D.r488OO4>r5#4N4N 88C==11O&/r5/I/I%J%J @ @ T.r488t99>>4 * "%T(+%T%T9>(/55AAQA &//55501 88DG,,-77&&r4v>> >r<cL|d|jDS)Ncg|] }|j SrVrrs r:rz9ConstDictVariable.unpack_var_sequence..s))))))r<rr`rs r:unpack_var_sequencez%ConstDictVariable.unpack_var_sequences, **,,,))dj))))r<Freversec|r||fn||f\}}t|rBt|r3||dgi}||d|gi|StjtS)Nr_r)r;r3r*r&NotImplemented)r`rrr>self_other_news r: nb_or_implzConstDictVariable.nb_or_impls*1C tUm v    <#7#7 ##BB77C OOB6(B 7 7 7J&~666r<c8||d|gi|SNr)r3)r`rrs r:nb_inplace_or_implz$ConstDictVariable.nb_inplace_or_impls% Xw333 r<c|s#t|||St|st |||t |}|jj ||C||j vr*tt|| gd|_|j |=n ||j |<tjdS)NrT)r.rYmp_ass_subscript_implr,r-rr+rrr%rFrrrrJr*r&)r`rrRrhkeyrbs r:rIz'ConstDictVariable.mp_ass_subscript_impls   A7700S%@@ @3 & S" % % % **,,,s## ''--- =4:%%(2S=S=S=U=U."s>   MQ       r<TFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr) rr9rgrhrihasattrrEr*r&rr SUPPORTABLE)r`rrrs` r:r2z"ConstDictVariable.call_obj_hasattrs     K3[5LM      6t}d++ 5'.t444}$$'.u555:: :%!233 ? %!899 ?>)()FGGG'.~>>>%,,R1FGGUU u r   >>"(1M1M1O1O-OPP Pr<c|dkr'tj||St||S)Nrb)r)rr8rY var_getattrr`rrrbs r:rkzConstDictVariable.var_getattr_sG ;  "(T-=-=-?-?@@ @ww""2t,,,r<rr0r4Nr4NF);__name__ __module__ __qualname__r9 _cpython_typer DICT_CONTAINSrDICT_NOT_CONTAINSrr)_nonvar_fieldsrrrZr]rostrr~rrrr8boolrr*rrrintrrrrrrrrrrrrr rrr3r=rDrGrIrLr*r2rr,rZrirk __classcell__rbs@r:rDrD_s(M!/N%7   'N)!)!O_45)!)! )!  )!)!)!)!)!)!VDT(N$sCx.NNNN,C,,,, DcN    G7GOGGGGMc?.B)CMMMMT    4     0 '0 ,0  0 ?# 0 c?23 0  0 0 0 0 d4 '4 ,4  4 ?# 4 c?23 4 sCx4  4 4 4 4 l S     t    &$t+&4C& &&&& M M M M-<-<-<-<^ISIIII-4C -4C "?T;Q &/&d&&&&IIII(5-(559/5J(5 (5(5(5(5TE 'EE  EEEE3-35D3 3333/://///Z? 'Z?Z??# Z? S/)* Z?  Z?Z?Z?Z?Z?Z?x*-* o **** 7 7  7 7 7  7 7 7 7   - '--% -  ------2:7:O::::  - 58     2'c'o''''''TN7NE#t).s?G))55e<<,,-?@@r<r#F)r^rr rPrr}rr )r`rrs ` r:rz MappingProxyVariable.reconstruct}s ; LT[LLL  ?04;00Y&2               21e<<=====r<c|jr@|jjr$d}t dd|jd|ddgdSdSdS)NaA dict has been modified while we have an existing mappingproxy object. A mapping proxy object, as the name suggest, proxies a mapping object (usually a dict). If the original dict object mutates, it is reflected in the proxy object as well. For an existing proxy object, we do not know the original dict it points to. Therefore, for correctness we graph break when there is dict mutation and we are trying to access a proxy object.z-mapping proxy affected by dictionary mutationrz, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)r^rrhas_existing_dict_mutationr)r`rrs r:_check_mutation_guardz*MappingProxyVariable._check_mutation_guards ; 291LLNN 7  GH4;HHHdh           r<rRcb|||j||Sr6)rr}r r s r:r z&MappingProxyVariable.mp_subscript_impls0 ""2&&&|--b#666r<rrcf|||j||||Sr6)rr}r3)r`rrrrGs r:r3z MappingProxyVariable.call_methods4 ""2&&&|''D$???r<c6|j|Sr6)r}rr<s r:rz!MappingProxyVariable.tp_iter_impls|((,,,r<rc8|j||Sr6r}rr`rrs r:rz MappingProxyVariable.sq_contains|''D111r<c6|j|Sr6)r}rLr<s r:rLzMappingProxyVariable.mp_lengths|%%b)))r<rr\c4ddlm}|||j||S)Nr#)generic_richcompare)object_protocolrr})r`rrr\rs r:riz%MappingProxyVariable.richcompare_impls/ 988888""2t|UB???r<c|tjur&tj||tjjvSt ||Sr6)r8rrr)r__dict__rYr2rls r:r2z%MappingProxyVariable.call_obj_hasattrsW     !7 7 7"(TU5K5T-TUU Uww''D111r<rm)rr1r4N)rprqrrrrrsrDrrZrr8rryrxrZr*r)r=rrr rwr9r3rrrLrir*r2rzr{s@r:r1r1esc*M 1ST &T&&&&*7*E#t)>>>6,7 '77  7777@ '@@?# @ S/)* @  @@@@-:-----2-25D2 2222 *7*O****@-@6E@KN@ @@@@2-2582 2222222222r<r1c6eZdZddZdddeeddfdZdS) NNModuleHooksDictVariabler4NcdSr6rVrss r:rz7NNModuleHooksDictVariable.install_dict_keys_match_guard r<rr1rcdSr6rVr`rrs r:rz5NNModuleHooksDictVariable.install_dict_contains_guard  r<rn)rprqrrrr*r)rrVr<r:rrsY     - 59/5J       r<rc eZdZUdZdZedzed<dededdffd Z e defdZ de fd Z d d deee ffd Ze deefd Zd d deefdZddZd d dedefdZd d defdZd d dedeedeeefdef fd Zd d defdZ d d d dede defdZ d d d dede defdZ d d d dede defdZ d d d dede defdZxZ S)!DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method Nkvr}rGr4c tjdi||jdvrtd|jt |t stdt |||_dS)N)rrrFz4Expected kv to be 'keys', 'values', or 'items', got rrV)rYrZrr\rLrDrr}rs r:rZzDictViewVariable.__init__s""6""" 75 5 5 RtwRR '#455 U !SDMM!S!STT T r<c||jtdt|jj|jS)Nzkv must not be None)rr\getattrr}rFrss r: view_itemszDictViewVariable.view_itemss7 7? !677 73wt|)4733555r<cdSrSrVrss r:r,zDictViewVariable.is_hashablerTr<rr1cVddlm}||d|ddSrVrWrYs r:rZzDictViewVariable.hash_implr[r<ctr6)r>rss r: view_items_vtzDictViewVariable.view_items_vt s "!r<c|jSr6)rr<s r:r=z$DictViewVariable.unpack_var_sequences !!r<rr0c|jtd||j||j|ddS)Nz#kv must not be None for reconstructr)rr\r}rr3rs r:rzDictViewVariable.reconstructsZ 7? !FGG G DG$$$Ar<rc|jtd||jvrt jdSt jdS)Nz(kv must not be None for call_obj_hasattrTF)rr\r8rr*r&)r`rrs r:r2z!DictViewVariable.call_obj_hasattrsV 7? !KLL L 4##%%. . .#*400 0&u---r<c|jdkr?dfd|jD}tjd|dS|jdkr?dfd|jD}tjd|dSdfd |jD}tjd |dS) Nrrwc3BK|]}t|jVdSr6r)rMrRrs r:rQz-DictViewVariable.repr_impl..(s/RR3l2sv66RRRRRRr< dict_keys([])rc38K|]}t|VdSr6)r!)rMrrs r:rQz-DictViewVariable.repr_impl..+s-SS%l2u55SSSSSSr< dict_values([c3rK|]1\}}dt|jdt|dV2dS)(rw)Nrrs r:rQz-DictViewVariable.repr_impl..-se  U G R(( F FLU,C,C F F F      r< dict_items([)rrzrr)rrs ` r:rzDictViewVariable.repr_impl&s 7f  IIRRRR$/RRRRRE"(-D5-D-D-DEE E 7h  IISSSS4?SSSSSE"(-FU-F-F-FGG G     "o     $R)A)A)A)ABBBr<rc |dkr|s|r2t||dt|dt|d|jjrBt |jjs)|jj|jjtj tt|j tSt||||S)Nrrrrr)r rr}r^rrrrr r+r*r)rr(rYr3r`rrrrGrbs r:r3zDictViewVariable.call_method3s > ! ! v #)4yy@@CKK@@@  |" F+=dl>Q+R+R F ,001DEEE1Xd01122.00 ww""2tT6:::r<cPtj|t|jS)z&Sequence length for dict view objects.)r)rrrr<s r: sq_lengthzDictViewVariable.sq_lengthKs$RT_)=)=>>>r<Frr>ctj|t||gi}||d|gi|SrFr)rr?rgr3r`rrr>ss r:rDzDictViewVariable.nb_or_implOsI  !"c * * 8 8dVR H H b(UGR000r<c|r||fn||f\}}tj|t||gi}||d|gi|S)Ndifference_updater)r`rrr>rArBrs r:nb_subtract_implz!DictViewVariable.nb_subtract_implZsd*1C tUm v  !"c * * 8 8eWb I I b-x<<<r<ctj|t||gi}||d|gi|S)Nintersection_updaterrs r: nb_and_implzDictViewVariable.nb_and_implfsJ  !"c * * 8 8dVR H H b/%"===r<ctj|t||gi}||d|gi|S)Nsymmetric_difference_updaterrs r: nb_xor_implzDictViewVariable.nb_xor_implqsJ  !"c * * 8 8dVR H H b7%"EEEr<rmro)!rprqrr__doc__rrw__annotations__rDrrZpropertyrrxr,rryrZr*r)rr=rr*r2rr9r3rrDrrrrzr{s@r:rrs7 Bd  1ST6C666X6 TN7NE#t).s......r<rrss r:rzDictKeysVariable.view_items_vts/.do....r<ctSr6)rrss r:r8zDictKeysVariable.python_typesr<rr1cddlm}|jjrBt |jjs)|jj|jj||jjS)Nr#)DictKeysIterator) rrr}r^rrrrrF)r`rrs r:rzDictKeysVariable.tp_iter_implsl****** <  B'9$,:M'N'N B I ( , ,T\-@ A A A 2333r<c|jsdSg}|jD]+}t|j}||,dd|zdzS)Nz dict_keys([])rrwrrrrnryrz)r`rFrqr|s r:r~zDictKeysVariable.debug_reprsg ;"?!E_ & &*1400 W%%%% 499U#3#33d: :r<rc8|j||Sr6rrs r:rzDictKeysVariable.sq_containsrr<rr\c ddlm}t|stjt S||t j|||t|||giSNr#r^ rcr_rBr*r&r@r dictview_richcomparergrr`rrr\r_s r:riz!DictKeysVariable.richcompare_impl /....."5)) ;#*>:: : ''  .  - !((-CB-GHH     r<rrrGc|dvr:t|j|}||dj}t|St||||SN)__and____iand____xor____ixor__rrrr/rYr3)r`rrrrGmrrbs r:r3zDictKeysVariable.call_methodsj    --A$q'#$$Aq>> !ww""2tT6:::r<)rprqrrrrsrrr?r)rr*rrr8rrwr~rrir9r3rzr{s@r:r(r(}sM B $3/$$$X$/tO4///X/T4:44444;C;;;;2-25D2 2222  - 6E KN     0; ';;?# ; S/)* ;  ;;;;;;;;;;r<r(ceZdZeZdZdZdZdZdZ dZ dZ dZ dZ defdZdddeeefffd ZedeefdZdefd Zdddefd Zddd ed edefd ZdefdZxZS)r-rNr4cdS)NTrVrss r:r,zDictValuesVariable.is_hashablestr<rr1cTtt||Sr6)rYrrZ)r`rrbs r:rZzDictValuesVariable.hash_impls"%t,,66r:::r<c*t|jSr6)r*rrss r:rz DictValuesVariable.view_items_vtsDO$$$r<ctSr6)rrss r:r8zDictValuesVariable.python_typesr<cddlm}|jjrBt |jjs)|jj|jj||jjS)Nr#)DictValuesIterator) rrr}r^rrrrrF)r`rrs r:rzDictValuesVariable.tp_iter_implsl,,,,,, <  B'9$,:M'N'N B I ( , ,T\-@ A A A!!$,"4555r<rr\c*ddlm}|||||S)Nr#)object_richcompare)rr)r`rrr\rs r:riz#DictValuesVariable.richcompare_impls- 877777!!$E2666r<c|jsdSg}|jD]&}t|}||'dd|zdzS)Nzdict_values([])rrwr)rrryrz)r`rFrOr}s r:r~zDictValuesVariable.debug_repr sf =$$!E_ & &*1-- W%%%%"TYYu%5%55< z.DictItemsVariable.set_items..sE   1 I3QT1I>> ? ?   r<rrss r:rzDictItemsVariable.set_itemss%      r<c$d|jDS)NcJg|] \}}tj|j|g!SrV)r r/rnrps r:rz3DictItemsVariable.view_items_vt..'s-OOOtq! 'q 22OOOr<rrss r:rzDictItemsVariable.view_items_vt$sPOtOOOOr<ctSr6)rrss r:r8zDictItemsVariable.python_type)sr<c|jsdSg}|jD]D\}}t|j}t|}|d|d|dEdd|zdzS)Nzdict_items([])rrwrrrrr{s r:r~zDictItemsVariable.debug_repr,s <##!E 8 81*1400*1-- 666G6667777!DIIe$4$44t; ;r<rr1rcddlm}t|s&t|d|dt |t jrt|j dkrv|j \}}t|}||j j vrtj |dS|j j |}tj |||S||j||S)Nr ) iter_containsr rF)utilsrr,rrrLr r/rrFr+r}r)ris_python_equalr)r`rrrrRvalkey_htstoreds r:rzDictItemsVariable.sq_contains7s *)))))4   R R!Pd6K6K6M6M!P!P!P Q Q Q dI3 4 4 JTZA9M9MzHC$S))FT\///&,R777\'/F"(S-@-@-H-HII I}T/r:::r<rr\c ddlm}t|stjt S||t j|||t|||giSrrrs r:riz"DictItemsVariable.richcompare_implKrr<cddlm}|jjrBt |jjs)|jj|jj||jjS)Nr#)DictItemsIterator) rrr}r^rrrrrF)r`rrs r:rzDictItemsVariable.tp_iter_implcsl++++++ <  B'9$,:M'N'N B I ( , ,T\-@ A A A  !3444r<rrrGc|dvr:t|j|}||dj}t|St||||Srr)r`rrrrGfn_hdlret_valrbs r:r3zDictItemsVariable.call_methodjsl    T^T22FfT!W.//Gw'' 'ww""2tT6:::r<)rprqrrrrsrrr?rr*r)rrr8rwr~rrirr9r3rzr{s@r:r'r'sM B  301   X PtO4PPPXPT >ruEE E3355EN**b)"EFFF--j999!225*EEEK?T"XX?? [QUVXQYQY [ [*5 r<rr1c ttjrjjdt jrjjd}t|tri}|j D]\}} |j }n7#t$r*tdd|j dgtjYnwxYwt|t"s"tdd|dgtj|||<|St&}fd| DS) Nrz#non-constant key in object __dict__zkey=z@Dynamo expects object __dict__ replacement keys to be constants.rz!non-string key in object __dict__z2Dynamo expects object __dict__ keys to be strings.c i|]E\}}|tj|jo"tt jd|FS)r)r^)r)rr^rr)rMrRrrrns r:rWz;SideEffectsProxyDict.get_value___dict__..sm   U &yN%jJ&G&GMM    r<)rLr r0rrhas_pending_mutation_of_attrr&GENERIC_SETATTR load_attrrDrFrnrr>rr rPrwrr r)rrndict_vtresultrRr key_valueexample_value_dicts`` r:get_value___dict__z'SideEffectsProxyDict.get_value___dict__s   3   i$AA  ,<    i,66r:FFG'#455 ")-"5"5"7"7..JC$'F$=$=$?$? .%$I$336OO(j"B$5$A"B &i55%$G$89$8$8(\"A$5$@"A  ).F9%% 2HHLL     16688    sB991C-,C-rNcl||_|jj|_||||_dSr6)rrrr item_dict)r`rrs r:rZzSideEffectsProxyDict.__init__s1 I200T::r<rRcft}t||r|jn|Sr6)r+rrnr)r`rRHashers r:_maybe_unwrap_keyz&SideEffectsProxyDict._maybe_unwrap_keys/ .4S&.A.AJsv((***sJr<cfdjjjiDS)Ncri|]3\}}jj|tj0||4SrV)rrrr& INSTANCE_DICT)rMrrr`s r:rWz;SideEffectsProxyDict.side_effects_table..sW   e == 4!1!?  %   r<)rstore_attr_mutationsrrrFrss`r:side_effects_tablez'SideEffectsProxyDict.side_effects_tablesT    #0EII 2  egg     r<c||}|j|j|tjr"|j|j|dS|j|SNT) deleted_ok)r"rrrr&r%rrr`rRrs r: __getitem__z SideEffectsProxyDict.__getitem__sm%%c**   9 9 It-;   Q$..ty$4.PP P~d##r<rc||}t|tstdt ||j|j||dS)NzExpected str key, got )r"rrwr\rrstore_instance_dict_attrr)r`rRrrs r: __setitem__z SideEffectsProxyDict.__setitem__sh%%c**dC   H !F$t**!F!FGG G 2249dEJJJJJr<c||}|j|j|t jdSr6)r"rr.rr rr+s r: __delitem__z SideEffectsProxyDict.__delitem__sL%%c** 22 ItY688     r<c||}|j|j|tjr=|j|j|d}t|tj  S||j vSr)) r"rrrr&r%rrLr rr)r`rRrrs r:rz!SideEffectsProxyDict.__contains__s%%c**   9 9 It-;   D%// 4D/QQE!%)BCCC Ct~%%r<c4td|DS)Nc3K|]}dVdS)r#NrV)rM_s r:rQz/SideEffectsProxyDict.__len__..s"##1######r<)rrss r:__len__zSideEffectsProxyDict.__len__s##d######r<c#lKt}|}|D]?\}}t|tjr |t j|V@|jD](\}}||vr|t j|V)dSr6) r+r'rFrLr rr*r&r)r`r!drqrOs r:__iter__zSideEffectsProxyDict.__iter__s   # # % %GGII 5 5DAq!Y677 &)03344 4 4 4 4N((** 9 9DAqzzf-4Q7788888 9 9r<)rprqrrr staticmethodr)r9rwrrrrZkVr"r'r,r/r1rxrryr6rr+r9rVr<r:r r s ?tCK7H\2/ '/ -</ c?" #/ / / \/ b;_;2M;RV;;;; KRKCKKKK  Do)=$>     $r$o$$$$KrK/KdKKKK r d    &&t&&&&$$$$$ 9(?3 9 9 9 9 9 9r<r c $eZdZdZedddeddfdZdedddeddffd Zd e d eddfd Z d e ddfd Z d e defdZ d e de fdZd e degefdefdZddd e deedee efdef fd ZddZdddeeddfdZxZS)DunderDictVariablezrepresents object.__dict__rr1rnr4c|jrtn t}|jot|jd}|||||S)Nr)rrr^)r^r$r%r)clsrrnr%r^s r:r&zDunderDictVariable.creates` 35)W,...AUAWAW@z")Z@@s "     r<rGNc ftjifi|t|||_dSr6)rYrZr rF)r`rnrrGrbs r:rZzDunderDictVariable.__init__(s9 &&v&&&)"b11 r<rrc||j|<dSr6rr)r`rrs r:setitemzDunderDictVariable.setitem1s  4r<c^||st||j|=dSr6)rrrFr`rs r:delitemzDunderDictVariable.delitem4s2}}T"" !4.. Jt   r<c|j|Sr6rrrDs r:getitemzDunderDictVariable.getitem9sz$r<c||jvSr6rrrDs r:rzDunderDictVariable.contains<stz!!r<defaultc||r||S|}||j|<|Sr6)rrGrF)r`rrIrs r:getitem_or_defaultz%DunderDictVariable.getitem_or_default?sF ==   <<%% %GIIE$DJt Lr<rc $|dkrf|s|r2t||dt|dt|dtt|jt dSt ||||S)Nr_rrr)rr^)r rrDr9rFr(rYr3rs r:r3zDunderDictVariable.call_methodKs 6>> v #)4yy@@CKK@@@  %TZ  0@0B0B4 ww""2tT6:::r<cdSr6rVrss r:rz0DunderDictVariable.install_dict_keys_match_guardarr<cdSr6rVrs r:rz.DunderDictVariable.install_dict_contains_guarddrr<rn)rprqrrr classmethodr)r&rrZrwrBrErGrxrrrKr*r9r3rrrzr{s@r:r=r=s$$  '          [  2 2 (2 2  222222!C!!D!!!!CD  C O    "S"T""""  "o-.      ; ';;?# ; S/)* ;  ;;;;;;,     - 59/5J         r<r=)Srrgrrcollections.abcrrtypingrrrtorch.utils._pytreerr#r r r bytecode_transformationr rrrrXrrrguardsrrr^rrrrrrrrrrrr r!r"rer$r%r&r'r(r)constantr*hashabler+r,r-rr.setsr/torch._dynamo.codegenr0torch._dynamo.symbolic_convertr1 functionsr2rxr;rBrDr1rrr(r-r'rwr;abcMutableMappingr r=rVr<r:r]s& ........,,,,,,,,,,******6666666666 LKKKKKKKKK00000000                      '&&&&&DDDDDDDDDD''''''0//////HHHHHH///////o/$//// B_BBBBBC -C -C -C -C -C -C -C -Lw2w2w2w2w2?w2w2w2t     1   PPPPPPPPfR;R;R;R;R;'R;R;R;jB=B=B=B=B=)B=B=B=Jd;d;d;d;d;(d;d;d;NsQ9Q9Q9Q9Q9;?9"o:MNQ9Q9Q9hR R R R R *R R R R R r<