Bj-dZddlZddlZddlZddlmZmZmZddlm Z m Z m Z m Z ddl mZddlmZmZmZddlmZmZmZmZdd lmZmZdd lmZmZdd lmZm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)d dl*m+Z+m,Z,m-Z-m.Z.m/Z/d dl0m1Z1d dl2m3Z3m4Z4m5Z5d dl6m7Z7e rddl8m9Z9ddl:m;Z;d dlGdde/Z?Gdde>Z@Gdde>ZAGdde/ZBGdd eBZCGd!d"eBZDGd#d$eBZEe3eFzZGGd%d&ejHjIeGe/fZJGd'd(e>ZKdS))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)CallableIteratorSequence)AnyLiteral TYPE_CHECKINGUnion) MappingKey)graph_break_hints polyfills variables)create_call_functioncreate_call_methodcreate_dup_topcreate_instruction)raise_observed_exception unimplemented) GuardBuilder install_guard) AttrSourceDictGetItemSourceis_constant_sourceis_from_local_source)cmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatch)AttributeMutationExistingAttributeMutationNewNO_SUCH_SUBOBJValueMutationNewVariableTracker)ConstantVariable)HashableTracker is_hashableraise_unhashable) SetVariable) PyCodegen)InstructionTranslator)UserFunctionVariablec eZdZeZejZejZ dhe j Z efdee e fde de ddffd Zde de fdZdee e ffdZdefd Zdee e ffd Zdee e ffd Zde fd Zd e defdZddddde dee deee fde f dZddddde dee deee fdee dfde fdZdefdZdefdZde dzde defdZd5d!Z d5d"Z!ddd#e de fd$Z"ddd#e de fd%Z#d#e de dzfd&Z$d#e ddfd'Z%d6d(Z&ddd)e'e ddfd*Z(ddd+e de fd,Z)ddd-ed)e'e deee fde f fd. Z*ddde'e fd/Z+ddde fd0Z,ddd-ede-fd1Z.de de ffd2 Z/defd3Z0ddd-effd4 Z1xZ2S)7ConstDictVariableuser_clsitemskwargsreturnNc Rd|vr|dd|vr|dtjd i|tt fd|DsJdt tdfddffd ||}|fd|D|_|j rt|j  nd |_ | |_ ||_dS) Noriginal_itemsshould_reconstruct_allc3vK|]3\}}t|tfot|tV4dSN) isinstancer&).0xvHashables `/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/torch/_dynamo/variables/dicts.py z-ConstDictVariable.__init__..is^  1 q?H5 6 6 /1o..      keyr(r4c>t|r|n |Sr9)r:)rBr>s r? make_hashablez1ConstDictVariable.__init__..make_hashableos&%S(33F33# FrAc.i|]\}}||SrF)r;r<r=rDs r? z.ConstDictVariable.__init__..us)MMMtq!}}Q//MMMrATrF)popsuper__init__r(allr2r r&_get_dict_cls_from_user_clssourcerr7copyr6r1)selfr2r1r3dict_clsr>rD __class__s @@r?rJzConstDictVariable.__init__Ws{ v % % JJ' ( ( ( #v - - JJ/ 0 0 0""6""""              G(99: G  G G G G G G 33H==XMMMMu{{}}MMMNN 6:[ J$T[11 1 1d #$jjll  rActtjtjf|vr|}n t fd|jD}|vs J||tjurt}|S)Nc3$K|] }|v|V dSr9rF)r;baseaccepted_dict_typess r?r@z@ConstDictVariable._get_dict_cls_from_user_cls..s7T=P5P5P5P5P5P5PrA)dict collections OrderedDict defaultdictnext__mro__)rOr1rPrUs @r?rLz-ConstDictVariable._get_dict_cls_from_user_cls~s#[%UV * * *HH!)!1H....... {. . .HrAcHd|jDS)Ncli|]1\}}|j|2SrF)vtas_proxyr;kr=s r?rGz.ConstDictVariable.as_proxy..s0MMM$!Q MMMrAr2rOs r?r_zConstDictVariable.as_proxys$MM$*:J:J:L:LMMMMrAcg}|jD]\}}t|jdrt|jjn|j}t|drt|jn|}||d|dd|zdzS)Nvaluez: {, })r2hasattrr^reprre debug_reprappendjoinrOr2rar=key_strval_strs r?rkzConstDictVariable.debug_reprsJ$$&& 2 2DAq*1!$*@*@Wd14:&&&adooFWFWG'.q'':':Nd17mmm G LLG00w00 1 1 1 1TYYu%%%++rAcHd|jDS)Ncli|]1\}}|j|2SrFr^as_python_constantr`s r?rGz8ConstDictVariable.as_python_constant..sH   1 D # # % %q';';'='=   rArbrcs r?rtz$ConstDictVariable.as_python_constants1   ((**    rAcp|d|jDS)NcHi|]\}}|j| SrFrsr`s r?rGz=ConstDictVariable.keys_as_python_constant..s,LLLA''))1LLLrAinstall_dict_keys_match_guardr2rcs r?keys_as_python_constantz)ConstDictVariable.keys_as_python_constants6 **,,,LL9I9I9K9KLLLLrAc|jSr9r1rcs r? python_typezConstDictVariable.python_types }rAr^ct|tsJt}t|sdS||}||jvo%t|j|t j SNF)r:r&r(r)r2rDeletedVariable)rOr^r>rBs r? __contains__zConstDictVariable.__contains__sp"o....."2 5hrlldj   JsOY6* * &  rAtxr- tree_map_fnr.map_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|SrFr^r; key_trackernew_items_hasheds r?rGz:ConstDictVariable.call_tree_map_branch..3" " "  N,[9" " " rATr2r6r7rM mutation_type) realizer:r0lenr2_tree_map_fallbackrltypeKeyError call_tree_mapcloner%) rOrrrrr other_dicts candidaterresibling_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|SrFrrs r?rGzDConstDictVariable.call_tree_map_with_path_branch..rrATr)rr:r0rr2_tree_map_with_path_fallbackrlrrr^rtr call_tree_map_with_pathrr%)rOrrrrrrrrrrer 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 dSr9)r:rrr;r<s r?r@z(ConstDictVariable.len..sC  => 1i788 8      rA)sumr2valuesrcs r?rzConstDictVariable.lens>  BF*BSBSBUBU      rAcvjp1tfdjDS)Nc3K|]:\}}j|j|V;dSr9) is_new_itemr6getr^)r;rBrerOs r?r@z2ConstDictVariable.has_new_items..!s^2 2 U   T044SV<U&&(( >U->->-@-@ >emmoo&&"U]]__*=*== =%yyBuII%%rAcodegenr,cHd}|jD]a\}}|j|j}|||s|jr ||j|||dz }b|td|dS)Nrr! BUILD_MAParg)r2r6rr^rr7 append_outputr)rOrnum_argsrBreitems r?reconstruct_kvs_into_new_dictz/ConstDictVariable.reconstruct_kvs_into_new_dict.s***,,  JC&**3622De,, 0K A 0(KKKLLLLLrAc|jtjur.fd|rȉgt ddt| t d| gtdtddS| t dddS|rtddtg||  td ddS| dS) NctdgS)NrX extend_outputcreate_load_python_modulerWcreate_load_attrrsr?z/ConstDictVariable.reconstruct..>s>--99+FF00??rArFupdater!POP_TOPrr DICT_UPDATE)r1rWrX add_push_null_contains_self_referencerrr add_cacher load_methodrrrrOrs `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;;;;;rArc|t|}||jvrtt||g|j|S)Nargs)r(r2rr)rOrrrBs r?'getitem_const_raise_exception_if_absentz9ConstDictVariable.getitem_const_raise_exception_if_absentisAc"" dj $Xr > > > >z#rAct|}||jvr3d|jd}tdd|j|dgtj|j|S)NzDictionary key z not found during tracingzkey not found in dictzKey z>Check if the key exists in the dictionary before accessing it.gb_typecontext explanationhints)r(r2rerr USER_ERROR)rOrrrBmsgs r? getitem_constzConstDictVariable.getitem_constqs{c"" dj HCIHHHC /*sy**T&1     z#rAcPt|}||jvrdS|j|Sr9)r(r2)rOrrBs r?maybe_getitem_constz%ConstDictVariable.maybe_getitem_consts-c"" dj 4z#rAcZ||vsJt|}t|j|}t|j|}t |tjr| dSdSr9) r(tupler2keysindexr6r:rLazyVariableTrackerr)rOrrBroriginal_key_vts r?realize_key_vtz ConstDictVariable.realize_key_vtsd{{{{c""djoo''((..s33 3 8 8 : :;;EB oy'D E E &  # # % % % % % & &rAcp|jr.t|tjdSdSr9)rMr make_guardrDICT_KEYS_MATCHrcs r?rxz/ConstDictVariable.install_dict_keys_match_guards< ; I $//,*FGG H H H H H I IrArc 4|jsdS|jj|rdS|d|v}|dj|drz|rt |jnt |j}t| tj ||d dS|djr5|r| |ddS|dSdS)Nr)rB)rMoutput side_effects is_modifiedis_python_constantrCONTAINS_GUARDNOT_CONTAINS_GUARDrr functoolspartialrtrrx)rOrrcontainsguard_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 5rArBc.|||Sr9)rrOrrBs r?mp_subscript_implz#ConstDictVariable.mp_subscript_impls;;BDDDrAnamec !ddlm}ddlm}t|dkri|jt g|Ri|}jj ||j |j tj dS|dkr|s|r2t|dt|dt|d||jr$jj|jt)|S|d krt|r"t|d t|d ||jr$jj|jt+|S|d kr|s|r2t|dt|dt|d||jr$jj|j|s|rt-t.t1|S|d kr||s|r2t|dt|dt|d||j t7dS|dkr|r|ot;|d}|st=|d||st|dkr2t|dt|dt|djj ||d|j |d<tj dS|dkr|r|ot;|d}|rw|d|_jj ||j |dtj dStC"|||S|dkrt|dvr"t|dt|d |ot;|d}|st=|d|d|vrE|#|t|dkrtj dS|dS|$|dS|dkr|rt|dvr"t|dt|d |ot;|d}|st=|d|d|vrF|#|t|dkrt-tJ|dSd|_jj ||j &|dS|dkr|rtO|j(t r>tO|j(tRj*st|rt||j st-tJdg|j(tRj*urt|dksd|vrt|dkr5|d+r|d,} nP|-dx} r)| +r| ,} nt||j .| \} } n|j .\} } d|_jj |t_j0| j1| gS|dkr|s|r2t|dt|dt|dd|_jj ||j 2tj dS|dkrh|rS|t|dk} t|dk} | s| rjj || rtg|dthr#|d|d}n"|t |d}|j |j | r;fd | D}|j |tj dStC"|||S|d!krt|s2t|d"t|dt|d|ot;|d}|st=|d|#||d|v}tkj6|S|d#krk|rVt|dvr2t|d$t|dt|d|ot;|d}|st=|d||st|dkr2t|d%t|dt|d|7|d}||St|dkrtj d}n|d}jj |||j |d<|S|d&kr#|jj ||d|vrt-tJd} t|dkr4|d+r|d,} |rEd|vrA|d+r'|-d,} |d}|j 8|| tj dS|d'krts|thrrt|dkr"t|d(t|d | ttj;<||dgiS|d)kr1tkj6|"d'||j= S|d*krt|dkr"t|d(t|d |d}ts|tht^j>t^j?frtg|t^j>r,|j@Jtg|j@thsJ|j@}|j(t ur |j(}|}n |j(}|}|||j t7d|+}|d|j |dj |St-t.d,|AjBd-|AjBd.gdS|d/kr|"d|||S|d0krqdd1lCmD}|jr8t|js$jj|j||Ft72StC"|||S)3Nr!)DictBuiltinVariable)SourcelessBuilderrJr2z0 args and 0 kwargsz args and z kwargsrz0 args argsrrN)r2rrM __setitem__rr z2 args and 0 kwargs __delitem__Tr)r!r z 1 or 2 argsrHpopitemzpopitem(): dictionary is emptyrlast)rclearrcTi|]$\}}tj||%SrF)r&build)r;rar=r>rs r?rGz1ConstDictVariable.call_method..sF''' Aq!!6r1!=!=>>'''rArzmore than 1 args and 0 kwargs setdefaultz1 or 2 args and 0 kwargszat most 2 args and 0 kwargs move_to_end__eq__1 args__ne____or__)r2rrMr1z$unsupported operand type(s) for |: 'z'and ''__ior____iter__ListIteratorVariabler)Grbuilderrr(call_custom_dictrVrrmutationr2rr'creater rrxrMguard_on_key_orderaddDictItemsVariableDictKeysVariabler TypeErrorDictValuesVariablerrNr% is_mutabler)r*r7rrI call_methodrrrrH issubclassr1rWrXrrtrrr TupleVariabler^rr:r0r&rrrrr dict___eq__ call_functionreUserDefinedDictVariableDefaultDictVariable_base_vtr|__name__listsr runpack_var_sequence)rOrrrr3rr temp_dict_vt arg_hashablerr=rahas_arg has_kwargsdict_vtkwargs_hashablerrer<rBrr1to_cpy new_dict_vtr r>rQs ` @r?rzConstDictVariable.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)) ) X   v #)4yy@@CKK@@@   . . 0 0 0{ > ,00=== 8v 8(B777%d++ + V^^  . . 0 0 0 v #)4yy@@CKK@@@  ::joo''7G7I7IRV ] " "t'8'8 "8KQ$8$8L . a"---  . . 0 0 0 Ta#)4yy@@CKK@@@  I " + +D 1 1 1,0GDJxxQ(( )#*400 0 ] " "t'8'8 "8KQ$8$8L C22444.2+ &//555 &&xxQ'8'8999'.t444ww**2tT6BBB U]]4yy&&#BmD =P=P=PQQQ8KQ$8$8L . a"---Awd""00T:::t99>>+24888Aw%%b$q'22 2 U]]t00]4yy&&#BmD =P=P=PQQQ8KQ$8$8L . a"---Awd""00T:::t99>>,Xr:::Aw*.D ' I " + +D 1 1 1:>>((47"3"344 4 Y  4??#4#4 4=$// ."4=+2IJJ .II . $B---: (8} 777D Q&F"2"2t99>>d1g&@&@&B&B>75577DD!::f---a213G3G3I3I2//11DD'D111z))t)4411z))++1*.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 0$ii1nGVqJ C* C &//555 5!$q'+<==Q==???"&q'"5"F"Fd1g##J%%gm4447'''''$*LLNN'''OJ%%o666'.t444ww**2tT6BBB ^ # #t99 #34yy@@CKK@@@  8KQ$8$8L . a"---  , ,R 6 6 6Aw$H"(X66 6 \ ! !doo&7&7 !4yy&&#.4yy@@CKK@@@  8KQ$8$8L . a"---  . . 0 0 0 TQ#14yy@@CKK@@@  ,,T!W55E  t99>>(/55AAQA &//55501 88DG,,- ] " "  . . 0 0 0 I " + +D 1 1 1Awd""(2666D4yyA~~$q'"<"<">">~Aw1133 ?&F**vf~/P/P/R/R*zz&))<<>>(47##C J " "3T " 2 2 2#*400 0 X  & ## # 4yyA~~#Bh3t998K8K8KLLL$++B 0EFFTTT47OR X  "($$R4@@FF X  4yyA~~#Bh3t998K8K8KLLLGE&%51- eY%FGG+ >555%en6GHHHHH!NE =,,#}H!FF$~H"F44666$ll*//++"2"4"4% + Q55777!((a777""(@t?O?O?Q?Q?Z@@ % 1 1 3 3 <@@@Y     R4 8 8 8K Z   3 3 3 3 3 3{ >#5dk#B#B > ,00===''((,,> >rAcL|d|jDS)Ncg|] }|j SrFrrs r? z9ConstDictVariable.unpack_var_sequence..:s))))))rArwrOrs r?r"z%ConstDictVariable.unpack_var_sequence8s* **,,,))dj))))rAcx|tj|t|jS)z Mapping length for dict objects.)rxr&rrr2r.s r? mp_lengthzConstDictVariable.mp_length<s/ **,,,$RTZ999rAcrtfdttjtjfDrKt j|rtjdSjturtjdSdjd}tddj|dgtj  dS) Nc3*K|] }j|uVdSr9r{)r;trOs r?r@z5ConstDictVariable.call_obj_hasattr..Fs>   MQ       rATFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr) rrVrWrXrYrir1r'rrr SUPPORTABLE)rOrrrs` r?call_obj_hasattrz"ConstDictVariable.call_obj_hasattrAs     K3[5LM      6t}d++ 5'.t444}$$'.u555.s?G))55e<<,,-?@@rAr!F)rMrr r4rrNrr)rOrrs ` r?rz MappingProxyVariable.reconstructzs ; LT[LLL  ?04;00Y&2               21e<<=====rAc|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 mutationrVz, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rMrrhas_existing_dict_mutationr)rOrrs r?_check_mutation_guardz*MappingProxyVariable._check_mutation_guards ; 291LLNN 7  GH4;HHHdh           rArBcb|||j||Sr9)rZrNrrs r?rz&MappingProxyVariable.mp_subscript_impls0 ""2&&&|--b#666rArrcf|||j||||Sr9)rZrNr)rOrrrr3s r?rz MappingProxyVariable.call_methods4 ""2&&&|''D$???rAc6|j|Sr9)rNr0r.s r?r0zMappingProxyVariable.mp_lengths|%%b)))rAc|tjur&tj||tjjvSt ||Sr9)r|rRrSr&r__dict__rIr5r=s r?r5z%MappingProxyVariable.call_obj_hasattrsW     !7 7 7"(TU5K5T-TUU Uww''D111rAr>)rr-r4N)r r@rArRrSrBr0rrJrr|rIr&r"rrZrrFrVrr0r'r5rJrKs@r?rMrMjs*M 1ST &T&&&&4&=4$BW4444>>>>6,7 #77  7777@ #@@?# @ S/)* @  @@@@*3*****2)2142 2222222222rArMc6eZdZddZdddeeddfdZdS) NNModuleHooksDictVariabler4NcdSr9rFrcs r?rxz7NNModuleHooksDictVariable.install_dict_keys_match_guard rArr-rcdSr9rFrOrrs r?rz5NNModuleHooksDictVariable.install_dict_contains_guard  rAr?)r r@rArxrIr&rrFrAr?rarasY     ) 15o1F       rArac eZdZejZ ddeeefdededzde ddf fd Z de ffd Z de ffd Zed ede fd Zd d dedeffd Zd d de deedee efdef fd Zd d de deffd ZddZxZS)rNr2r1default_factoryr3r4c tj||fi||tjusJ|t jd}||_dSr9)rIrJrWrYr'rrh)rOr2r1rhr3rQs r?rJzDefaultDictVariable.__init__s^ 33F333;22222  ".5d;;O.rAc|jtttfvr |jsdSt Sr~)rhrIrrVr2rIrrOrQs r?rz&DefaultDictVariable.is_python_constants<  eT': : :4: :5ww))+++rAc|jJd|jdtdS)Nz defaultdict(rg))rhrkrIrks r?rkzDefaultDictVariable.debug_reprsL#/// W4/::<< W W@R@R@T@T W W W rArct|tjtjjtjjfp"t|tjo|jduSr9)r:rBaseBuiltinVariable functionsBaseUserFunctionVariablePolyfilledFunctionVariabler'rers r?is_supported_argz$DefaultDictVariable.is_supported_argsY -#<#>   Qi899Oci4>O QrArr-rBc&||vr|||S|jrtt||gdS|j|gi}t |d||gi|S)Nrr)rrhis_constant_nonerrrrIr)rOrrB default_varrQs r?rz%DefaultDictVariable.mp_subscript_impls $;;%%b#.. .   0 0 2 2  $Xr > > > > > >.<> > X  4yyA~~#Bh3t998K8K8KLLL"(Y-BCCQQT47OR 77&&r4v>> >rAcb|dkr|jSt||S)Nrh)rhrIr<r=s r?r<zDefaultDictVariable.var_getattr+s4 $ $ $' 'ww""2t,,,rArr,cfd|jgtddt |t d|gtdtddS)NctdgS)NrYrrsr?rz1DefaultDictVariable.reconstruct..7s>G))55kBB,,];;rAr!Frr) rrhrrrrrrrrrrs `r?rzDefaultDictVariable.reconstruct4s         $%%% %a//       $n..///H%%% **7333 #A&& "9--      rAr9r>)r r@rArWrYrBrVr&rrrJrGrrFrk staticmethodrsrrIrr<rrJrKs@r?rrs+M 37 / /O_45 / /)4/ /  /  / / / / / /,D,,,,,, C      QoQ$QQQ\Q #  $? #???# ? S/)* ?  ??????8- #--  ------        rArc eZdZUdZdZedzed<dededdffd Z e defdZ e de e fd Zd d de e fd ZddZd d dedefdZd d dede e deee fde f fd Zd d de fdZxZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method NkvrNr3r4c tjdi||jdvsJt|tsJ||_dS)N)rrr2rF)rIrJrr:r0rNrPs r?rJzDictViewVariable.__init__[sV""6"""w55555'#455555 rAcb|jJt|jj|jSr9)rgetattrrNr2rcs r? view_itemszDictViewVariable.view_itemsas/w"""3wt|)4733555rActr9)NotImplementedErrorrcs r? view_items_vtzDictViewVariable.view_items_vtfs "!rArr-c|jSr9)rr.s r?r"z$DictViewVariable.unpack_var_sequencels !!rArr,c|jJ||j||j|ddS)Nr)rrNrrrs r?rzDictViewVariable.reconstructosRw""" DG$$$ArArc|jJ||jvrtjdStjdS)NTF)rr|r_r'r)rOrrs r?r5z!DictViewVariable.call_obj_hasattrusNw""" 4##%%. . .#*400 0&u---rArc|dkr$ddlm}||jtS|dkr't j||St||||S)Nrr!r r __repr__) r!r rr%r&rrkrIr)rOrrrr3r rQs r?rzDictViewVariable.call_method}s :   3 3 3 3 3 3''"2B2D2D Z  "(T__->->?? ?ww""2tT6:::rAcPtj|t|jS)z&Sequence length for dict view objects.)r&rrrr.s r? sq_lengthzDictViewVariable.sq_lengths$RT_)=)=>>>rAr>)r r@rA__doc__rrF__annotations__r0rrJpropertyrrIr&rr"rr'r5rVrrrJrKs@r?r~r~Rs Bd  1ST 6C666X6"tO4"""X" "&="$BW"""" .).14. ....; #;;?# ; S/)* ;  ;;;;;;"?3?????????rAr~c eZdZeZdZedeefdZ ede efdZ de fdZ defdZddd ed e ed eeefdef fd ZxZS) rrr4c*t|jSr9)setrrcs r? set_itemszDictKeysVariable.set_itemss4?###rAc$d|jDS)Ncg|] }|j SrFrrs r?r-z2DictKeysVariable.view_items_vt..s......rArrcs r?rzDictKeysVariable.view_items_vts/.do....rActSr9)rrcs r?r|zDictKeysVariable.python_typesrAc|jsdSg}|jD]^}t|jdrt|jjn|j}||_dd|zdzS)Nz dict_keys([])rez dict_keys([,])rrir^rjrerkrlrm)rOr2raros r?rkzDictKeysVariable.debug_reprs :"?!E_ & &(/g(>(>UD$$$ADOODUDU W%%%% 388E??2T9 9rArr-rrr3c,|dkr|j||||S|dvr:t|j|}||dj}t |S|t vrt |dttjttfstj |tStj |t ||j|djSt||||S)Nr__and____iand__rr__sub____isub____xor____ixor__r)rNrrrr+rr:rUserDefinedSetVariablerrr&rNotImplementedrI)rOrrrr3mrrQs r?rzDictKeysVariable.call_methods > ! !<++BdFCC C     --A$q'#$$Aq>> ! + + +Q4%$  A',R@@@"(&t,T^T!W=NOO ww""2tT6:::rA)r r@rArrBrrrr&rrIrrr|rFrkrVrrJrKs@r?rrsM B $3/$$$X$/tO4///X/T :C : : : :&; #&;&;?# &; S/)* &;  &;&;&;&;&;&;&;&;&;&;rArcVeZdZeZdZedeefdZ de fdZ de fdZ dS)rrr4c*t|jSr9)rIrrcs r?rz DictValuesVariable.view_items_vtsDO$$$rActSr9)rrcs r?r|zDictValuesVariable.python_typesrAc|jsdSg}|jD]O}t|drt|jn|}||Pdd|zdzS)Nzdict_values([])rez dict_values([rr)rrirjrerkrlrm)rOr2r=rps r?rkzDictValuesVariable.debug_reprs <$$!E_ & &+21g+>+>R$qw---ALLNN W%%%%"SXXe__4t; ;rAN)r r@rArrBrrrIr&rrr|rFrkrFrAr?rrsM B %tO4%%%X%Tz.DictItemsVariable.set_items..sE   1 I3QT1I>> ? ?   rArrcs r?rzDictItemsVariable.set_itemss%      rAc$d|jDS)NcJg|] \}}tj|j|g!SrF)rrr^r`s r?r-z3DictItemsVariable.view_items_vt..s-OOOtq! 'q 22OOOrArrcs r?rzDictItemsVariable.view_items_vtsPOtOOOOrActSr9)rrcs r?r|zDictItemsVariable.python_typesrAc|jsdSg}|jD]\}}t|jdrt|jjn|j}t|drt|jn|}|d|d|ddd|zdzS) Nzdict_items([])re(rgrmz dict_items([rrrrns r?rkzDictItemsVariable.debug_reprs ;##!E 8 81(/g(>(>UD$$$ADOODUDU,31g+>+>R$qw---ALLNN 666G6667777!CHHUOO3d: :rArr-rrr3c8|dkrt|dkr"t||dt|dt|dtr)|j|d|djgiSt|dt tjttfr9tj |t|j |dj z dkStjdS|dkr$ddlm}||jt% S|d vr:t'|j |}||dj }t |S|t(vrt|dt tjttfstj |t*Stj |t)||j |dj St-||||S) Nrr!rrrFrr r r)rr r:rrNrr+rrrr&rrr'rr!r rr%rrrrI) rOrrrr3r fn_hdlret_valrQs r?rzDictItemsVariable.call_methods* 8  4yyA~~#Bh3t998K8K8KLLL$q'#455 |//HtAw>OQSTTTQ4%$  ',a)::;;q@$*511 1 Z   3 3 3 3 3 3''"2B2D2D     T^T22FfT!W.//Gw'' ' + + +Q4%$  A',R@@@"(&t,T^T!W=NOO ww""2tT6:::rAFcdS)zC Dictionary item views are not hashable in Python. FrFrcs r?r9z$DictItemsVariable.is_python_hashableVr:rA)r r@rArrBrrrrrIr&rrr|rFrkrVrrr9rJrKs@r?rrs8M B  301   X PtO4PPPXPT ;C ; ; ; ;?; #?;?;?# ?; S/)* ?;  ?;?;?;?;?;?;BGENrArc eZdZdZededeeeffdZ edddedeeeffdZ dedddd fd Z d e defd Z deeeffd Zd e defdZd e dedd fdZd e dd fdZd e defdZdefdZdeefdZd S)SideEffectsProxyDictzm A proxy dict that allows us to track mutations to the dict using side effects table as storage. r^r4ct|tjriSt|tjrt |jS|}|turJt|tj r| dSt |dStddt|dt|gt jdS)Nr_z-unsupported variable type for __dict__ accesszVariableTracker type: z.Dynamo does not know how to get __dict__ from r)rrNestedUserFunctionVariabler:LocalGeneratorFunctionVariablerget_example_value_dictr^get_real_python_backed_valuer$UserDefinedObjectVariable_getattr_staticobject__getattribute__rrr DYNAMO_BUG)r^res r?rz+SideEffectsProxyDict.get_example_value_dictfs "i: ; ; I ID E E '>>ruEE E3355EN**b)"EFFF--j999!225*EEEK?T"XX?? [QUVXQYQY [ [*5 rArr-c|t}fd|DS)Nc i|]E\}}|tj|jo"tt jd|FS)r_)rM)r&rrMrr)r;rBrerr^s r?rGz;SideEffectsProxyDict.get_value___dict__..sm   U &yN%jJ&G&GMM    rA)rrr2)rr^example_value_dicts`` r?get_value___dict__z'SideEffectsProxyDict.get_value___dict__sX2HHLL     16688    rArNcl||_|jj|_||||_dSr9)rrrr item_dict)rOrrs r?rJzSideEffectsProxyDict.__init__s1 I200T::rArBcft}t||r|jn|Sr9)r(rr^rt)rOrBHashers r?_maybe_unwrap_keyz&SideEffectsProxyDict._maybe_unwrap_keys/ .4S&.A.AJsv((***sJrAcL|jj|jiSr9)rstore_attr_mutationsrrrcs r?side_effects_tablez'SideEffectsProxyDict.side_effects_tables  599$)RHHHrAc||}|j|j|r"|j|j|dS|j|S)NT) deleted_ok)rrhas_pending_mutation_of_attrr load_attrrrOrBrs r? __getitem__z SideEffectsProxyDict.__getitem__s`%%c**   9 9$)T J J Q$..ty$4.PP P~d##rArec||}t|tsJ|j|j||dSr9)rrrFr store_attrr)rOrBrers r?rz SideEffectsProxyDict.__setitem__sN%%c**dC      $$TYe<<<<) )rAc4td|DS)Nc3K|]}dVdS)r!NrF)r;_s r?r@z/SideEffectsProxyDict.__len__..s"##1######rA)rrcs r?__len__zSideEffectsProxyDict.__len__s##d######rAc#lKt}|}|D]?\}}t|tjr |t j|V@|jD](\}}||vr|t j|V)dSr9) r(rr2r:rrr'rr)rOrdrar=s r?rzSideEffectsProxyDict.__iter__s   # # % %GGII 5 5DAq!Y677 &)03344 4 4 4 4N((** 9 9DAqzzf-4Q7788888 9 9rA)r r@rArr|r&rVrFrrrrJkVrrrrrrGrrHrrr(rrFrAr?rr`s ?tCK7H\2  #  )8  c?" #    \  ;_;2I;d;;;; KRKCKKKKIDo)=$>IIII$r$o$$$$ =r=/=d==== SrSdSSSS**t****$$$$$ 9(?3 9 9 9 9 9 9rArceZdZdZedddeddfdZdedddeddffd Zd e d eddfd Z d e defd Z d e de fdZ d e degefdefdZddZdddeeddfdZxZS)DunderDictVariablezrepresents object.__dict__rr-r^r4c|jrtn t}|jot|jd}|||||S)Nr_)rrrM)rMr"r#r)clsrr^rrMs r?rzDunderDictVariable.creates` 35)W,...AUAWAW@z")Z@@s "     rAr3Nc ftjifi|t|||_dSr9)rIrJrr2)rOr^rr3rQs r?rJzDunderDictVariable.__init__s9 &&v&&&)"b11 rArrec||j|<dSr9rb)rOrres r?setitemzDunderDictVariable.setitems  4rAc|j|Sr9rbrOrs r?getitemzDunderDictVariable.getitemsz$rAc||jvSr9rbrs r?rzDunderDictVariable.containsstz!!rAdefaultc||r||S|}||j|<|Sr9)rrr2)rOrrres r?getitem_or_defaultz%DunderDictVariable.getitem_or_defaultsF ==   <<%% %GIIE$DJt LrAcdSr9rFrcs r?rxz0DunderDictVariable.install_dict_keys_match_guardrcrArcdSr9rFres r?rz.DunderDictVariable.install_dict_contains_guardrfrAr?)r r@rAr classmethodr&rrrJrFrrrGrrrrxrIrrJrKs@r?rrs$$  #          [  2 2 $2 2  222222!C!!D!!!! C O    "S"T""""  "o-.           ) 15o1F         rAr)LrrWrrRcollections.abcrrrtypingrrrr torch.utils._pytreer r r r rbytecode_transformationrrrrexcrrguardsrrrMrrrrutilsrrrrrr rTr"r#r$r%r&constantr'hashabler(r)r*setsr+torch._dynamo.codegenr,torch._dynamo.symbolic_convertr-rpr.r0rMrarr~rrrrFrabcMutableMappingrrrFrAr?rsn& 8888888888555555555555******6666666666 :999999900000000 '&&&&&DDDDDDDDDD0//////DDDDDD//////\ -\ -\ -\ -\ -\ -\ -\ -~]2]2]2]2]2?]2]2]2@     1   z z z z z +z z z z>?>?>?>?>?>?>?>?BD;D;D;D;D;'D;D;D;N<<<<<)<<<2ggggg(gggTsc9c9c9c9c9;?9"o:MNc9c9c9L9 9 9 9 9 *9 9 9 9 9 rA