o j9:j-@sdZddlZddlZddlZddlmZmZmZddlm 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 dlGddde/Z?Gddde>Z@Gddde>ZAGddde/ZBGdd d eBZCGd!d"d"eBZDGd#d$d$eBZEe3eFBZGGd%d&d&ejHjIeGe/fZJGd'd(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)UserFunctionVariablecseZdZeZejZejZ dhe j Z efdee e fde de ddffdd Zde de fdd Zdee e ffd d Zdefd d Zdee e ffddZdee e ffddZde fddZde defddZddddde dee deee fde f ddZddddde dee deee fd ee d!fde fd"d#Zdefd$d%Zdefd&d'Zd(e dBd)e defd*d+ZdRd.d/Z dRd0d1Z!ddd2e de fd3d4Z"ddd2e de fd5d6Z#d2e de dBfd7d8Z$d2e ddfd9d:Z%dSd;d<Z&ddd=e'e ddfd>d?Z(ddd@e de fdAdBZ)dddCed=e'e deee fde f fdDdE Z*ddde'e fdFdGZ+ddde fdHdIZ,dddCede-fdJdKZ.de de ffdLdM Z/defdNdOZ0dddCeffdPdQ Z1Z2S)TConstDictVariableuser_clsitemskwargsreturnNc sd|vr |dd|vr|dtjd i|ttfdd|Ds,Jdttdfddffdd ||}|fd d |D|_|j rWt |j  nd |_ | |_ ||_dS)Noriginal_itemsshould_reconstruct_allc3s,|]\}}t|tfot|tVqdSN) isinstancer%.0xvHashabled/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/_dynamo/variables/dicts.py is  z-ConstDictVariable.__init__..keyr'r2cst|r|S|Sr5)r6r@r;r=r> make_hashableosz1ConstDictVariable.__init__..make_hashablecsi|] \}}||qSr=r=r7)rBr=r> usz.ConstDictVariable.__init__..Tr=)popsuper__init__r'allr0rr%_get_dict_cls_from_user_clssourcerr4copyr3r/)selfr0r/r1dict_cls __class__)r<rBr>rFWs(        zConstDictVariable.__init__csVttjtjf|vr|}n tfdd|jD}|vs"J||tjur)t}|S)Nc3s|] }|vr|VqdSr5r=)r8baseaccepted_dict_typesr=r>r?sz@ConstDictVariable._get_dict_cls_from_user_cls..)dict collections OrderedDict defaultdictnext__mro__)rKr/rLr=rPr>rH~s  z-ConstDictVariable._get_dict_cls_from_user_clscCdd|jDS)NcS i|] \}}|j|qSr=)vtas_proxyr8kr:r=r=r>rC z.ConstDictVariable.as_proxy..r0rKr=r=r>r[szConstDictVariable.as_proxycCs|g}|jD]-\}}t|jdrt|jjn|j}t|dr&t|jn|}||d|qdd|dS)Nvaluez: {, })r0hasattrrZreprra debug_reprappendjoinrKr0r]r:key_strval_strr=r=r>rgs "zConstDictVariable.debug_reprcCrX)NcSrYr=rZas_python_constantr\r=r=r>rCsz8ConstDictVariable.as_python_constant..r_r`r=r=r>rnsz$ConstDictVariable.as_python_constantcCs|dd|jDS)NcSsi|] \}}|j|qSr=rmr\r=r=r>rCsz=ConstDictVariable.keys_as_python_constant..install_dict_keys_match_guardr0r`r=r=r>keys_as_python_constantsz)ConstDictVariable.keys_as_python_constantcC|jSr5r/r`r=r=r> python_typezConstDictVariable.python_typerZcCsDt|tsJt}t|sdS||}||jvo!t|j|tj SNF)r6r%r'r(r0r DeletedVariable)rKrZr<r@r=r=r> __contains__s  zConstDictVariable.__contains__txr, tree_map_fnr-map_fnresttree_map_kwargsc sg}|D]%}|}t|trt|jt|jkr$||||||S||qt|j|jD]6\}} g} |D]"}z | |j|Wq=ty_||||||YSw| |||| ||<q5fddD} |j | ddt dS)Nci|]}|j|qSr=rZr8 key_trackernew_items_hashedr=r>rC z:ConstDictVariable.call_tree_map_branch..Tr0r3r4rI mutation_type) realizer6r.lenr0_tree_map_fallbackrhtypeKeyError call_tree_mapcloner$) rKryrzr{r|r} other_dicts candidaterrasibling_leavesupdated_original_itemsr=rr>call_tree_map_branchsP       z&ConstDictVariable.call_tree_map_branchkeypath.c sg}|D]&}|}t|trt|jt|jkr%|||||||S||qt|j|jD]D\} } g} |D]#}z | |j| Wq>tya|||||||YSw| j } |t | f} | |||| || | <q6fddD}|j |ddtdS)Ncr~r=rrrr=r>rCrzDConstDictVariable.call_tree_map_with_path_branch..Tr)rr6r.rr0_tree_map_with_path_fallbackrhrrrZrnr call_tree_map_with_pathrr$)rKryrzr{r|r}rrrrrar key_const child_keypathrr=rr>call_tree_map_with_path_branchsV         z0ConstDictVariable.call_tree_map_with_path_branchcCstdd|jDS)Ncss|] }t|tj VqdSr5)r6r rwr8r9r=r=r>r?s z(ConstDictVariable.len..)sumr0valuesr`r=r=r>rszConstDictVariable.lencs"jptfddjDS)Nc3s*|]\}}j|j|VqdSr5) is_new_itemr3getrZr8r@rar`r=r>r?!s  z2ConstDictVariable.has_new_items..)r4anyr0r`r=r`r> has_new_items szConstDictVariable.has_new_itemsraothercCs<|r|r|rt|t|kSt|t|kSr5) is_realizedidr)rKrarr=r=r>r&szConstDictVariable.is_new_itemcodegenr+cCshd}|jD]!\}}|j|j}|||s|jr(||j|||d7}q|td|ddS)Nrr BUILD_MAParg)r0r3rrZrr4 append_outputr)rKrnum_argsr@raitemr=r=r>reconstruct_kvs_into_new_dict.s z/ConstDictVariable.reconstruct_kvs_into_new_dictcs|jtjurSfdd|rDgtddt| t d| gt dt ddS| tdddS|rxt ddd tg||  t d dd dS| dS) NctdgS)NrT extend_outputcreate_load_python_modulerScreate_load_attrr=rr=r>> z/ConstDictVariable.reconstruct..rFupdater POP_TOPrr DICT_UPDATE)r/rSrT add_push_null_contains_self_referencerrr add_cacher load_methodrrrrKrr=rr> reconstruct:sF          zConstDictVariable.reconstructrcCs,t|}||jvrtt||gd|j|S)Nargs)r'r0rr)rKryrr@r=r=r>'getitem_const_raise_exception_if_absentis  z9ConstDictVariable.getitem_const_raise_exception_if_absentcCsJt|}||jvr d|jd}tdd|j|dgtjd|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'r0rarr USER_ERROR)rKryrr@msgr=r=r> getitem_constqs   zConstDictVariable.getitem_constcCs t|}||jvr dS|j|Sr5)r'r0)rKrr@r=r=r>maybe_getitem_consts  z%ConstDictVariable.maybe_getitem_constcCsV||vsJt|}t|j|}t|j|}t|tjr)| dSdSr5) r'tupler0keysindexr3r6r LazyVariableTrackerr)rKrr@roriginal_key_vtr=r=r>realize_key_vts   z ConstDictVariable.realize_key_vtcCs|jr t|tjdSdSr5)rIr make_guardrDICT_KEYS_MATCHr`r=r=r>rpsz/ConstDictVariable.install_dict_keys_match_guardrcCs|jsdS|jj|rdS|d|v}|djdur?|dr?|r(t|jnt|j}t| t j ||d ddS|djrU|rO| |ddS|dSdS)NrrA)rIoutput side_effects is_modifiedis_python_constantrCONTAINS_GUARDNOT_CONTAINS_GUARDrr functoolspartialrnrrp)rKryrcontainsguard_fnr=r=r>install_dict_contains_guards*    z-ConstDictVariable.install_dict_contains_guardr@cCs |||Sr5)rrKryr@r=r=r>mp_subscript_impls z#ConstDictVariable.mp_subscript_implnamec s ddlm}ddlm}t|dkr2|jtg|Ri|}jj ||j |j t dS|dkr^|s:|rKt|dt|dt|d||jrZjj|jt|S|d krt|rrt|d t|d ||jrjj|jt|S|d kr|s|rt|dt|dt|d||jrjj|j|s|rttt|S|d kr||s|rt|dt|dt|d|j|j tddS|dkr4|r4|ot|d}|st|d||s t|dkrt|dt|dt|djj ||d|j |d<t dS|dkrp|rp|oFt|d}|rg|d|_jj ||j |dt dSt!"|||S|dkrt|dvrt|dt|d |ot|d}|st|d|d|vr|#|t|dkrt dS|dS|$|dS|dkr!|r!t|dvrt|dt|d |ot|d}|st|d|d|vr |#|t|dkr tt%|dSd|_jj ||j &|dS|dkr|rt'|j(trDt'|j(t)j*sDt|rDt||j sPtt%dgd|j(t)j*urt|dkscd|vrt|dkrx|d+rx|d,} n|-d} r| +r| ,} nt||j j.| d\} } n|j .\} } d|_jj |t/0| j1| gS|dkr|s|rt|dt|dt|dd|_jj ||j 2t dS|dkrR|rR|t|dk} t|dk} | s| rIjj || r/t3|dt4r|d|d}n |t|d}|j |j | rDfd d!| D}|j |t dSt!"|||S|d"krt|smt|d#t|dt|d|out|d}|st|d|#||d|v}t56|S|d$kr|rt|dvrt|d%t|dt|d|ot|d}|st|d||st|dkrt|d&t|dt|d|7|d}|dur|St|dkrt d}n|d}jj |||j |d<|S|d'krr|jj ||d|vr3tt%d} t|dkrI|d+rI|d,} |r_d|vr_|d+r_|-d,} |d}|j j8|| dt dS|d(krt9|t4rt|dkrt|d)t|d | t:j;<||dgiS|d*krt56|"d(||j= S|d+kr?t|dkrt|d)t|d |d}t9|t4t/j>t/j?fr(t3|t/j>r|j@dusJt3|j@t4sJ|j@}|j(tur|j(}|}n|j(}|}||j|j td|d,}|d|j |dj |Sttd-|AjBd.|AjBd/gddS|d0krN|"d|||S|d1krudd2lCmD}|jrktE|jskjj|j||Ftd3St!"|||S)4Nr )DictBuiltinVariable)SourcelessBuilderrFr0z0 args and 0 kwargsz args and z kwargsrz0 args argsrrJ)r0rrI __setitem__rr z2 args and 0 kwargs __delitem__Tr)r r z 1 or 2 argsrDpopitemzpopitem(): dictionary is emptyrlast)rclearrcs"i|] \}}t||qSr=)r%buildr\r<ryr=r>rCsz1ConstDictVariable.call_method..rxzmore 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__)r0rrIr/z$unsupported operand type(s) for |: 'z'and ''__ior____iter__ListIteratorVariabler)Grbuilderrr'call_custom_dictrRrrmutationr0rr&createrrrprIguard_on_key_orderaddDictItemsVariableDictKeysVariabler TypeErrorDictValuesVariablerrJr$ is_mutabler(r)r4rrE call_methodrrrrD issubclassr/rSrTrrnrrr TupleVariablerZrr6r.r%rrrrr dict___eq__ call_functionraUserDefinedDictVariableDefaultDictVariable_base_vtrt__name__listsrrunpack_var_sequence)rKryrrr1rr temp_dict_vt arg_hashablerr:r]has_arg has_kwargsdict_vtkwargs_hashablerrar9r@rr/to_cpy new_dict_vtrrMrr>rs<                                                   zConstDictVariable.call_methodcCs|dd|jDS)NcSg|]}|jqSr=rrr=r=r> :z9ConstDictVariable.unpack_var_sequence..rorKryr=r=r>r 8sz%ConstDictVariable.unpack_var_sequencecCs|t|t|jS)z Mapping length for dict objects.)rpr%rrr0rr=r=r> mp_length<szConstDictVariable.mp_lengthcs|tfddttjtjfDr%tj|rtdSjtur%tdSdjd}t ddj|d gt j d dS) Nc3s|]}j|uVqdSr5rs)r8tr`r=r>r?Fs  z5ConstDictVariable.call_obj_hasattr..TFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr) rrRrSrTrUrer/r&rrr SUPPORTABLE)rKryrrr=r`r>call_obj_hasattrAs"        z"ConstDictVariable.call_obj_hasattrc s|tjdi|SNr=)rprEr)rKr1rMr=r>rZszConstDictVariable.clonecCdS)zP Dictionaries are mutable and therefore not hashable in Python. Fr=r`r=r=r>is_python_hashable^z$ConstDictVariable.is_python_hashablecs&|dkr t||St||S)NrN)r%rrtrE var_getattrrKryrrMr=r>r dszConstDictVariable.var_getattrrr+r2Nr2N)3r  __module__ __qualname__rR _cpython_typer DICT_CONTAINSrDICT_NOT_CONTAINSrr%_nonvar_fieldsrrrFrHr[strrgrnrqrtboolrxrrrrintrrrrrrrrrrplistrrrr rr&rrrr  __classcell__r=r=rMr>r.Ks '   2   6  /    *   r r.c seZdZejZdededdffdd Zde fddZ d d de e fd d Z dddZdddZd d de de fddZd d dede e deee fde f ddZd d de fddZd d dedeffdd ZZS) MappingProxyVariabledv_dictr1r2Nc s*tjdi|t|tsJ||_dSr)rErFr6r.r0rKr0r1rMr=r>rFos zMappingProxyVariable.__init__cCstjSr5)typesMappingProxyTyper`r=r=r>rttruz MappingProxyVariable.python_typeryr,cC |j|Sr5)r0r rr=r=r>r w z(MappingProxyVariable.unpack_var_sequencerr+csd|jrd|jd}tdd|j|dgtjdfdd|jtd d 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.rcr)Nr3)rrr2rr=rr=r>rrz2MappingProxyVariable.reconstruct..r F)rIrr rrr0rr)rKrrr=rr>rzs"    z MappingProxyVariable.reconstructcCs@|jr|jjrd}tdd|jd|ddgddSdSdS)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 mutationr6z, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rIrrhas_existing_dict_mutationr)rKryrr=r=r>_check_mutation_guards   z*MappingProxyVariable._check_mutation_guardr@cCs|||j||Sr5)r8r0rrr=r=r>rs z&MappingProxyVariable.mp_subscript_implrrcCs|||j||||Sr5)r8r0rrKryrrr1r=r=r>rs z MappingProxyVariable.call_methodcCr4r5)r0rrr=r=r>rr5zMappingProxyVariable.mp_lengthcs0|tjurt||tjjvSt||Sr5)rtr2r3r%r__dict__rErr!rMr=r>rsz%MappingProxyVariable.call_obj_hasattrr")ryr,r2N)r r$r%r2r3r&r.rrFrrtr-r%r rr8rr*rRrrr&rr.r=r=rMr>r/jsB       r/c@s0eZdZd ddZdddeeddfdd ZdS) NNModuleHooksDictVariabler2NcCdSr5r=r`r=r=r>rpz7NNModuleHooksDictVariable.install_dict_keys_match_guardryr,rcCr<r5r=rKryrr=r=r>rz5NNModuleHooksDictVariable.install_dict_contains_guardr#)r r$r%rpr-r%rr=r=r=r>r;s r;c seZdZejZ ddeeefdededBde ddf fdd Z de ffd d Z de ffd d Zed ede fddZdddedeffdd Zddde deedee efdef fdd Zddde deffdd Zd ddZZS)!rNr0r/default_factoryr1r2c s@tj||fi||tjusJ|durtd}||_dSr5)rErFrSrUr&rr@)rKr0r/r@r1rMr=r>rFs   zDefaultDictVariable.__init__cs$|jtttfvr |js dStSrv)r@r-rrRr0rErr`rMr=r>rs z&DefaultDictVariable.is_python_constantcs,|jdusJd|jdtdS)Nz defaultdict(rc))r@rgrEr`rMr=r>rgszDefaultDictVariable.debug_reprrcCs0t|tjtjjtjjfpt|tjo|jduSr5)r6r BaseBuiltinVariable functionsBaseUserFunctionVariablePolyfilledFunctionVariabler&rarr=r=r>is_supported_argsz$DefaultDictVariable.is_supported_argryr,r@cs\||vr |||S|jrtt||gddS|j|gi}t|d||gi|S)Nrr)rr@is_constant_nonerrrrEr)rKryr@ default_varrMr=r>rs  z%DefaultDictVariable.mp_subscript_implrrcs|dkrH|jrHt|dkrt||dt|dt|dtr?|djdkr?||dr?|jj ||d|_ t dSt ||||S|dkrnt|dkr^t||d t|dt|tj|||dgiSt ||||S) N __setattr__r z2 argsrrr@r rr)rrrrr&rarFrrrr@rrErr%rr rrr9rMr=r>rs&      zDefaultDictVariable.call_methodcs|dkr|jSt||S)Nr@)r@rEr r!rMr=r>r +szDefaultDictVariable.var_getattrrr+cs~fdd|jgtddt|td|gt dt ddS)Ncr)NrUrr=rr=r>r7rz1DefaultDictVariable.reconstruct..r Frr) rr@rrrrrrrrrrr=rr>r4s,      zDefaultDictVariable.reconstructr5r")r r$r%rSrUr&rRr%rrrFr+rr*rg staticmethodrFrr-rr rr.r=r=rMr>rsX     rc seZdZUdZdZedBed<dededdffdd Z e defd d Z e de e fd d Zd dde e fddZdddZd ddedefddZd ddede e deee fde f fdd Zd dde fddZZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method Nkvr0r1r2c s8tjdi||jdvsJt|tsJ||_dS)N)rrr0r=)rErFrLr6r.r0r1rMr=r>rF[s zDictViewVariable.__init__cCs |jdusJt|jj|jSr5)rLgetattrr0r0r`r=r=r> view_itemsaszDictViewVariable.view_itemscCstr5)NotImplementedErrorr`r=r=r> view_items_vtfrzDictViewVariable.view_items_vtryr,cCrrr5)rPrr=r=r>r lruz$DictViewVariable.unpack_var_sequencerr+cCs2|jdusJ||j||j|ddS)Nr)rLr0rrrr=r=r>ros  zDictViewVariable.reconstructrcCs0|jdusJ||jvrtdStdS)NTF)rLrtr:r&rr!r=r=r>rus  z!DictViewVariable.call_obj_hasattrrcsN|dkrddlm}||jtdS|dkrt||St||||S)Nrr rr__repr__) r rrPr$r%rrgrEr)rKryrrr1rrMr=r>r}s zDictViewVariable.call_methodcCst|t|jS)z&Sequence length for dict view objects.)r%rrrNrr=r=r> sq_lengthszDictViewVariable.sq_lengthr")r r$r%__doc__rLr*__annotations__r.rrFpropertyrNr-r%rPr rr&rrRrrRr.r=r=rMr>rKRs:    rKc seZdZeZdZedeefddZ ede efddZ de fddZ defd d Zd d d ede edeeefdef fdd ZZS)rrr2cC t|jSr5)setrNr`r=r=r> set_items zDictKeysVariable.set_itemscCdd|jDS)NcSrr=rrr=r=r>rrz2DictKeysVariable.view_items_vt..rNr`r=r=r>rPzDictKeysVariable.view_items_vtcCtSr5)rr`r=r=r>rtr=zDictKeysVariable.python_typecCsX|jsdSg}|jD]}t|jdrt|jjn|j}||q dd|dS)Nz dict_keys([])raz dict_keys([,])rNrerZrfrargrhri)rKr0r]rkr=r=r>rgs   zDictKeysVariable.debug_reprryr,rrr1cs|dkr |j||||S|dvr"t|j|}||dj}t|S|tvrGt|dttjt t fs8t |t St |t||j|djSt||||S)Nrx__and____iand__rr__sub____isub____xor____ixor__r)r0rrMrXr*rr6r UserDefinedSetVariablerrr%rNotImplementedrE)rKryrrr1mrrMr=r>rs*  zDictKeysVariable.call_method)r r$r%rr&rLrUrWr%rXr-rPrrtr*rgrRrr.r=r=rMr>rs(  rc@sFeZdZeZdZedeefddZ de fddZ de fddZ d S) rrr2cCrVr5)r-rNr`r=r=r>rPrYz DictValuesVariable.view_items_vtcCr]r5)rr`r=r=r>rtr=zDictValuesVariable.python_typecCsR|jsdSg}|jD]}t|drt|jn|}||q dd|dS)Nzdict_values([])raz dict_values([r^r_)rNrerfrargrhri)rKr0r:rlr=r=r>rgs  zDictValuesVariable.debug_reprN)r r$r%rr&rLrUr-r%rPrrtr*rgr=r=r=r>rsrc seZdZeZdZededfddZede e fddZ de fdd Z defd d Zd d dede e deee fde f fdd ZdedfddZZS)rr0r2r'cCrZ)NcSs$h|]\}}tt|j|gqSr=)r'r rrZr\r=r=r> sz.DictItemsVariable.set_items..r[r`r=r=r>rXszDictItemsVariable.set_itemscCrZ)NcSs g|] \}}t|j|gqSr=)r rrZr\r=r=r>rr^z3DictItemsVariable.view_items_vt..r[r`r=r=r>rPr\zDictItemsVariable.view_items_vtcCr]r5)rr`r=r=r>rtr=zDictItemsVariable.python_typecCs|jsdSg}|jD]/\}}t|jdrt|jjn|j}t|dr)t|jn|}|d|d|dq dd|dS) Nzdict_items([])ra(rcrAz dict_items([r^r_r`rjr=r=r>rgs zDictItemsVariable.debug_reprryr,rrr1cs@|dkrKt|dkrt||dt|dt|dtr*|j|d|djgiSt|dttjtt frFt |t|j |dj AdkSt dS|dkr]ddlm}||jtd S|d vrrt|j |}||dj }t|S|tvrt|dttjtt fst |tSt |t||j |dj St||||S) Nrr rrrFrrrra)rrr6rr0rr*r rhrr%rrXr&rr rrPr$rMrrirE)rKryrrr1rfn_hdlret_valrMr=r>rsT      zDictItemsVariable.call_methodFcCr)zC Dictionary item views are not hashable in Python. Fr=r`r=r=r>rVrz$DictItemsVariable.is_python_hashable)r r$r%rr&rLrUrWrXr-r%rPrrtr*rgrRrrrr.r=r=rMr>rs*  Arc @seZdZdZededeeeffddZ edddedeeeffdd Z d edddd fd d Z de defddZ deeeffddZde defddZde dedd fddZde dd fddZde defddZdefddZdeefddZd S) SideEffectsProxyDictzm A proxy dict that allows us to track mutations to the dict using side effects table as storage. rZr2cCst|tjriSt|tjrt|jS|}|t ur-t|tj r'| dSt |dStddt|dt|gtjddS)Nr:z-unsupported variable type for __dict__ accesszVariableTracker type: z.Dynamo does not know how to get __dict__ from r)rr NestedUserFunctionVariabler6LocalGeneratorFunctionVariablerpget_example_value_dictrZget_real_python_backed_valuer#UserDefinedObjectVariable_getattr_staticobject__getattribute__rrr DYNAMO_BUG)rZrar=r=r>rsfs"         z+SideEffectsProxyDict.get_example_value_dictryr,cs"t}fdd|DS)Nc s6i|]\}}|tj|jottjd|dqS)r:)rI)r%rrIrrrryrZr=r>rCsz;SideEffectsProxyDict.get_value___dict__..)rprsr0)ryrZexample_value_dictr=rzr>get_value___dict__s  z'SideEffectsProxyDict.get_value___dict__rNcCs"||_|jj|_||||_dSr5)rrrr| item_dict)rKrryr=r=r>rFs zSideEffectsProxyDict.__init__r@cCst}t||r |jS|Sr5)r'rrZrn)rKr@Hasherr=r=r>_maybe_unwrap_keysz&SideEffectsProxyDict._maybe_unwrap_keycCs|jj|jiSr5)rstore_attr_mutationsrrr`r=r=r>side_effects_tablez'SideEffectsProxyDict.side_effects_tablecCs8||}|j|j|r|jj|j|ddS|j|S)NT) deleted_ok)rrhas_pending_mutation_of_attrr load_attrr}rKr@rr=r=r> __getitem__s  z SideEffectsProxyDict.__getitem__racCs.||}t|ts J|j|j||dSr5)rrr*r store_attrr)rKr@rarr=r=r>rs z SideEffectsProxyDict.__setitem__cCs$||}|j|j|tdSr5)rrrrr rwrr=r=r>rs z SideEffectsProxyDict.__delitem__cCs6||}|}||vrt||tj S||jvSr5)rrr6r rwr})rKr@rtabler=r=r>rxs  z!SideEffectsProxyDict.__contains__cCstdd|DS)Ncss|]}dVqdS)r Nr=)r8_r=r=r>r?sz/SideEffectsProxyDict.__len__..)rr`r=r=r>__len__rzSideEffectsProxyDict.__len__ccsnt}|}|D]\}}t|tjrq |t|Vq |jD]\}}||vr4|t|Vq$dSr5) r'rr0r6r rwr&rr})rKr~dr]r:r=r=r>rs zSideEffectsProxyDict.__iter__)r r$r%rSrJr%rRr*rwrsr|rFkVrrrrrr+rxr,rrr'rr=r=r=r>rp`s*   rpcseZdZdZedddeddfddZdedddedd ffd d Zd e d edd fddZ d e defddZ d e de fddZ d e degefdefddZdddZdddeedd fddZZS)DunderDictVariablezrepresents object.__dict__ryr,rZr2cCs4|jrtnt}|jot|jd}|||||dS)Nr:)ryrrI)rIr!r"r)clsryrZrrIr=r=r>rszDunderDictVariable.creater1Nc s$tjifi|t|||_dSr5)rErFrpr0)rKrZryr1rMr=r>rFszDunderDictVariable.__init__rracCs||j|<dSr5r_)rKrrar=r=r>setitemszDunderDictVariable.setitemcCs |j|Sr5r_rKrr=r=r>getitem zDunderDictVariable.getitemcCs ||jvSr5r_rr=r=r>rrzDunderDictVariable.containsdefaultcCs(||r ||S|}||j|<|Sr5)rrr0)rKrrrar=r=r>getitem_or_defaults   z%DunderDictVariable.getitem_or_defaultcCr<r5r=r`r=r=r>rpr=z0DunderDictVariable.install_dict_keys_match_guardrcCr<r5r=r>r=r=r>rr?z.DunderDictVariable.install_dict_contains_guardr#)r r$r%rS classmethodr%rrrFr*rrr+rrrrpr-rr.r=r=rMr>rsJ    r)LrSrSrr2collections.abcrrrtypingrrrrtorch.utils._pytreer rr r r bytecode_transformationrrrrexcrrguardsrrrIrrrrutilsrrrrrrrOr!r"r#r$r%constantr&hashabler'r(r)setsr*torch._dynamo.codegenr+torch._dynamo.symbolic_convertr,rCr-r.r/r;rrKrrrr*rabcMutableMappingrprr=r=r=r>sL       %` }AGjf