9izdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z mZmZddlmZmZddlmZmZdd lmZmZdd lmZmZdd lmZmZmZm Z m!Z!m"Z"m#Z#d d l$m%Z%m&Z&d dl'm(Z(d dl)m*Z*e rddl+m,Z,ddl-m.Z.d dl/m0Z0dede1fdZ2 d2de&de dddfdZ3de&de1fdZ4Gdde&Z5Gdde&Z6Gd d!e5Z7Gd"d#e5Z8Gd$d%e5Z9Gd&d'e9Z:Gd(d)e9Z;Gd*d+e&Z<Gd,d-e<Z=Gd.d/e<Z>Gd0d1e<Z?dS)3a 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) - Sets and frozensets (implemented internally using dictionaries) 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. Sets are implemented as dictionaries with None values for efficiency and code reuse. N)Sequence)AnyOptional TYPE_CHECKINGUnion)graph_break_hints polyfills variables)create_call_functioncreate_instruction)raise_observed_exception unimplemented) GuardBuilder install_guard)is_constant_sourceis_from_local_source)cmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatchspecialize_symnode)ValueMutationNewVariableTracker)ConstantVariable)ListIteratorVariable) PyCodegenInstructionTranslator)UserFunctionVariableobjreturncRt|jddS)N__instancecheck__F)type__dict__get)r$s P/var/www/icac/venv/lib/python3.11/site-packages/torch/_dynamo/variables/dicts.pywas_instancecheck_overrider,;s" 99  ! !"5u = ==argtxr"c 2|ddlm}|j} |}n#t$rt |}YnwxYwt t|td|dt | gdS)Nrr!zunhashable type: z and variable tracker = args) torch._dynamo.symbolic_convertr" current_tx python_type Exceptionr(r TypeErrorrrealize)r.r/r"arg_types r+raise_unhashabler:?s zHHHHHH - " - / /??$$ 99 ]H]]S[[]]H[H[]]   s.A  A xct|tjr*|s|rdS|S)NT) isinstancer LazyVariableTracker is_realized is_hashableis_python_hashable)r;s r+r@r@VsT 1i344  MMOO t   ! !!r-c eZdZejZdhejZGddZe fde eefde de ddffd Z de de fd Z de e e ffd Zdefd Zde e e ffd Zde e effd Zde fdZdedefdZdddddedeede eefdef dZdefdZdefdZdeededefdZd0d Zd0d!Zddd"edefd#Zddd"edefd$Z d"edeefd%Z!d"eddfd&Z"d1d'Z#ddd(e$eddfd)Z%ddd*ed(e$ede eefdef fd+ Z&ddde$efd,Z'ddd*ede(fd-Z)de deffd. Z*d/Z+xZ,S)2ConstDictVariableuser_clsc:eZdZdZdeddfdZdefdZdefdZ dS)"ConstDictVariable._HashableTrackeraB Auxiliary opaque internal class that wraps a VariableTracker and makes it hashable This should not be seen or touched by anything outside of ConstDictVariable and its children Note that it's also fine to put VTs into dictionaries and sets, but doing so does not take into account aliasing vtr%Ncnt|}t|st|||_dSN)rr@r:rG)selfrGs r+__init__z+ConstDictVariable._HashableTracker.__init__ss8#B''Br?? % $$$DGGGr-c"t|jtjrX|js?|jr&t |jS|jS)a Computes the hash value for the wrapped VariableTracker. For unrealized LazyVariableTrackers, uses the hash of the original value to avoid realizing the tracker and inserting unnecessary guards. For all other cases, delegates to the VariableTracker's get_python_hash method. Returns: The hash value of the underlying variable tracker ) r=rGr r>r?r@hashoriginal_valueget_python_hashrJs r+__hash__z+ConstDictVariable._HashableTracker.__hash__|s{47I$ABB 6++-- 6G'')) 6 DG22445557**,, ,r-c`|j|jurdS|j|jS)a Checks equality between two _HashableTracker instances. Delegates to the VariableTracker's is_python_equal method to compare the underlying variable trackers for Python-level equality. Args: other: Another _HashableTracker instance to compare with Returns: True if the underlying variable trackers are Python-equal, False otherwise T)rGis_python_equalrJothers r+__eq__z)ConstDictVariable._HashableTracker.__eq__s0w%(""t7**5844 4r-) __name__ __module__ __qualname____doc__rrKintrQboolrVr-r+_HashableTrackerrFlsv    4     -c - - - -& 54 5 5 5 5 5 5r-r^itemskwargsr%Nc \d|vr|dd|vr|dtjd i|tjt fd|DsJdttdfddffd | |}|fd|D|_|j rt|j  nd |_ | |_||_dS) Noriginal_itemsshould_reconstruct_allc3vK|]3\}}t|tfot|tV4dSrI)r=r).0r;vHashables r+ z-ConstDictVariable.__init__..s^  1 q?H5 6 6 /1o..      r-keyrFr%c>t|r|n |SrI)r=)rirgs r+ make_hashablez1ConstDictVariable.__init__..make_hashables&%S(33F33# Fr-c.i|]\}}||Sr]r])rer;rfrks r+ z.ConstDictVariable.__init__..s)MMMtq!}}Q//MMMr-Tr])popsuperrKrCr^allr_rr_get_dict_cls_from_user_clssourcerrccopyrbrD)rJr_rDr`dict_clsrgrk __class__s @@r+rKzConstDictVariable.__init__s~ v % % JJ' ( ( ( #v - - JJ/ 0 0 0""6"""$5              G(LLM G 1 G G G G G G 33H==XMMMMu{{}}MMMNN 6:[ J$T[11 1 1d #$jjll  r-cttjtjf|vr|}n t fd|jD}|vs J||tjurt}|S)Nc3$K|] }|v|V dSrIr])rebaseaccepted_dict_typess r+rhz@ConstDictVariable._get_dict_cls_from_user_cls..s7T=P5P5P5P5P5P5Pr-)dict collections OrderedDict defaultdictnext__mro__)rJrDrtrys @r+rqz-ConstDictVariable._get_dict_cls_from_user_clss#[%UV * * *HH!)!1H....... {. . .Hr-cHd|jDS)Ncli|]1\}}|j|2Sr]rGas_proxyrekrfs r+rmz.ConstDictVariable.as_proxy..s0MMM$!Q MMMr-r_rPs r+rzConstDictVariable.as_proxys$MM$*:J:J:L:LMMMMr-czddd|jDzdzS)N{, c3|K|]7\}}|jd|V8dS)z: NrG debug_reprrs r+rhz/ConstDictVariable.debug_repr..sY=AQ14??$$88 88r-})joinr_rPs r+rzConstDictVariable.debug_reprsT iiEIZEUEUEWEW    r-cHd|jDS)Ncli|]1\}}|j|2Sr]rGas_python_constantrs r+rmz8ConstDictVariable.as_python_constant..sH   1 D # # % %q';';'='=   r-rrPs r+rz$ConstDictVariable.as_python_constants1   ((**    r-cp|d|jDS)NcHi|]\}}|j| Sr]rrs r+rmz=ConstDictVariable.keys_as_python_constant..s,LLLA''))1LLLr-install_dict_keys_match_guardr_rPs r+keys_as_python_constantz)ConstDictVariable.keys_as_python_constants6 **,,,LL9I9I9K9KLLLLr-c|jSrIrDrPs r+r5zConstDictVariable.python_types }r-rGct|tsJtj}|o@|||jvo.t|j||t j SrI)r=rrCr^rAr_r DeletedVariable)rJrGrgs r+ __contains__zConstDictVariable.__contains__st"o.....$5  ! ! # # T  * Ttz((2,,79RSSS r-r/r" 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|Sr]rG)re key_trackernew_items_hasheds r+rmz:ConstDictVariable.call_tree_map_branch..!s3" " "  N,[9" " " r-T)r_rbrcrr mutation_type) r8r=rClenr__tree_map_fallbackappendr(KeyError call_tree_mapcloner) rJr/rrrr other_dicts candidatervaluesibling_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..#DDcbtd|jDS)Nc3LK|]}t|tj V dSrI)r=r rrer;s r+rhz(ConstDictVariable.len../sC  => 1i788 8      r-)sumr_valuesrPs r+rzConstDictVariable.len.s>  BF*BSBSBUBU      r-cvjp1tfdjDS)Nc3K|]:\}}j|j|V;dSrI) is_new_itemrbr*rG)rerirrJs r+rhz2ConstDictVariable.has_new_items..4s^2 2 U   T044SV<U&&(( >U->->-@-@ >emmoo&&"U]]__*=*== =%yyBuII%%r-codegenr cHd}|jD]a\}}|j|j}|||s|jr ||j|||dz }b|td|dS)Nrr BUILD_MAPr.)r_rbr*rGrrc append_outputr )rJrnum_argsriritems r+reconstruct_kvs_into_new_dictz/ConstDictVariable.reconstruct_kvs_into_new_dictAs***,,  JC&**3622De,, 0K A 0(KKKLLLLLr-c|jtjurRfd|t dddS|dS)NctdgS)Nr| extend_outputcreate_load_python_moduler{create_load_attrrsr+z/ConstDictVariable.reconstruct..Qs>--99+FF00??r-rF)rDr{r| add_push_nullrrr rJrs `r+ reconstructzConstDictVariable.reconstructMs =K3 3 3  ! !     . .w 7 7 7  ! !"6q%"@"@ A A A A A  . .w 7 7 7 7 7r-r.cLt|}||jvru dt|d|}n4#t $r't jdt|}YnwxYwtt||g|j|S)NzDict key lookup failed for z%. Debug representation of the key is r1) rCr^r_strrr6rcreaterr)rJr/r.ri error_messages r+'getitem_const_raise_exception_if_absentz9ConstDictVariable.getitem_const_raise_exception_if_absent]s 0055 dj  O#c((OO:=..:J:JOO     0 7<#c((<<!!   %Xr H H H Hz#s'A .A>=A>ct|}||jvr3d|jd}t dd|j|dgt j|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)rCr^r_rrr USER_ERROR)rJr/r.rimsgs r+ getitem_constzConstDictVariable.getitem_constns 0055 dj HCIHHHC /*sy**T&1     z#r-cft|}||jvrdS|j|SrI)rCr^r_)rJr.ris r+maybe_getitem_constz%ConstDictVariable.maybe_getitem_consts30055 dj 4z#r-cp||vsJt|}t|j|}t|j|}t|tj r| dSdSrI) rCr^tupler_keysindexrbr=r r>r8)rJr.riroriginal_key_vts r+realize_key_vtz ConstDictVariable.realize_key_vtsd{{{{0055djoo''((..s33 3 8 8 : :;;EB oy'D E E &  # # % % % % % & &r-cp|jr.t|tjdSdSrI)rrr make_guardrDICT_KEYS_MATCHrPs r+rz/ConstDictVariable.install_dict_keys_match_guards< ; I $//,*FGG H H H H H I Ir-r2c |jsdS|jj|rdS|d|v}|dj~|drdt |tjt|j |d | dS|djr5|r| |ddS| dSdS)Nr)riinvert)rroutput side_effects is_modifiedis_python_constantrr functoolspartialr(CONTAINS_GUARDrrr)rJr/r2containss r+install_dict_contains_guardz-ConstDictVariable.install_dict_contains_guards"{  F 9 ! - -d 3 3  F7d? 7> !d1g&@&@&B&B ! %T 1 G6688#+|     !W^ 5 5##DG,,,,,2244444  5 5r-namec "ddlm}mtj|dkrvt jt j|g|Ri|}|jj ||j |j j dS|dkrQt|dkr"t||dt|d|||dS|dkr|s|r2t||d t|d t|d ||jr$|jj|jt+|S|d krt|r"t||d t|d||jr$|jj|jt-|S|dkr|s|r2t||d t|d t|d ||jr$|jj|j|s|rt/t0|t3|S|dkr||s|r2t||d t|d t|d ||j t9dS|dkrl|s|r2t||d t|d t|d |j t|j S|dkr|r|ot=|d}|st?|d|||st|dkr2t||dt|d t|d |jj ||d|j |d<j dS|dkr|r|ot=|d}|rs|d|_ |jj ||j !|dj dStE#||||S|dkrt|dvr"t||dt|d|ot=|d}|st?|d||d|vrA|$||t|dkrj dS|dS|%||dS|dkr|rt|dvr"t||dt|d|ot=|d}|st?|d||d|vrF|$||t|dkrt/tL||dSd|_ |jj ||j '|dS|dkr|rtQ|j)t r>tQ|j)tTj+st|rt|||j s(j d}t/tL||g|j)tTj+urt|dksd|vrt|dkr5|d,r|d-} nP|.dx} r)| ,r| -} nt|||j /| \} } n|j /\} } d|_ |jj |t j0| j1| gS|d kr|s|r2t||d t|d t|d d|_ |jj ||j 2j dS|d!krd|rO|t|dk} t|dk} | s| r|jj || rtg|dtr#|d|d}n"|4|t |d}|j |j | r;fd"| D}|j |j dStE#||||S|d#krt|s2t||d$t|d t|d |ot=|d}|st?|d||$|||d|v}j |S|d%krg|rRt|dvr2t||d&t|d t|d |ot=|d}|st?|d|||st|dkr2t||d't|d t|d |5|d}||St|dkrj d}n|d}|jj |||j |d<|S|d(kr||jj ||d|vrt/tL|d} t|dkr4|d,r|d-} |rEd|vrA|d,r'|.d-} |d}|j 6|| j dS|d)krto|trpt|dkr"t||dt|dt j8trj:;|||dgiS|d*kr,j |#|d)||j< S|d+krxt|dkr"t||dt|d|d}to|tt j=fsOd,|>j?d-|>j?d.}t/t0||g|j)|j)h}td/|Dr tTj+nt }|||j t9d|0}|d|j |dj |S|d1kr|#|d!|||S|d2kro|jr8t|js$|jj|jt|C|t93StE#||||S)4Nr)BuiltinVariablerrK __getitem__1 args argsrr_0 args and 0 kwargs args and  kwargsrz0 argsrrs)r_rrr__len__ __setitem__rz2 args and 0 kwargs __delitem__Tr*)rrz 1 or 2 argsrnpopitemzpopitem(): dictionary is emptyr1last)rclearupdatecJi|]\}}j|| Sr])r)rerrfrrgs r+rmz1ConstDictVariable.call_method..sH''' Aq!!8!1!8!;!;<.s/JJ!z![%<==JJJJJJr-)r_rrrrD__ior____iter__r)DrrrCr^r rz call_dictrrmutationr_rrrrrrrrguard_on_key_orderaddDictItemsVariableDictKeysVariablerr7DictValuesVariablerrsr is_mutabler@r:rcrro call_methodrrrrnr rDr{r|rrr*r TupleVariablerGrr=call_custom_dictrrrr#r dict___eq__ call_functionrUserDefinedDictVariabler5rWrrrunpack_var_sequence)rJr/rr2r`r temp_dict_vt arg_hashablerrrfrhas_arg has_kwargsdict_vtkwargs_hashablerrr;rirUerr_msgtsrD new_dict_vtrrgrus @@r+rzConstDictVariable.call_methods 87777777$5 :  D94T::D#L I " + +D 1 1 1 J  l0 1 1 1*#*400 0 ] " "4yyA~~#Bh3t998K8K8KLLL??DGLL L 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 Y   v #)4yy@@CKK@@@   . . 0 0 0*#*3tz??;; ; ] " "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>>2+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---: C-&-.NOO(2SEBBBB} 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 &//5555!$q'+<==VQ==???59!W"1"B"B2tTRSW"U"UJ%%gm4447'''''$*LLNN'''OJ%%o666.'.t444ww**2tT6BBB ^ # #t99 #34yy@@CKK@@@  8KQ$8$8L . a"---  , ,R 6 6 6Aw$H*#*844 4 \ ! !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~~#Bh3t998K8K8KLLL1)2GHHVVT47OR X  *#*$$R4@@FF X  4yyA~~#Bh3t998K8K8KLLLGE&)9+LM H<4;K;K;M;M;V<<!--//8<<<)BgYGGGG-0BJJrJJJJJ ''   . . 0 0 0**joo''.00! %K G 1 1 3 3 3   $ $T!W] 3 3 3  Y     R4 8 8 8K Z  { >#5dk#B#B > ,00==='((,,> >r-cL|d|jDS)Ncg|] }|j Sr]rrs r+ z9ConstDictVariable.unpack_var_sequence..s))))))r-rrJr/s r+r!z%ConstDictVariable.unpack_var_sequences* **,,,))dj))))r-crtfdttjtjfDrKt j|rtjdSjturtjdSdjd}tddj|dgtj  dS) Nc3*K|] }j|uVdSrIr)rerrJs r+rhz5ConstDictVariable.call_obj_hasattr..&s>   MQ       r-TFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr) rrzr{r|r}hasattrrDrrrr SUPPORTABLE)rJr/rrs` r+call_obj_hasattrz"ConstDictVariable.call_obj_hasattr!s     K3[5LM      6t}d++ 5'.t444}$$'.u555 ur-rr r%Nr%N)-rWrXrYr DICT_CONTAINSrr_nonvar_fieldsr^rzr(rrKrqrrrrrr5r\rrrr[rrrrrrrrrrrlistrrr!rr3rrA __classcell__rus@r+rCrCds!/N   'N 2525252525252525n%!%!O_45%!%! %!  %!%!%!%!%!%!NDT&N$sCx.NNNN C     DcN    Mc?.B)CMMMMT  4    0 #0 ,0  0 ' 0 c?23 0  0 0 0 0 d S     t    &o.&7F& &&&& M M M M8888 )0? ")0? "8O;T &/&d&&&&IIII&5)&515o1F&5 &5&5&5&5Pa? #a?a??# a? S/)* a?  a?a?a?a?a?a?F *&=*$BW**** ) 14     2'c'o''''''r-rCc eZdZdededdffd ZdefdZdddee fd Z dd Z ddd e dee de e e fde f dZddd e deffd ZxZS)MappingProxyVariabledv_dictr`r%Nc vtjdi|t|tsJ||_dSr5)rorKr=rCrArJrAr`rus r+rKzMappingProxyVariable.__init__GsB""6"""'#455555 r-ctjSrI)typesMappingProxyTyperPs r+r5z MappingProxyVariable.python_typeLs %%r-r/r"c6|j|SrI)rAr!r.s r+r!z(MappingProxyVariable.unpack_var_sequenceOs|//333r-rr c|jr3d|jd}tdd|j|dgtjfd|jtdd dS) Nz*Preexisting MappingProxyVariable (source: zS) cannot be reconstructed because the connection to the original dict will be lost.z%mapping proxy cannot be reconstructedSource: zCUse a mapping proxy constructed in the same `torch.compile` region.rctdgS)NrF)rrrErrsr+rz2MappingProxyVariable.reconstruct..cs?G))55e<<,,-?@@r-rF)rrrr r2rrArr )rJrrs ` r+rz MappingProxyVariable.reconstructRs ; LT[LLL  ?04;00Y&2               21e<<=====r-rr2c|jr>|jjr d}t dd|jd|ddg|j||||S)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 mutationrIz, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rrrrhas_existing_dict_mutationrrAr)rJr/rr2r`rs r+rz MappingProxyVariable.call_methodms ; 291LLNN 7  GH4;HHHdh     |''D$???r-c|tjur%tj|tjjvSt ||SrI)r5rErFrrr)ror3)rJr/rrus r+r3z%MappingProxyVariable.call_obj_hasattrsU     !7 7 7#*453I3R+RSS Sww''D111r-r8)rWrXrYrCrrKr(r5r<rr!rrrzrrr3r=r>s@r+r@r@Es< 1ST &T&&&&4&=4$BW4444>>>>6@ #@@?# @ S/)* @  @@@@:2)2142 2222222222r-r@c6eZdZddZdddeeddfdZdS) NNModuleHooksDictVariabler%NcdSrIr]rPs r+rz7NNModuleHooksDictVariable.install_dict_keys_match_guards r-r/r"r2cdSrIr]rJr/r2s r+rz5NNModuleHooksDictVariable.install_dict_contains_guards  r-r9)rWrXrYrr<rrr]r-r+rOrOsY     ) 15o1F       r-rOc eZdZ ddeeefdedeededdf fd Zde ffd Z de ffd Z e d ede fd Zd d de deedee efdef fd ZddZxZS)DefaultDictVariableNr_rDdefault_factoryr`r%c tj||fi||tjusJ|t jd}||_dSrI)rorKr{r}rrrU)rJr_rDrUr`rus r+rKzDefaultDictVariable.__init__s^ 33F333;22222  ".5d;;O.r-c|jtttfvr |jsdSt S)NF)rUr<rrzr_rorrJrus r+rz&DefaultDictVariable.is_python_constants<  eT': : :4: :5ww))+++r-c|jJd|jdtdS)Nz defaultdict(r))rUrrorXs r+rzDefaultDictVariable.debug_reprsL#/// W4/::<< W W@R@R@T@T W W W r-r.ct|tjr!|jtt t tfvSt|tjj tjj fSrI) r=r rfnr<rrzset functionsBaseUserFunctionVariablePolyfilledFunctionVariablers r+is_supported_argz$DefaultDictVariable.is_supported_argsY c94 5 5 6dE455 5'@'B r-r/r"rr2c.|dkrt|dkr"t||dt|d|d|vr|||dSt|jt r,|jj tt||dgdS|j |gi}t |d|d|g||St ||||S)Nrrrrrr1r) rrrrrUrrrrrror)rJr/rr2r` default_varrus r+rzDefaultDictVariable.call_methods = 4yyA~~#Bh3t998K8K8KLLLAw$))"d1g6664/1ABB ',2:,Xra JJJJJJ"&"6"D"DRR"P"PKGG''MDG[+A6'&77&&r4v>> >r-rr cfd|j|t dddS)NctdgS)Nr}rrsr+rz1DefaultDictVariable.reconstruct..s>G))55kBB,,];;r-rF)rrUrrr rs `r+rzDefaultDictVariable.reconstructs}        $%%% **733321e<<=====r-rIr8)rWrXrYrzrr(rrrKr\rrr staticmethodrar<rrr=r>s@r+rTrTso 6: / /O_45 / /"/2 /  /  / / / / / /,D,,,,,, C       o $   \ ? #???# ? S/)* ?  ??????8 > > > > > > > >r-rTc VeZdZdZejZdeede ddffd Z de fdZ e dedfd Zedefd Zde fd Zdefd Zde fd ZddZddde deedee efdef dZddde deedee efdef fd ZdddedefdZddZxZS) SetVariablez.We model a sets as dictionary with None valuesr_r`r%Nc t|t}t j|fi|dSrI)rzfromkeysrh_default_valuerorK)rJr_r`rus r+rKzSetVariable.__init__sI  e[%?%?%A%ABB))&)))))r-ch|jsdSddd|jDzdzS)Nzset()r,c3HK|]}|jVdSrIrrers r+rhz)SetVariable.debug_repr...!H!H!$//"3"3!H!H!H!H!H!Hr-rr_rrPs r+rzSetVariable.debug_reprs?z O7!H!HTZ!H!H!HHHH3N Nr-rFcNt|jSrI)r]r_rrPs r+ set_itemszSetVariable.set_itemss4:??$$%%%r-c*tjdSrI)rrr]r-r+rkzSetVariable._default_value s &t,,,r-c$d|jDS)Nc@h|]}|jSr]rros r+ z'SetVariable.as_proxy..s"888A 888r-rsrPs r+rzSetVariable.as_proxys888888r-ctSrI)r]rPs r+r5zSetVariable.python_types r-c$d|jDS)Nc@h|]}|jSr]rros r+rwz1SetVariable.as_python_constant..s&BBBa''))BBBr-rxrPs r+rzSetVariable.as_python_constantsBB4>BBBBr-rr c|d|jD|tdt |jdS)Ncg|] }|j Sr]rrs r+r-z+SetVariable.reconstruct..666!666r- BUILD_SETr)foreachrsrr rrs r+rzSetVariable.reconstructsW66t~6667770#dnBUBUVVVWWWWWr-r/r"r\r2c R |d|g|Did|D}n`#t$rS}tt||t t t j|jYd}~nd}~wwxYwtj ||S)Nc6g|]}|Sr]rrs r+r-z0SetVariable._fast_set_method..'s$@@@Q!&&((@@@r-c>i|]\}}||Sr]rrs r+rmz0SetVariable._fast_set_method..(s*HHHA1a**,,HHHr-r1) r_r6rr(r<maprrr2rbuild)rJr/r\r2r`resexcs r+_fast_set_methodzSetVariable._fast_set_methods "@@4-$-@@@HHHHHCC    $S 2D-=-Dch)O)O$P$P           $R---s14 BA B  Brc ddlm}|dvra|||rU|tur:|}||t ||||S|dkrt jtj|g|Ri|}|j j ||j |j |j tjdS|dkrl|st#|dkr2t%||dt#|dt#|d d }|d t&g} nU|d kr|s|r2t%||d t#|dt#|d  |jj}nX#t0$rK} t3t0|t5t7tj| jYd} ~ nd} ~ wwxYwt;|||g||S|dkr|st#|dkr2t%||dt#|dt#|d t jt@j!"|||d giS|dkrZ|r"t%||dt#|d t jt@j#"||g|iS|dkrZ|r"t%||dt#|d t jt@j$"||g|iS|dkrZ|r"t%||dt#|d t jt@j%"||g|iS|dkrZ|r"t%||dt#|d t jt@j&"||g|iS|dkrZ|r"t%||dt#|d t jt@j'"||g|iS|dkr}|st#|dkr2t%||dt#|dt#|d t jt@j("||g|iS|dkr}|st#|dkr2t%||dt#|dt#|d t jt@j)"||g|iS|dkrn|*rZ|r"t%||dt#|d t jt@j+"||g|iS|dkr|st#|dkr2t%||dt#|dt#|d |d |vrt3t0||t;|d ||S|dkr|st#|dkr2t%||dt#|dt#|d |d |vr$t;|d ||StjdS|dvrt#|dkr"t%||dt#|dtXj-tXj.d} |d /} ta| t&s/t jt"|| gi} t j| 1|"||| giS|d vrddddd 1|} te|d t&t j3fs`tjd!|d"|4d#|d 4d$} t3tj|| g| J||| ||S|d%vrte|d t&t j3fs`tjd!|d"|4d#|d 4d$} t3tj|| gddddd%1|} | J||| |||S|d&kr}te|d t&t j3fstjd'S||d||}tjt#|jd kS|tlvrxte|d t&t j3fstjtnStjtm||j|d jSt;||||S)(Nr)check_constant_args) isdisjointunion intersection differencesymmetric_differencerKrrz1 args and 0 kwargsrrrrrnrr1rrz0 kwargsintersection_updaterrzExpect: 0 kwargs, Actual: difference_updatersymmetric_difference_updaterremovediscard)r)issubset issupersetrr)__and__r __xor____sub__z unsupported operand type(s) for z: 'z' and 'r )__iand__r__ixor____isub__rVF)8utilsrr5r]rgetattrr rcall_setrrrr_rrrrrrrhrkrsrnrGrrr<rr2rorr#r set_isdisjointrset_intersectionset_intersection_update set_unionset_differenceset_difference_updateset_symmetric_differenceset_symmetric_difference_updater set_updateoperatorleger8rr*r=UserDefinedSetVariablepython_type_namer7rNotImplemented)rJr/rr2r`rpy_type temp_set_vtresulteoprUmrrrus r+rzSetVariable.call_method1s  0/////   $#D&11   ""c))&&((G((WWd-C-CT6RR R :  A)3C88A"VtVVVvVVK I " + +D 1 1 1 J      J  k/ 0 0 0#*400 0 U]] Ta#)4yy@@CKK@@@  !DG[7799:DD U]]  #)4yy@@CKK@@@  *..*<*<*>*>*A   (btC0@0G,P,P'Q'Q  GG  D6(F ; ; ;M \ ! ! Ta#)4yy@@CKK@@@  1(mBtAw44 5^ # # S#BjS[[:Q:Q:QRRR1*mB  r22 3* * * S#BjS[[:Q:Q:QRRR11mB  r22 3W__ S#BjS[[:Q:Q:QRRR1)2EFFTTTMDM2 \ ! ! #O3v;;OOO1(mB  r22 3( ( ( S#BjS[[:Q:Q:QRRR1/mB  r22 3+ + + Ta#)4yy@@CKK@@@  12mB  r22 32 2 2 Ta#)4yy@@CKK@@@  19mB  r22 3X  $//"3"3  S#BjS[[:Q:Q:QRRR1)2FGGUUTMDM2 X   Ta#)4yy@@CKK@@@  Awd""(2DAAAA77&&r5$?? ? Y   Ta#)4yy@@CKK@@@  Aw$ww**2udFCCC'.T:::: / / /4yyA~~#Bh3t998K8K8KLLL%K&kBGOO%%E%-- V!1#66DDR%RTUU,RVVD\\::HHT5M2 @ @ @)!1'  c$ii  d1g Y5U'VWW D&-}t}}@U@U@W@W}}`def`g`x`x`z`z}}})BcUCCCC===##B488 8 D D Dd1g Y5U'VWW D&-}t}}@U@U@W@W}}`def`g`x`x`z`z}}})BcUCCCC1#9/  c$ii  ===   RD& 1 1 1K X  d1g Y5U'VWW 6'.u555  %;T6JJA#*3q{+;+;q+@AA A + + +d1g Y5U'VWW ?'.~>>>#*&t,T^T!W=NOO ww""2tT6:::s5G H)AH$$H)r.c td)NzIllegal to getitem on a set) RuntimeError)rJr/r.s r+rzSetVariable.getitem_consts8999r-cdSrIr]rPs r+rz)SetVariable.install_dict_keys_match_guard r-r8r9)rWrXrYrZr SET_CONTAINSrr<rrrKrrpropertyr]rsrfrkrr(r5rrrzrrrrr=r>s@r+rhrhsJ88!.N*O$**  ******OCOOOO &3CD&&&X&-O---\-9#9999TCCCCCCXXXX. #. .?# . S/)* .  ....&P; #P;P;?# P; S/)* P;  P;P;P;P;P;P;d:):0?: ::::         r-rhc eZdZdefdZededfdZdefdZ de fdZ dd Z d d d ede edeeefdef fd ZdZdZdZxZS)FrozensetVariabler%ch|jsdSddd|jDzdzS)Nz frozenset()rrmc3HK|]}|jVdSrIrros r+rhz/FrozensetVariable.debug_repr..rpr-rrqrPs r+rzFrozensetVariable.debug_reprs?z O =!H!HTZ!H!H!HHHH3N Nr-rFc4|jSrI)r_rrPs r+rszFrozensetVariable.set_itemssz   r-ctSrI) frozensetrPs r+r5zFrozensetVariable.python_typer-c>td|jDS)Nc@h|]}|jSr]rros r+rwz7FrozensetVariable.as_python_constant..s&LLL!$1133LLLr-)rrsrPs r+rz$FrozensetVariable.as_python_constants!LLT^LLLMMMr-rr Ncd|jDfdt dddS)Ncg|] }|j Sr]rrs r+r-z1FrozensetVariable.reconstruct..r~r-cVdgS)Nr)rcreate_load_globalrsr+rz/FrozensetVariable.reconstruct..!s-G))..{;;r-rF)rrsrrr rs `r+rzFrozensetVariable.reconstructsy66t~666777        21e<<=====r-r/r"rr2r`c&|dvrtd|d|dkrtjdS|dvr8t||||}t |jSt||||S)NrrnrrrrIllegal call_method z on a frozensetrK)rsrrr)rrrrorrr_)rJr/rr2r`rrus r+rzFrozensetVariable.call_method)s I I IKdKKKLL L Z  $*400 0    ##BdF;;A$QW-- -ww""2tT6:::r-cdS)zB Frozensets are immutable and hashable in Python. Tr]rPs r+rAz$FrozensetVariable.is_python_hashableEs tr-cDt|SrI)rMrrPs r+rOz!FrozensetVariable.get_python_hashKsD++--...r-cV||kSrIrrTs r+rSz!FrozensetVariable.is_python_equalNs%&&((E,D,D,F,FFFr-r8)rWrXrYrrrr]rsr(r5rrrr<rrzrrArOrSr=r>s@r+rr sLOCOOOO !3CD!!!X!TNCNNNN > > > >; #;;?# ; S/)* ;  ;;;;;;8 ///GGGGGGGr-rc eZdZdefdZddZdddeeddfdZe de fd Z de fd Z de fd Zddd edeed eeefdef fd ZxZS)DictKeySetVariabler%ch|jsdSddd|jDzdzS)Nz dict_keys([])z dict_keys([rmc3HK|]}|jVdSrIrros r+rhz0DictKeySetVariable.debug_repr..Xs.(O(Oq):):(O(O(O(O(O(Or-z])rqrPs r+rzDictKeySetVariable.debug_reprSsCz "?(O(ODJ(O(O(O O OORVV r-NcdSrIr]rPs r+rz0DictKeySetVariable.install_dict_keys_match_guard[rr-r/r"r2cdSrIr]rRs r+rz.DictKeySetVariable.install_dict_contains_guard_s  r-c|jSrIrrPs r+rszDictKeySetVariable.set_itemses zr-ctSrIrrPs r+r5zDictKeySetVariable.python_typeirr-cztd|jDdS)Nc@h|]}|jSr]rros r+rwz8DictKeySetVariable.as_python_constant..ns& ? ? ?1QT $ $ & & ? ? ?r-)rzrjrsrrPs r+rz%DictKeySetVariable.as_python_constantls6}} ? ? ? ? ?  $&& r-rr`cz|dvrtd|dt||||S)Nrrz on a dict_keys)rrorrJr/rr2r`rus r+rzDictKeySetVariable.call_methodqsI I I IKdKKKLL Lww""2tT6:::r-r9)rWrXrYrrrr<rrrrrsr(r5rrzrr=r>s@r+rrRs9C     ) 15o1F     3XTC ; # ; ;?# ; S/)* ;  ; ; ; ; ; ; ; ; ; ;r-rc eZdZUdZdZeeed<dede ddffd Z e de fdZ e de efd Zd d de efd ZddZd d dedefdZd d dede edeeefdef fd ZxZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method NkvrAr`r%c tjdi||jdvsJt|tsJ||_dS)N)rrr_r])rorKrr=rCrArCs r+rKzDictViewVariable.__init__sV""6"""w55555'#455555 r-cb|jJt|jj|jSrI)rrrAr_rPs r+ view_itemszDictViewVariable.view_itemss/w"""3wt|)4733555r-ctrI)NotImplementedErrorrPs r+ view_items_vtzDictViewVariable.view_items_vts "!r-r/r"c|jSrI)rr.s r+r!z$DictViewVariable.unpack_var_sequences !!r-rr c|jJ||j||j|ddS)Nr)rrA load_methodrrs r+rzDictViewVariable.reconstructsRw""" DG$$$Ar-rc|jJ||jvrtjdStjdS)NTF)rr5r)rr)rJr/rs r+r3z!DictViewVariable.call_obj_hasattrsNw""" 4##%%. . .#*400 0&u---r-r2c|dkr|j||||S|dkr"t|jt St ||||S)Nrrr)rArrrrrors r+rzDictViewVariable.call_methods} 9  <++BdFCC C Z  '"2B2D2D ww""2tT6:::r-r8)rWrXrYrZrrr__annotations__rCrrKrrr<rrr!rrr3rzrr=r>s@r+rr}s B  1ST 6C666X6"tO4"""X" "&="$BW"""" .).14. .... ; # ; ;?# ; S/)* ;  ; ; ; ; ; ; ; ; ; ;r-rc eZdZdZedeefdZedeefdZ de fdZ ddde d eed e e efdef fd ZxZS) rrr%c*t|jSrI)r]rrPs r+rszDictKeysVariable.set_itemss4?###r-c$d|jDS)Ncg|] }|j Sr]rrs r+r-z2DictKeysVariable.view_items_vt..s......r-rrPs r+rzDictKeysVariable.view_items_vts/.do....r-ctSrIrrPs r+r5zDictKeysVariable.python_typerr-r/r"rr2r`c|dkr|j||||S|dvr:t|j|}||dj}t |S|t vrst |dttfstj tStj t ||j|djSt||||S)Nr)rrr rrrrrr) rArrrsrhrr=rrrrro)rJr/rr2r`rrrus r+rzDictKeysVariable.call_methods > ! !<++BdFCC C     --A$q'#$$Aq>> ! ) ) )d1g 5E'FGG ?'.~>>>#*&t,T^T!W=NOO ww""2tT6:::r-)rWrXrYrrr]rrsr<rr(r5rrzrr=r>s@r+rrs B $3/$$$X$/tO4///X/T; #;;?# ; S/)* ;  ;;;;;;;;;;r-rcFeZdZdZedeefdZdefdZ dS)rrr%c*t|jSrI)r<rrPs r+rz DictValuesVariable.view_items_vtsDO$$$r-ctSrI)rrPs r+r5zDictValuesVariable.python_typesr-N) rWrXrYrrr<rrr(r5r]r-r+rrs^ B %tO4%%%X%Tr-rc eZdZdZedeefdZdefdZ ddde deed e e efdef fd Z d Z xZS) rr_r%c$d|jDS)NcJg|] \}}tj|j|g!Sr])r rrGrs r+r-z3DictItemsVariable.view_items_vt..s-OOOtq! 'q 22OOOr-rrPs r+rzDictItemsVariable.view_items_vtsPOtOOOOr-ctSrI)rrPs r+r5zDictItemsVariable.python_typesr-r/r"rr2r`cr|dkrt|dkr"t||dt|dt|dtr)|j|d|djgiSt jdSt||||S)NrVrrrrF) rrr=rrArrrrors r+rzDictItemsVariable.call_methods 8  4yyA~~#Bh3t998K8K8KLLL$q'#455 U|//HtAw>OQSTTT#*511 1ww""2tT6:::r-cdS)zC Dictionary item views are not hashable in Python. Fr]rPs r+rAz$DictItemsVariable.is_python_hashabler7r-)rWrXrYrrr<rrr(r5rrzrrAr=r>s@r+rrs B PtO4PPPXPT; #;;?# ; S/)* ;  ;;;;;;"r-rrI)@rZr{rrrEcollections.abcrtypingrrrrrr r r bytecode_transformationr r rrrguardsrrrrrrrrrrrrrrrxrrconstantrlistsrtorch._dynamo.codegenr r3r"r^r#r\r,r:r@rCr@rOrTrhrrrrrrr]r-r+rs_( $$$$$$6666666666666666666666NNNNNNNN9999999900000000========43333333&&&&&&''''''0//////DDDDDD//////>C>D>>>> CG &'>? . "? "t " " " "^ ^ ^ ^ ^ ^ ^ ^ BJ2J2J2J2J2?J2J2J2Z     1   P>P>P>P>P>+P>P>P>lW W W W W #W W W tBGBGBGBGBG BGBGBGJ(;(;(;(;(;(;(;(;V8;8;8;8;8;8;8;8;v,;,;,;,;,;',;,;,;^     )        (     r-