o eiy@sJdZddlZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZmZddlmZmZmZmZdd lmZmZdd lmZmZdd lm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*d dl+m,Z,m-Z-d dl.m/Z/m0Z0e rddl1m2Z2ddl3m4Z4ddl5m6Z6d dl7m8Z8de de9fddZ: d:de-de dddfddZ;de-de9fddZGd$d%d%e=Z?Gd&d'd'e=Z@Gd(d)d)e=ZAGd*d+d+e-ZBGd,d-d-eAZCGd.d/d/eAZDGd0d1d1eAZEGd2d3d3e-ZFGd4d5d5eFZGGd6d7d7eFZHGd8d9d9eFZIdS);a 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)IterableSequence)AnyLiteralOptional TYPE_CHECKINGUnion OrderedSet)graph_break_hints polyfills variables)create_call_functioncreate_call_methodcreate_dup_topcreate_instruction)raise_observed_exception unimplemented) GuardBuilder install_guard) AttrSourceis_constant_sourceis_from_local_source)cmp_name_to_op_mapping dict_items dict_keys dict_valuesistyperaise_args_mismatchspecialize_symnode)ValueMutationNewVariableTracker)CONSTANT_VARIABLE_NONEConstantVariable) PyCodegenInstructionTranslator)BuiltinVariable)UserFunctionVariableobjreturncCst|jddS)N__instancecheck__F)type__dict__get)r+r1g/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/_dynamo/variables/dicts.pywas_instancecheck_overrideBsr3argtxr(c Csddlm}|durddlm}|}z|}Wn ty't|}Ynwtt || |d|dt| gddS)Nr!SourcelessBuilderrr'zunhashable type: z and variable tracker = args) builderr7torch._dynamo.symbolic_convertr( current_tx python_type Exceptionr.r TypeErrorcreaterealize)r4r5r7r(arg_typer1r1r2raise_unhashableFs$      rCxcCs(t|tjr|s|rdS|S)NT) isinstancerLazyVariableTracker is_realized is_hashableis_python_hashable)rDr1r1r2rH`s rHc sJeZdZejZdhejZGdddZe fde eefde de ddffdd Z de de fd d Z de e e ffd d ZdefddZde e e ffddZde e effddZde fddZdedefddZdddddedeede eefdef d d!Zdefd"d#Zdefd$d%Zd&eed'edefd(d)ZdId,d-ZdId.d/Zddd0edefd1d2Zddd0edefd3d4Z d0edeefd5d6Z!d0eddfd7d8Z"dJd9d:Z#ddd;e$eddfded;e$ede eefdef fd?d@ Z&ddde$efdAdBZ'ddd>ede(fdCdDZ)de deffdEdF Z*defdGdHZ+Z,S)KConstDictVariableuser_clsc@sBeZdZdZdeddfddZdefddZd ede fd d Z 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,NcCs"t|}t|s t|||_dSN)r rHrCrM)selfrMr1r1r2__init__}s z+ConstDictVariable._HashableTracker.__init__cCs:t|jtjr|js|jrt|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 ) rErMrrFrGrHhashoriginal_valueget_python_hashrOr1r1r2__hash__s  z+ConstDictVariable._HashableTracker.__hash__othercCs.t|tjsdS|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 FT)rErJ_HashableTrackerrMis_python_equalrOrVr1r1r2__eq__s  z)ConstDictVariable._HashableTracker.__eq__) __name__ __module__ __qualname____doc__r#rPintrUobjectboolrZr1r1r1r2rWvs  rWitemskwargsr,Nc sd|vr |dd|vr|dtjd i|tjtfdd|Ds-Jdttdfddffdd | |}|fd d |D|_|j rXt |j  nd |_ | |_||_dS)Noriginal_itemsshould_reconstruct_allc3s,|]\}}t|tfot|tVqdSrN)rEr#.0rDvHashabler1r2 s  z-ConstDictVariable.__init__..keyrLr,cst|r|S|SrN)rE)rlrir1r2 make_hashablesz1ConstDictVariable.__init__..make_hashablecsi|] \}}||qSr1r1rf)rmr1r2 z.ConstDictVariable.__init__..Tr1)popsuperrPrJrWallrbrr#_get_dict_cls_from_user_clssourcerrecopyrdrK)rOrbrKrcdict_cls __class__)rjrmr2rPs(        zConstDictVariable.__init__csVttjtjf|vr|}n tfdd|jD}|vs"J||tjur)t}|S)Nc3s|] }|vr|VqdSrNr1)rgbaseaccepted_dict_typesr1r2rksz@ConstDictVariable._get_dict_cls_from_user_cls..)dict collections OrderedDict defaultdictnext__mro__)rOrKrvr1rzr2rss  z-ConstDictVariable._get_dict_cls_from_user_clscCdd|jDS)NcS i|] \}}|j|qSr1rMas_proxyrgkrhr1r1r2rn z.ConstDictVariable.as_proxy..rbrTr1r1r2rzConstDictVariable.as_proxycCs|g}|jD]-\}}t|jdrt|jjn|j}t|dr&t|jn|}||d|qdd|dS)Nvaluez: {, }rbhasattrrMreprr debug_reprappendjoinrOrbrrhkey_strval_strr1r1r2rs "zConstDictVariable.debug_reprcCr)NcSrr1rMas_python_constantrr1r1r2rnz8ConstDictVariable.as_python_constant..rrTr1r1r2rsz$ConstDictVariable.as_python_constantcCs|dd|jDS)NcSsi|] \}}|j|qSr1rrr1r1r2rnsz=ConstDictVariable.keys_as_python_constant..install_dict_keys_match_guardrbrTr1r1r2keys_as_python_constantsz)ConstDictVariable.keys_as_python_constantcC|jSrNrKrTr1r1r2r=zConstDictVariable.python_typerMcCsBt|tsJtj}|o |||jvo t|j||tj SrN)rEr#rJrWrIrbrDeletedVariable)rOrMrjr1r1r2 __contains__s zConstDictVariable.__contains__r5r( 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)Ncsi|]}|j|qSr1rM)rg key_trackernew_items_hashedr1r2rn,s z:ConstDictVariable.call_tree_map_branch..T)rbrdrert mutation_type) rArErJlenrb_tree_map_fallbackrr.KeyError call_tree_mapcloner") rOr5rrrr other_dicts candidaterrsibling_leavesupdated_original_itemsr1rr2call_tree_map_branchsP       z&ConstDictVariable.call_tree_map_branchcCstdd|jDS)Ncss|] }t|tj VqdSrN)rErrrgrDr1r1r2rk:s z(ConstDictVariable.len..)sumrbvaluesrTr1r1r2r9szConstDictVariable.lencs"jptfddjDS)Nc3s*|]\}}j|j|VqdSrN) is_new_itemrdr0rM)rgrlrrTr1r2rk?s  z2ConstDictVariable.has_new_items..)reanyrbrTr1rTr2 has_new_items>szConstDictVariable.has_new_itemsrrVcCs<|r|r|rt|t|kSt|t|kSrN)rGidrA)rOrrVr1r1r2rDszConstDictVariable.is_new_itemcodegenr&cCshd}|jD]!\}}|j|j}|||s|jr(||j|||d7}q|td|ddS)Nrr! BUILD_MAPr4)rbrdr0rMrre append_outputr)rOrnum_argsrlritemr1r1r2reconstruct_kvs_into_new_dictLs z/ConstDictVariable.reconstruct_kvs_into_new_dictcs|jtjurSfdd|rDgtddt| t d| gt dt ddS| tdddS|rxt ddd tg||  t d dd dS| dS) NctdgS)Nr~ extend_outputcreate_load_python_moduler}create_load_attrr1rr1r2\ z/ConstDictVariable.reconstruct..rFupdater!POP_TOPrr DICT_UPDATE)rKr}r~ add_push_null_contains_self_referencerrr add_cacher load_methodrrrrOrr1rr2 reconstructXsF          zConstDictVariable.reconstructr4cCsrt|}||jvr4zdt|d|}Wnty+tdt|}Ynwtt ||gd|j|S)NzDict key lookup failed for z%. Debug representation of the key is r8) rJrWrbstrrr>r%r@rr)rOr5r4rl error_messager1r1r2'getitem_const_raise_exception_if_absents      z9ConstDictVariable.getitem_const_raise_exception_if_absentcCsLt|}||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)rJrWrbrrr USER_ERROR)rOr5r4rlmsgr1r1r2 getitem_consts    zConstDictVariable.getitem_constcCs"t|}||jvr dS|j|SrN)rJrWrb)rOr4rlr1r1r2maybe_getitem_consts   z%ConstDictVariable.maybe_getitem_constcCsX||vsJt|}t|j|}t|j|}t|tj r*| dSdSrN) rJrWtuplerbkeysindexrdrErrFrA)rOr4rlroriginal_key_vtr1r1r2realize_key_vts    z ConstDictVariable.realize_key_vtcCs|jr t|tjdSdSrN)rtr make_guardrDICT_KEYS_MATCHrTr1r1r2rsz/ConstDictVariable.install_dict_keys_match_guardr9cCs|jsdS|jj|rdS|d|v}|djdur8|dr8t|tjt |j |d | ddS|djrN|rH| |ddS| dSdS)Nr)rlinvert)rtoutput side_effects is_modifiedis_python_constantrr functoolspartialr.CONTAINS_GUARDrrr)rOr5r9containsr1r1r2install_dict_contains_guards(    z-ConstDictVariable.install_dict_contains_guardnamecsN ddlm}mddlm}tj|dkr5||tj |g|Ri|}|j j ||j |j tS|dkrSt|dkrKt||dt|d|||dS|d kr|s[|rlt||d t|d t|d ||jr{|j j|jt|S|d krt|rt||dt|d||jr|j j|jt|S|dkr|s|rt||d t|d t|d ||jr|j j|j|s|rtt|t|S|dkr||s|rt||d t|d t|d |j|j tddS|dkr-|s|r!t||d t|d t|d |t|j S|dkr}|r}|o?t |d}|sJt!|d|||sXt|dkrit||dt|d t|d |j j ||d|j |d<tS|dkr|r|ot |d}|r|d|_"|j j ||j #|dtSt$%||||S|dkrt|dvrt||dt|d|ot |d}|st!|d||d|vr|&||t|dkrtS|dS|'||dS|dkrd|rdt|dvr t||dt|d|o(t |d}|s3t!|d||d|vrP|&||t|dkrLtt(||dSd|_"|j j ||j )|dS|dkr|rt*|j+trt*|j+t,j-st|rt|||j sd} tt(|| gd|j+t,j-urt|dksd|vrt|dkr|d.r|d/} n|0d} r| .r| /} nt|||j j1| d \} } n|j 1\} } d|_"|j j |t23| j4| gS|d!kr'|s|rt||d t|d t|d d|_"|j j ||j 5tS|d"kr|r|t|dk} t|dk}| sG|r|j j || rtt6|dtrd|d|d}n |7|t|d}|j |j |rfd#d$| D}|j |tSt$%||||S|d%krt|st||d&t|d t|d |ot |d}|st!|d||&|||d|v}|S|d'krU|rUt|dvrt||d(t|d t|d |ot |d}|st!|d|||st|dkr't||d)t|d t|d |8|d}|dur5|St|dkr?t}n|d}|j j |||j |d<|S|d*kr||j j ||d|vrqtt(|d} t|dkr|d.r|d/} |rd|vr|d.r|0d/} |d}|j j9|| d tS|d+krt:|trt|dkrt||dt|d||t;j<=|||dgiS|d,kr|%|d+||j> S|d-krht|dkrt||dt|d|d}t:|tt2j?t2j@frO|j+tur |j+}|}n t6|ts(J|j+}|}||j|j td|d.}|d|j |dj |Sd/|AjBd0|AjBd1}tt||gddS|d2krw|%|d"|||S|d3krdd4lCmD}|jrtE|js|j j|j||F|td5St$%||||S)6Nr!)r)r%r6rP __getitem__1 args argsrrb0 args and 0 kwargs args and  kwargsrz0 argsrru)rbrrt__len__ __setitem__r z2 args and 0 kwargs __delitem__Tr0)r!r z 1 or 2 argsrppopitemzpopitem(): dictionary is emptyr8last)rclearrcs i|] \}}||qSr1)r@rr%rjr1r2rnrz1ConstDictVariable.call_method..rzmore than 1 args and 0 kwargs setdefaultz1 or 2 args and 0 kwargszat most 2 args and 0 kwargs move_to_endrZ__ne____or__)rbrrtrKz$unsupported operand type(s) for |: 'z'and ''__ior____iter__ListIteratorVariabler)Gr)r%r:r7rJrWr@r| call_dictrrmutationrbrr$rrrrrtguard_on_key_orderaddDictItemsVariableDictKeysVariablerr?DictValuesVariablerrur" is_mutablerHrCrerrq call_methodrrrrp issubclassrKr}r~rrr0rr TupleVariablerMrrEcall_custom_dictrrrr dict___eq__ call_functionrUserDefinedDictVariableDefaultDictVariabler=r[listsrrunpack_var_sequence)rOr5rr9rcr)r7 temp_dict_vt arg_hashablerrrhrhas_arg has_kwargsdict_vtkwargs_hashablerrrDrlrVrKto_cpy new_dict_vterr_msgrrwrr2rs>                                              zConstDictVariable.call_methodcCs|dd|jDS)NcSg|]}|jqSr1rrr1r1r2 Tz9ConstDictVariable.unpack_var_sequence..rrOr5r1r1r2rRsz%ConstDictVariable.unpack_var_sequencecs|tfddttjtjfDr%tj|rtdSjtur%tdSdjd}t ddj|d gt j d dS) Nc3s|]}j|uVqdSrNr)rgtrTr1r2rk[s  z5ConstDictVariable.call_obj_hasattr..TFz hasattr on z is not supportedzunsupported hasattr operationzClass z+Consider using a regular dictionary insteadr) rr|r}r~rrrKr%r@rr SUPPORTABLE)rOr5rrr1rTr2call_obj_hasattrVs"        z"ConstDictVariable.call_obj_hasattrc s|tjdi|SNr1)rrqrrOrcrwr1r2roszConstDictVariable.clonecCdS)zP Dictionaries are mutable and therefore not hashable in Python. Fr1rTr1r1r2rIsz$ConstDictVariable.is_python_hashablerr&r,Nr,N)-r[r\r]r DICT_CONTAINSrr#_nonvar_fieldsrWr|r.rrPrsrrrrrr=rarrrr_rrrrrrrrrrrlistrrrr%r,rrI __classcell__r1r1rwr2rJns9 '   2  /    ( p rJc seZdZdededdffdd ZdefddZd d dee fd d Z dddZ d d de dee de e e fde f ddZd d de deffdd ZZS)MappingProxyVariabledv_dictrcr,Nc s*tjdi|t|tsJ||_dSr-)rqrPrErJr8rOr8rcrwr1r2rP|s zMappingProxyVariable.__init__cCstjSrN)typesMappingProxyTyperTr1r1r2r=rz MappingProxyVariable.python_typer5r(cCs |j|SrN)r8rr)r1r1r2r 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)Nr;)rrr:rr1rr1r2rrz2MappingProxyVariable.reconstruct..r!F)rtrr r+rr8rr)rOrrr1rr2rs"    z MappingProxyVariable.reconstructrr9cCsF|jr|jjrd}tdd|jd|ddgd|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 mutationr=z, Dict mutation detectedzNAvoid modifying dictionaries that might be referenced by mapping proxy objectszROr avoid using the mapping proxy objects after modifying its underlying dictionaryr)rtrrhas_existing_dict_mutationrr8r)rOr5rr9rcrr1r1r2rs   z MappingProxyVariable.call_methodcs.|tjurt|tjjvSt||SrN)r=r:r;r%r@r/rqr,rOr5rrwr1r2r,sz%MappingProxyVariable.call_obj_hasattrr1)r[r\r]rJrrPr.r=r5r#rrrr|rr%r,r6r1r1rwr2r7zs.   r7c@s0eZdZd ddZdddeeddfdd ZdS) NNModuleHooksDictVariabler,NcCdSrNr1rTr1r1r2rz7NNModuleHooksDictVariable.install_dict_keys_match_guardr5r(r9cCrArNr1rOr5r9r1r1r2rz5NNModuleHooksDictVariable.install_dict_contains_guardr2)r[r\r]rr5r#rr1r1r1r2r@s r@c seZdZ ddeeefdedeededdf fdd Zde ffd d Z de ffd d Z e d ede fddZddde deedee efdef fdd Zddde deffdd ZdddZZS)rNrbrKdefault_factoryrcr,c s:tj||fi||tjusJ|durt}||_dSrN)rqrPr}rr$rE)rOrbrKrErcrwr1r2rPs  zDefaultDictVariable.__init__cs$|jtttfvr |js dStS)NF)rEr5rr|rbrqrrTrwr1r2rs z&DefaultDictVariable.is_python_constantcs,|jdusJd|jdtdS)Nz defaultdict(r))rErrqrTrwr1r2rszDefaultDictVariable.debug_reprr4cCs0t|tjtjjtjjfpt|tjo|jduSrN)rErr) functionsBaseUserFunctionVariablePolyfilledFunctionVariabler%rrr1r1r2is_supported_argsz$DefaultDictVariable.is_supported_argr5r(rr9cs|dkrSt|dkrt||dt|d|d|vr$|||dSt|jtr<|jjdur.r!Frr) rrErrrrrrrrrrr1rr2r/s,      zDefaultDictVariable.reconstructrNr1)r[r\r]r|r#r.rrrPrarrr staticmethodrJr5rrMrr6r1r1rwr2rsH    . rc s$eZdZdZejZdeede ddffdd Z de fdd Z e ded fd d Zedefd dZde fddZdefddZde fddZd*ddZddde deedee efdef ddZddde deedee efdef fd d! Zd+d#d$Zddd%edefd&d'Zd,d(d)ZZS)- SetVariablez.We model a sets as dictionary with None valuesrbrcr,Nc s\g}|D]}t|tjr||q||qt|t}t j |fi|dSrN) rErJrWrrAr|fromkeysrO_default_valuerqrP)rOrbrcrealized_itemsrrwr1r2rPUs  zSetVariable.__init__cCsh|jsdSg}|jD] }t|tjr|jn|}t|dr!t|jn|}| |q dd |dS)Nzset()rr,r) rbrErJrWrMrrrrrr)rOrbrhrMrr1r1r2rks  zSetVariable.debug_reprrLcCst|jSrN)setrbrrTr1r1r2 set_itemsvszSetVariable.set_itemscCtSrN)r$r1r1r1r2rQzrDzSetVariable._default_valuecCdd|jDS)NcSh|]}|jqSr1rrgrr1r1r2 z'SetVariable.as_proxy..rUrTr1r1r2rzSetVariable.as_proxycCrVrN)rTrTr1r1r2r=rBzSetVariable.python_typecCrW)NcSrXr1rrYr1r1r2rZr[z1SetVariable.as_python_constant..r\rTr1r1r2rr]zSetVariable.as_python_constantrr&cCs2|dd|jD|tdt|jddS)NcSr&r1rrr1r1r2r'r(z+SetVariable.reconstruct.. BUILD_SETr)foreachrUrrrrr1r1r2rszSetVariable.reconstructr5r(fnr9c Csz|dd|g|Didd|D}Wn"ty:}ztt||tttj|jdWYd}~nd}~wwt ||S)NcSsg|]}|qSr1rrr1r1r2r'sz0SetVariable._fast_set_method..cSsi|] \}}||qSr1rarr1r1r2rnroz0SetVariable._fast_set_method..r8) rbr>rr.r5mapr%r@r9r#build)rOr5r`r9rcresexcr1r1r2_fast_set_methods  zSetVariable._fast_set_methodrc sddlm}ddlm}|dvr*|||r*|tur*|}||t||||S|dkrS||tj |g|Ri|}|j j ||j |j |j tS|dkr||s_t|dkrpt||dt|d t|d d }|d tg}nH|d kr|s|rt||dt|d t|d z|jj} Wn ty} ztt|tttj| jdWYd} ~ nd} ~ wwt||| g|| S|dkr|st|dkrt||dt|d t|d ||t j!"|||d giS|dkr|r t||dt|d ||t j#"||g|d|$iS|dkrA|r2t||dt|d ||t j%"||g|iS|dkrh|rUt||dt|d ||t j&"||g|d|$iS|dkr|r|t||dt|d ||t j'"||g|d|$iS|dkr|rt||dt|d ||t j("||g|iS|dkr|st|dkrt||dt|d t|d ||t j)"||g|d|$iS|dkr|st|dkrt||dt|d t|d ||t j*"||g|iS|dkr<|+r<|r-t||dt|d ||t j,"||g|iS|dkrs|sKt|dkr\t||dt|d t|d |d |vrjtt||dt|d ||S|dkr|st|dkrt||dt|d t|d |d |vrt|d ||StS|dvrt|dkrt||dt|d t-j.t-j/d} |d 0} t1| ts||t"|| gi} ||| 2|"||| giS|d!vr2ddddd!2|} t3|d tt4j5fs#td"|d#|6d$|d 6d%}tt7||gd| dus*J||| ||S|d&vrzt3|d tt4j5fs_td"|d#|6d$|d 6d%}tt7||gdddddd&2|} | duspJ||| |||S|d'krt3|d tt4j5fstd(S||d||}tt|jd kS|t8vrt3|d tt4j5fstt9Stt8||j|d jSt||||S))Nr )check_constant_argsr!r6) isdisjointunion intersection differencesymmetric_differencerPrz1 args and 0 kwargsrrrrrprr8rhrjz0 kwargsclsintersection_updaterirkzExpect: 0 kwargs, Actual: difference_updaterlsymmetric_difference_updaterremovediscard)issubset issupersetrr)__and__r__xor____sub__z unsupported operand type(s) for z: 'z' and 'r)__iand__r__ixor____isub__rZF):utilsrgr:r7r=rTrfgetattrr@call_setrrr rbrrr$rrrOrQrUrprMrrr5rbr%r9rqrr set_isdisjointrset_intersectionpython_type_varset_intersection_update set_unionset_differenceset_difference_updateset_symmetric_differenceset_symmetric_difference_updater set_updateoperatorlegerArr0rErUserDefinedSetVariablepython_type_namer?rNotImplemented)rOr5rr9rcrgr7py_type temp_set_vtresulteoprVmrrrwr1r2rs                        " "    zSetVariable.call_methodr)cC ttSrN)rr)rTrTr1r1r2r zSetVariable.python_type_varr4cCstd)NzIllegal to getitem on a set) RuntimeError)rOr5r4r1r1r2rszSetVariable.getitem_constcCrArNr1rTr1r1r2rz)SetVariable.install_dict_keys_match_guardr1r,r)r2)r[r\r]r^r SET_CONTAINSrrr#rrPrrpropertyrTrUrNrQrr.r=rrr5r|rfrrrrr6r1r1rwr2rOPsf      b rOc seZdZdeddffdd ZdeeefddZdd d ede ffd d Z dd d ed e e de ee fde f fdd Z dd d ee de ee fddfddZZS)OrderedSetClassVariablercr,Nc stjdi|dSr-)rqrPr.rwr1r2rPsz OrderedSetClassVariable.__init__cCrVrNr rTr1r1r2rrBz*OrderedSetClassVariable.as_python_constantr5r(rcsH|dkrddlm}|jrt|j|}nd}||||dSt||S)N__new__r!)GetAttrVariable)rt)miscrrtrrqrM)rOr5rr attr_sourcerwr1r2rMs z#OrderedSetClassVariable.var_getattrr9csddlm}|dkr+t|dks|r#t||dt|dt|dtjgtdStt|}||vrJt |d tj rJ|d  |||dd|St ||||S) Nr!) set_methodsrr z'OrderedSet.__new__ only accepts one argrrr r) builtinrrrrOrderedSetVariabler"r|rTrErOrrq)rOr5rr9rcr resolved_fnrwr1r2rs"  z#OrderedSetClassVariable.call_methodrcCsbt|dks|rt|ddt|dt|dt|dkr"g}n|d|}tj|tdS)Nr!r zOrderedSet only accepts one argrrrr )rrforce_unpack_var_sequencerrr")rOr5r9rcrbr1r1r2rs z%OrderedSetClassVariable.call_function)r[r\r]rrPr.r rrr#rMr5r|rrrr6r1r1rwr2rs0   rc@sZeZdZdefddZdeefddZdeeefddZ de fdd Z dd dZ d S)rr,cCs\|jsdSg}|jD]\}}t|jdrt|jjn|j}||q dd|dS)NzOrderedSet([])rz OrderedSet([rS])r)rOrbrrhrr1r1r2rs  zOrderedSetVariable.debug_reprcCtdd|jDS)NcSsg|]}|jqSr1rrYr1r1r2r'r[z9OrderedSetVariable.as_python_constant..)r rUrTr1r1r2rrz%OrderedSetVariable.as_python_constantcCrVrNr rTr1r1r2r=rBzOrderedSetVariable.python_typecCstSrN)rrTr1r1r2rrz"OrderedSetVariable.python_type_varrr&NcsTfdddd|jDtdt|jdtdddS) Ncs ddS)Nztorch.utils._ordered_setr )load_import_fromr1rr1r2rs z0OrderedSetVariable.reconstruct..cSr&r1rrr1r1r2r'r(z2OrderedSetVariable.reconstruct.. BUILD_LISTrr!F)rr_rUrrrrrrr1rr2rs  zOrderedSetVariable.reconstructr1) r[r\r]rrr rrr.r=rrrr1r1r1r2rs  rc seZdZdefddZededfddZdefddZ d"d d Z de fd d Z d#ddZ dddedeedeeefdef fdd ZdedfddZdefddZdedefd d!ZZS)$FrozensetVariabler,cCX|jsdSg}|jD]}t|jdrt|jjn|j}||q dd|dS)Nz frozenset()rrrSrrrOrbrrr1r1r2r   zFrozensetVariable.debug_reprrLcCs |jSrN)rbrrTr1r1r2rU zFrozensetVariable.set_itemscCrVrN) frozensetrTr1r1r2r=rBzFrozensetVariable.python_typer)cCrrN)rr)rrTr1r1r2rrz!FrozensetVariable.python_type_varcCr)NcSrXr1rrYr1r1r2rZr[z7FrozensetVariable.as_python_constant..)rrUrTr1r1r2rrz$FrozensetVariable.as_python_constantrr&NcsPfdddd|jDtdt|jdgtdddS) NcsdgS)Nr)rcreate_load_globalr1rr1r2r sz/FrozensetVariable.reconstruct..cSr&r1rrr1r1r2r'r(z1FrozensetVariable.reconstruct..rrr!F)rr_rUrrrrrr1rr2r s zFrozensetVariable.reconstructr5r(rr9rccsZ|dvr td|d|dkrtS|dvr$t||||}t|jSt||||S)NrrprrqrrrIllegal call_method z on a frozensetrP)rurkrjrl)rr$rqrrrb)rOr5rr9rcrrwr1r2rs zFrozensetVariable.call_methodTcCr/)zB Frozensets are immutable and hashable in Python. Tr1rTr1r1r2rI6r0z$FrozensetVariable.is_python_hashablecCs t|SrN)rQrrTr1r1r2rS<r<z!FrozensetVariable.get_python_hashrVcCst|to ||kSrN)rEr#rrYr1r1r2rX?s z!FrozensetVariable.is_python_equalrr1)r[r\r]rrrrTrUr.r=rrrrr5r#r|rrrIr_rSr`rarXr6r1r1rwr2rs,    rc seZdZdefddZdddZddd eeddfd d Ze de fd d Z de fddZ de fddZddded eedeeefdef fdd ZZS)DictKeySetVariabler,cCrNz dict_keys([])rz dict_keys([rSrrrr1r1r2rGrzDictKeySetVariable.debug_reprNcCrArNr1rTr1r1r2rSrz0DictKeySetVariable.install_dict_keys_match_guardr5r(r9cCrArNr1rCr1r1r2rWr0z.DictKeySetVariable.install_dict_contains_guardcCrrNrrTr1r1r2rU]szDictKeySetVariable.set_itemscCrVrNrrTr1r1r2r=arBzDictKeySetVariable.python_typecCstdd|jDdS)NcSrXr1rrYr1r1r2rZfr[z8DictKeySetVariable.as_python_constant..)r|rPrUrrTr1r1r2rds z%DictKeySetVariable.as_python_constantrrccs*|dvr td|dt||||S)Nrrz on a dict_keys)rrqr)rOr5rr9rcrwr1r2riszDictKeySetVariable.call_methodr2)r[r\r]rrrr5r#rrrrUr.r=rr|rr6r1r1rwr2rFs2   rc seZdZUdZdZeeed<dede ddffdd Z e de fd d Z e de efd d Zd dde efddZdddZd ddedefddZd ddede edeeefdef fdd ZZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method Nkvr8rcr,c s8tjdi||jdvsJt|tsJ||_dS)N)rrrbr1)rqrPrrErJr8r9rwr1r2rP~s zDictViewVariable.__init__cCs |jdusJt|jj|jSrN)rr|r8rbrTr1r1r2 view_itemsszDictViewVariable.view_itemscCstrN)NotImplementedErrorrTr1r1r2 view_items_vtr0zDictViewVariable.view_items_vtr5r(cCrrN)rr)r1r1r2rrz$DictViewVariable.unpack_var_sequencerr&cCs2|jdusJ||j||j|ddS)Nr)rr8rrrr1r1r2rs  zDictViewVariable.reconstructrcCs0|jdusJ||jvrtdStdS)NTF)rr=r/r%r@r?r1r1r2r,s  z!DictViewVariable.call_obj_hasattrr9csf|dkr |j||||S|dkrddlm}||jtdS|dkr*t|St ||||S)Nrrr!rr __repr__) r8rrrrr"r%r@rrqrOr5rr9rcrrwr1r2rs zDictViewVariable.call_methodr1)r[r\r]r^rrr__annotations__rJrrPrrr5r#rrrr%r,r|rr6r1r1rwr2rus8    rc seZdZdZedeefddZedeefddZ de fddZ de fd d Z d d d e deedee efdef fdd ZZS)rrr,cC t|jSrN)rTrrTr1r1r2rUrzDictKeysVariable.set_itemscCrW)NcSr&r1rrr1r1r2r'r(z2DictKeysVariable.view_items_vt..rrTr1r1r2rzDictKeysVariable.view_items_vtcCrVrNrrTr1r1r2r=rBzDictKeysVariable.python_typecCrrrrrMrrrrrrr1r1r2rrzDictKeysVariable.debug_reprr5r(rr9rccs|dkr |j||||S|dvr"t|j|}||dj}t|S|tvrBt|dttfs4t t St t||j|djSt ||||S)Nr)rurxrrrwrzrvryr) r8rr|rUrOrrErr%r@rrq)rOr5rr9rcrrrwr1r2rs  zDictKeysVariable.call_method)r[r\r]rrrTr#rUr5rr.r=rrr|rr6r1r1rwr2rs&  rc@sBeZdZdZedeefddZdefddZ de fddZ d S) rrr,cCrrN)r5rrTr1r1r2rrz DictValuesVariable.view_items_vtcCrVrN)rrTr1r1r2r=rBzDictValuesVariable.python_typecCsR|jsdSg}|jD]}t|drt|jn|}||q dd|dS)Nzdict_values([])rz dict_values([rSr)rrrrrrr)rOrbrhrr1r1r2rs  zDictValuesVariable.debug_reprN) r[r\r]rrr5r#rr.r=rrr1r1r1r2rs rc seZdZdZedeefddZdefddZ de fddZ d d d e d eed e e efdef fdd Z dedfddZZS)rrbr,cCrW)NcSs g|] \}}t|j|gqSr1)rrrMrr1r1r2r' rz3DictItemsVariable.view_items_vt..rrTr1r1r2rrzDictItemsVariable.view_items_vtcCrVrN)rrTr1r1r2r= rBzDictItemsVariable.python_typecCs|jsdSg}|jD]/\}}t|jdrt|jjn|j}t|dr)t|jn|}|d|d|dq dd|dS) Nzdict_items([])r(rrFz dict_items([rSrrrr1r1r2rs zDictItemsVariable.debug_reprr5r(rr9rccs|dkr/t|dkrt||dt|dt|dtr*|j|d|djgiStdS|dkrAddlm }||j t d St ||||S) NrZr!rrrFrrr ) rrrErr8rr%r@rrrr"rqrrwr1r2rs   zDictItemsVariable.call_methodFcCr/)zC Dictionary item views are not hashable in Python. Fr1rTr1r1r2rI4r0z$DictItemsVariable.is_python_hashable)r[r\r]rrr5r#rr.r=rrr|rrrIr6r1r1rwr2rs$  rrN)Jr^r}rrr:collections.abcrrtypingrrrrrtorch.utils._ordered_setr r r r rbytecode_transformationrrrrrerrguardsrrrtrrrr{rrrrrrr ryr"r#constantr$r%torch._dynamo.codegenr&r;r(torch._dynamo.variables.builtinr)rGr*rar3rCrHrJr7r@rrOrrrrrrrrr1r1r1r2sh $      M ~?C V/?;