o ߗi5B@sJddlZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddl mZddlmZmZdd lmZmZdd lmZe rLd d lmZmZGd ddeZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZ ddZ!defddZ"Gd d!d!e#Z$Gd"d#d#e$d$Z%d%d&Z&dS)'N)Enum)AnyCallableDictListOptional TYPE_CHECKING) variables)current_scope_id unimplemented) GuardBuilder install_guard) AttrSourceSource)istype)InstructionTranslatorInstructionTranslatorBasec@seZdZdZdZdZdS) SourceTypea This Enum divides VariableTracker into 2 cases, depending on the variable it represents: - already existed that Dynamo began tracking while introspection (Existing) - is a new variable that is created during Dynamo introspection (New) In general, we have these invariants: 1. for `VariableTracker` associated with `Existing`, its `source` field must not be None. 2. for `VariableTracker` associated with `New`, most of the time its `source` field is None, except for cases like side effect codegen for `AttributeMutationNew`, during which we generate a `LocalSource('tmp...')` for such variable, to facilitate codegen. rrN)__name__ __module__ __qualname____doc__ExistingNewrrZ/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_dynamo/variables/base.pyrsrc@s"eZdZdZdeddfddZdS) MutationTypez Base class for Variable.mutation_type. It encodes information about 1. The type of mutation Dynamo allows on the variable. 2. Whether the value represented by this variable already existed before Dynamo tracing. typreturnNcCs<|tjur d|_dS|tjurt|_dStd|dS)NrzUnsupported SourceType: )rrscoperr r selfr rrr__init__.s    zMutationType.__init__)rrrrrr%rrrrr&srcs2eZdZdZd fdd ZddZdd ZZS) ValueMutationNewa This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value itself (rather than its attributes). 2. The value is created by the bytecode Dynamo is tracing through. For instance, Dynamo could model a newly created list with this marker, indicating that while we need to model mutations to this list, we don't have to emit bytecode for these mutations if the list doesn't escape into the Python world. r!NcttjdSN)superr%rrr$ __class__rrr%ZzValueMutationNew.__init__cCst|Sr()idr*rrr__hash__]zValueMutationNew.__hash__cCs||uSr(r)r$otherrrr__eq__`r0zValueMutationNew.__eq__r!N)rrrrr%r/r2 __classcell__rrr+rr&Ns  r&cs2eZdZUdZeed<ddeffdd ZZS)ValueMutationExistinga This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value itself (rather than its attributes). 2. The value exists before Dynamo tracing started. For instance, Dynamo could model a pre-existing list with this marker, indicating that if we encounter mutations to this list, we need to buffer and re-apply those mutations after the graph runs, since the list might be used afterwards in Python. is_modifiedFcttj||_dSr()r)r%rrr6)r$r6r+rrr%u zValueMutationExisting.__init__)F)rrrrbool__annotations__r%r4rrr+rr5ds r5cs&eZdZdZdeffdd ZZS)AttributeMutationz This case of VariableTracker.mutation_type marker indicates that Dynamo allows mutation on the value's attributes. r cst|dSr()r)r%r#r+rrr%zAttributeMutation.__init__)rrrrrr%r4rrr+rr;zsr;cs eZdZdZfddZZS)AttributeMutationExistinga This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value's attributes. 2. The value exists before Dynamo tracing started. For instance, Dynamo could model a pre-existing object with this marker, indicating that if we encounter mutations to this object, we need to buffer then re-apply those mutations after the graph runs, since the object might be used afterwards in Python. cr'r()r)r%rrr*r+rrr%r-z"AttributeMutationExisting.__init__)rrrrr%r4rrr+rr=s r=cs,eZdZdZddeeffdd ZZS)AttributeMutationNewa This case of VariableTracker.mutation_type marker indicates 1. Dynamo allows mutation on the value's attributes. 2. The value is created by the bytecode Dynamo is tracing through. For instance, Dynamo could model a newly created object with this marker, indicating that while we need to model mutations to this object, we don't have to emit bytecode for these mutations if the object doesn't escape into the Python world. N cls_sourcecr7r()r)r%rrr?)r$r?r+rrr%r8zAttributeMutationNew.__init__r()rrrrrrr%r4rrr+rr>s r>cCs|dkS)Nrr)scope_idrrr_is_top_level_scoper0rAmcCst}t|r dS|j|kS)NT)r rAr")rBr@rrris_side_effect_safes rCcs0eZdZgZdefddZdfdd ZZS)VariableTrackerMetar!cCs0ttj|r|ttjfvr|}t||S)z-Make isinstance work with LazyVariableTracker)type__instancecheck__r LazyVariableTrackerVariableTrackerrealize)clsinstancerrrrFs z%VariableTrackerMeta.__instancecheck__Ncs t|||tj|dSr()r)r%rDall_subclassesappend)rJnamebasesattrsr+rrr%szVariableTrackerMeta.__init__r3)rrrrLr9rFr%r4rrr+rrDs rDc s eZdZdZhdZddZe dUdedgdfdede e e efd dfd d Z d e fd d ZddZddZddZddZddZddZddde d efddZddde d dfdd Zd!d"Zd#d$Zd%d&Zd'd(Zd edfd)d*Zd edfd+d,Zd efd-d.Zd efd/d0Z d ee fd1d2Z!ddde d dfd3d4Z"   5 6 7 8 dVd9d:Z# 5 6 7 8 dWd;d<Z$d=d>Z%dXd?d@Z&dXdAdBZ'dCdDZ(dEdFZ)dGdHZ*dIdJZ+dKdLZ,e- dUddMdedNe e.d efdOdPZ/dddQdNe.dRe0d dffdSdTZ1Z2S)YrHz Base class for tracked locals and stack values VariableTracker instances are immutable and should be copied in order to change them. Prefer the factory function VariableTracker.build() over VariableTracker.__init__(). >valueguardssource mutation_typeparents_trackeruser_code_variable_namecKs$t|j}|||jdi|S)z)Shallow copy with some (optional) changesNr)dict__dict__updater,)r$kwargsargsrrrclones  zVariableTracker.cloneNfnrQcacher!cCs|duri}t|}||vrdS|||<t|tr?|}|||}|j}|jD]\}}||vr<||||q-dSt|t t frT|D] }||||qHdSt|t t j frj|D] }||||q`dSdS)zM Walk value and call fn on all the VariableTracker instances N)r. isinstancerHunwrap_nonvar_fieldsrXitemsvisitrlisttuplerW collections OrderedDictvalues)rJr]rQr^idxnonvarskeysubvaluerrrrcs2   zVariableTracker.visitcCs|jjdS)Nz())r,rr*rrr__repr__szVariableTracker.__repr__cCs*zt|WStyt|YSwr()repras_python_constantNotImplementedErrorr*rrr debug_reprs   zVariableTracker.debug_reprcCs.zt|WStyt|ddw)a Abstract method to be implemented by subclasses of VariableTracker. This method should return the type represented by the instance of the subclass. The purpose is to provide a standardized way to retrieve the Python type information of the variable being tracked. Returns: type: The Python type (such as int, str, list, etc.) of the variable tracked by the subclass. If the type cannot be determined or is not relevant, leaving it undefined or invoking super() is always sound. Note: This is an abstract method and may be overridden in subclasses. Example: class SetVariable(VariableTracker): def python_type(self): return set Raises: NotImplementedError: If the method is not implemented in a subclass. z has no typeN)rErorpr*rrr python_types  zVariableTracker.python_typecCst|d)z For constantsz is not a constantrpr*rrrro+sz"VariableTracker.as_python_constantc Cs>z|WSty}z tt|WYd}~dSd}~ww)zcSimilar to as_python_constant(), but add ID_MATCH guards to try to force things to become constantsN)rorpr str)r$errrguard_as_python_constant/s  z(VariableTracker.guard_as_python_constantcC$z|WdStyYdSwNTF)rorpr*rrris_python_constant6  z"VariableTracker.is_python_constantcCs|jr |j|Str()rS make_guardrp)r$r]rrrr{=s zVariableTracker.make_guardtxrrNcCt)z/getattr(self, name) returning a python constantrsr$r|rNrrr const_getattrBzVariableTracker.const_getattrcCsR|||}tj|st|jot|j|}|r!t|t j tjj ||dS)z,getattr(self, name) returning a new variable)rS) rr ConstantVariable is_literalrprSrrr{rCONSTANT_MATCHcreate)r$r|rNrQrSrrr var_getattrFs  zVariableTracker.var_getattrcCrwrx)as_proxyrpr*rrris_proxyPrzzVariableTracker.is_proxycCs tt|r()rprtr*rrrrWs zVariableTracker.as_proxycCsBz|}ddl}t||jjr|jWSWdSty YdSw)Nr)rtorch.fxr_fxProxynoderp)r$proxytorchrrr maybe_fx_nodeZs zVariableTracker.maybe_fx_nodecCr}r(rs)r$codegenrrr reconstructezVariableTracker.reconstructcCr}r(rsr$r|rrrunpack_var_sequencehrz#VariableTracker.unpack_var_sequencecC ||Sr()rrrrrforce_unpack_var_sequenceks z)VariableTracker.force_unpack_var_sequencecCs&z||WdStyYdSwrx)rrprrrrhas_unpack_var_sequencets   z'VariableTracker.has_unpack_var_sequencecCrr()rrrrrhas_force_unpack_var_sequence|s z-VariableTracker.has_force_unpack_var_sequencecCstd|dS)Nzinspect_parameter_names: r r*rrrinspect_parameter_namesr-z'VariableTracker.inspect_parameter_namescCstd|jjd|dS)Nzhasattr  )r r,rr~rrr call_hasattrszVariableTracker.call_hasattrr[List[VariableTracker]rZDict[str, VariableTracker]cCstd|d|d|dS)Nzcall_function rr )r$r|r[rZrrr call_functionszVariableTracker.call_functionc Cs|dkr||r|s |rJtjt||S|dkr6t|dkr6|dr6|s6|||dSt d|d|d|d|dS)N__len__ __getattr__rrz call_method r) rr rrlenrryrror )r$r|rNr[rZrrr call_methods   $zVariableTracker.call_methodcCsdSr(r)r$rNrrr set_name_hintrzVariableTracker.set_name_hintcC|S)z=Used by LazyVariableTracker to build the real VariableTrackerrr*rrrrIrzVariableTracker.realizecCr)zSUsed by LazyVariableTracker to return the real VariableTracker if it already existsrr*rrrr`rzVariableTracker.unwrapcCsdS)z:Used by LazyVariableTracker to indicate an unrealized nodeTrr*rrr is_realizedrzVariableTracker.is_realizedcCstd|ddS)Nznext()r rrrr next_variableszVariableTracker.next_variablecCs|jo||Sr()strict_checks_fnrrrris_strict_moder<zVariableTracker.is_strict_modecCs | S)z0Whether Dynamo allows mutation on this variable.) is_immutabler*rrr is_mutable zVariableTracker.is_mutablecCs |jduS)z.Whether Dynamo bans mutation on this variable.N)rTr*rrrrrzVariableTracker.is_immutablerrScCs2ddlm}|dur|j||S||||S)z=Create a new VariableTracker from a value and optional Sourcer)builderN)rSourcelessBuilderrVariableBuilder)r|rQrSrrrrbuilds zVariableTracker.build)rSrTrTcsbt||_||_|dur-t|ttfr|dusJdSt|ttfs'J|dus/JdSdSr() r)r%rSrTr_r&r>r5r=)r$rSrTr+rrr%s   zVariableTracker.__init__r()r|rr[rrZrr!rH)r[rrZrr!rH)r!rH)3rrrrrar\ classmethodrrrrintrcrtrmrqrrrorvryr{rrrrrrrrrr9rrrrrrrrIr`rrrrr staticmethodrrrr%r4rrr+rrHs    !        rH) metaclasscGs>t|dkr|\}t|trt|St|jSdtt|S)Nrr) rr_rHrtrErjoinmaptypestr)objsobjrrrrs   r)'rfenumrtypingrrrrrrrr r excr rRrrrSrrutilsrsymbolic_convertrrrrr&r5r;r=r>rArCrErDrHrrrrrs2      (   !