o
    -i                     @   s  d Z ddlZddlZddlZddlZddlZddlmZmZ ddl	m
Z
 ddlmZmZmZ ddlZddlmZ ddlmZmZmZ dd	lmZmZmZmZ dd
lmZ ddlmZm Z  ddl!m"Z"m#Z#m$Z$m%Z% ddlm&Z&m'Z'm(Z( ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 ddl1m2Z2 erddl3m4Z4 ddl5m6Z6 ddl7m8Z8 de9eef de9eef de:ddfddZ;de<e de<e ddfddZ=G dd dZ>ej?dd ded! fd"d#Z@ej?dd ded! fd$d%ZAej?dd ded! fd&d'ZBdS )(a  
Side effect tracking and management for TorchDynamo's compilation system.

This module provides infrastructure for tracking and managing side effects that occur
during symbolic execution, including:

- Tracking mutations to objects, attributes, and variables
- Managing context changes (cell variables, global namespace modifications)
- Handling aliasing and object identity preservation
- Managing stack frame state and local variable changes
- Tracking function calls with side effects

Key classes:
- SideEffects: Main container for tracking all side effects during execution
- MutableSideEffects: Specialization for mutable object tracking
- AttributeMutation/ValueMutation: Track specific types of mutations
- Various specialized side effect classes for different scenarios

The side effect system ensures that mutations performed during symbolic execution
are properly replayed during runtime, maintaining the correctness of compiled code
while enabling optimizations where safe.
    N)	GeneratorMutableMapping)CellType)AnyOptionalTYPE_CHECKINGAutogradFunctionContextVariable   )graph_break_hintsutils	variables)bytecode_from_templatecreate_call_functioncreate_call_methodcreate_instruction)	PyCodegen)SideEffectsErrorunimplemented_v2)GlobalSourceLocalCellSourceLocalSourceSource)is_frozen_dataclassnn_module_new
object_new)AttributeMutationAttributeMutationExistingAttributeMutationNewis_side_effect_safeValueMutationExistingValueMutationNewVariableTracker)FrozenDataClassVariable)OutputGraph)InstructionTranslatorBase)ListVariable	dict_fromdict_to	mro_indexreturnc                 C   s<   t |j| }|| |  D ]\}}|||| qd S N)type__mro__clearitems__setitem__)r'   r(   r)   
dict_classkv r4   S/home/ubuntu/LTX-2/.venv/lib/python3.10/site-packages/torch/_dynamo/side_effects.py_manual_dict_setitemA   s
   
r6   	list_fromlist_toc                 C   s   t | t ||  d S r+   )listr.   extend)r7   r8   r4   r4   r5   _manual_list_updateN   s   
r;   c                       s`  e Zd ZU dZeeef ed< eeeeef f ed< e	e
 ed< 					ddddeeeef  deeeeeef f  dee	e
  dee	eee	e f   d	eeeed
edef f  ddf fddZdeddfddZdeddfddZdedefddZdd dee fddZdddZde
defddZde
defddZdefd d!Zdefd"d#Zdefd$d%Zdedefd&d'Zded(ed)eddfd*d+Z	,	,dded(ed-ed.edef
d/d0Zd1ed)eddfd2d3Zd1edefd4d5Zd6ed(edefd7d8Z d6ed(ed)eddfd9d:Z!e"d;e#defd<d=Z$dedefd>d?Z%dedefd@dAZ&ded(edefdBdCZ'dedefdDdEZ(e)fde
dFedGe#defdHdIZ*e*Z+de
dFedefdJdKZ,dLe-dMe
dNe
dOeee
f def
dPdQZ.dMe#de#fdRdSZ/dTedUedVe	e de
fdWdXZ0dTedUedVe	e defdYdZZ1defd[d\Z2d]ee- d^e3d_edefd`daZ4d]e-de
defdbdcZ5ddedee	e ddfdfdgZ6		 		ddhdiZ7ddldmZ8deddfdndoZ9defdpdqZ:de	e fdrdsZ;dte<ddfdudvZ=dwd
dxedydd(eddf
dzd{Z>d|eddfd}d~Z?dte<ddfddZ@dddZAdte<ddfddZBdefddZCdddZD  ZES )SideEffectsah  
    Maintain records of mutations and provide methods to apply them during code generation.

    Handles tracking and applying side effects during PyTorch Dynamo compilation,
    maintaining Python semantics by managing mutations, attribute modifications,
    and other side effects that occur during program execution.

    Key responsibilities:
    - Tracks mutations to Python objects, lists, and dictionaries that need to be
    applied after an FX graph is run.
    - Manages attribute modifications and deletions
    - Handles tensor hooks and backward pass state
    - Tracks cell variable mutations and global variable changes
    - Ensures correct ordering and application of side effects after graph execution

    This ensures that optimized code behaves identically to the original Python code with
    respect to object mutations and other side effects.
    id_to_variablestore_attr_mutations	keepaliveNoutput_graphr$   save_for_backwardtensor_hookszvariables.TensorVariablez!variables.RemovableHandleVariabler*   c                    s`   t    t|| _|pi | _|pi | _|pg | _|pg | _|p"i | _	d| _
d | _t | _d S )NF)super__init__weakrefrefoutput_graph_weakrefr=   r>   r?   rA   rB   _has_existing_dict_mutationca_final_callbacks_varset"ignore_mutation_on_these_variables)selfr@   r=   r>   r?   rA   rB   	__class__r4   r5   rD   k   s   





	zSideEffects.__init__varc                 C   s   | j | dS )zMutations to this variable will be executed but not not tracked,
        typically used for temporary mutations that are later restored.N)rK   addrL   rO   r4   r4   r5   ignore_mutations_on   s   zSideEffects.ignore_mutations_onc                 C   s   || j v r| j | dS dS )zQRemove a variable from the skip mutation set, restoring normal mutation tracking.N)rK   removerQ   r4   r4   r5   stop_ignoring_mutations_on   s   
z&SideEffects.stop_ignoring_mutations_onotherc                 C   s>   t |tsJ | j|jko| j|jko| j|jko| j|jkS r+   )
isinstancer<   r=   r>   rA   rB   )rL   rU   r4   r4   r5   __eq__   s   


zSideEffects.__eq__c                 C   s   | j |j kr| j  }|j  }||krd| d| S dS | j|jkr<| j }|j }||kr:d| d| S dS | j|jkrDdS | j|jkrLdS d S )Nzid_to_variable keys: z != zid_to_variable: unknown diffzstore_attr_mutations keys: z"store_attr_mutations: unknown diffrA   rB   )r=   keysr>   rA   rB   )rL   rU   sk_itvok_itvsk_samok_samr4   r4   r5   diff   s"   



zSideEffects.diffc                 C   sJ   |   }|dus
J | j|t| jdd | j D t| j| j| j	dS )zCreate a shallow copyNc                 S   s   i | ]	\}}|t |qS r4   )dict.0r2   r3   r4   r4   r5   
<dictcomp>   s    z%SideEffects.clone.<locals>.<dictcomp>)r@   r=   r>   r?   rA   rB   )
rG   rN   r^   r=   r>   r/   r9   r?   rA   rB   )rL   rF   r4   r4   r5   clone   s   zSideEffects.cloneitemc                 C   s   t || jv S r+   )idr=   rL   rc   r4   r4   r5   __contains__      zSideEffects.__contains__c                 C   s   | j t| S r+   )r=   rd   re   r4   r4   r5   __getitem__   rg   zSideEffects.__getitem__c                 C   s(   |   }t|o|jjjjo|jjjjS r+   )rG   bool
current_txoutputcurrent_tracerunder_activation_checkpoint#allow_side_effects_under_checkpointrL   r@   r4   r4   r5   *should_allow_side_effects_under_checkpoint   s   

z6SideEffects.should_allow_side_effects_under_checkpointc                 C      |   }t|o|jjjjS r+   )rG   ri   rj   rk   rl   ,unsafe_allow_externally_visible_side_effectsro   r4   r4   r5   9should_allow_externally_visible_side_effects_in_subtracer   s
   
zESideEffects.should_allow_externally_visible_side_effects_in_subtracerc                 C   rq   r+   )rG   ri   rj   rk   rl   is_reconstructing_generatorro   r4   r4   r5   rt      s
   
z'SideEffects.is_reconstructing_generatorc                 C   s`   ddl m} t||rdS |  rdS |  rdS |  r!tdt|js.t	dddg d d	S )
Nr   r   TzCannot reconstruct a generator with variable mutations. Dynamo needs to fully exhaust the generator, which may cause unintended variable modifications.zOHigherOrderOperator: Mutating a variable not in the current scope (SideEffects) This is not supported.gb_typecontextexplanationhintsF)
torch._dynamo.variables.miscr	   rV   rs   rp   rt   r   r   mutation_typer   )rL   rc   r	   r4   r4   r5   check_allowed_side_effect   s&   

z%SideEffects.check_allowed_side_effectnamevaluec                 C   s>   |  |sJ | | || jvri | j|< || j| |< d S r+   )is_attribute_mutationr~   r>   )rL   rc   r   r   r4   r4   r5   
store_attr  s
   


zSideEffects.store_attrF
deleted_okcheckc                 C   sV   |r	|  |s	J | j| | }|s)t|tjr)tdd| d| dg tjd |S )Nz$Attempted to read a deleted variablezitem: z, name: ru   rw   )r   r>   rV   r   DeletedVariabler   r   
USER_ERROR)rL   rc   r   r   r   resultr4   r4   r5   	load_attr  s   zSideEffects.load_attrcellvarc                 C   s\   |  rtdd| d| dg tjd t|tjsJ t|tjs%J | |d| d S )NzWrite to immutable cellz	cellvar: z	, value: zFDynamo doesn't support writing to immutable/sourceless cell variables.rw   cell_contents)	is_immutabler   r   	DIFFICULTrV   r   CellVariabler"   r   )rL   r   r   r4   r4   r5   
store_cell(  s   zSideEffects.store_cellc                 C   sV   t |tjsJ | |dr| j|dddS |jr|jS tdt|dg tj	d d S )Nr   F)r   zRead uninitialized cellzBAttempted to read a cell variable that has not been populated yet.rw   )
rV   r   r   has_pending_mutation_of_attrr   pre_existing_contentsr   strr   r   )rL   r   r4   r4   r5   	load_cell4  s   
zSideEffects.load_cellgvarc                 C   s   t |tjsJ | ||S r+   )rV   r   r"   r   )rL   r   r   r4   r4   r5   load_globalA  s   zSideEffects.load_globalc                 C   s2   t |tjsJ t |tjsJ | ||| d S r+   )rV   r   r"   r   )rL   r   r   r   r4   r4   r5   store_globalE  s   zSideEffects.store_globalclsc              
   C   s6   t | dd tjtjtjtjtjtjt	jt
jtjf	v S )N__getattribute__)inspectgetattr_staticobjectr   r^   rJ   	frozensetintr   r9   tupleBaseException)r   r4   r4   r5   "cls_supports_mutation_side_effectsL  s   z.SideEffects.cls_supports_mutation_side_effectsc                 C   s   t |jtS r+   )rV   r}   r   re   r4   r4   r5   r   Z     z!SideEffects.is_attribute_mutationc                 C   s   |  |ot| j|S r+   )r   ri   r>   getre   r4   r4   r5   has_pending_mutation]  s   
z SideEffects.has_pending_mutationc                 C   s   |  |o|| j|dv S )Nr4   )r   r>   r   )rL   rc   r   r4   r4   r5   r   b  s
   z(SideEffects.has_pending_mutation_of_attrc                 C   s\   |  rdS t|jttfrdS t|tjr || jv p|| S | 	|r*|| jv S |jj
S )NFT)r   rV   r}   r   r!   r   UserDefinedObjectVariabler>   is_underlying_vt_modifiedr   is_modifiedre   r4   r4   r5   r   g  s   

zSideEffects.is_modifiedvariablemutation_type_clsc                 C   s\   t || jv rt| d|j d| jt | j d| |_|| jt |< | j| |S )z7Start tracking an existing or new variable for mutationz is already tracked for mutation. This could be because you are not using VariableBuilder to construct the variable tracker. Source of new object: z'. Source of previously tracked object: .)rd   r=   AssertionErrorsourcer}   r?   append)rL   rc   r   r   r4   r4   r5   
_track_objx  s   zSideEffects._track_objc                 C   s   | j ||tdS )N)r   )r   r   )rL   rc   r   r4   r4   r5   track_object_existing  s
   z!SideEffects.track_object_existing
cls_sourceuser_clsvariable_clsoptionsc                 C   s   |t jjju r#tjdd t j }W d    n1 sw   Y  nt|}||fdt|i|}|| j	t
|< | j| |S )NT)recordr}   )torchautogradfunctionFunctionCtxwarningscatch_warningsFunctionr   r   r=   rd   r?   r   )rL   r   r   r   r   objr   r4   r4   r5   track_object_new  s    zSideEffects.track_object_newc                 C   s  ddl m} ddlm} ddlm} ddlm} tj	}t
||r(||r(|}n]t|dr9t|dr9||s9|}nLt
|tjjrDtj}nAt
|ttjfrPtj}n5t
|ttfr[tj}n*t
|trdtj}n!t
|trmtj}nt
|trvtj}nt|r}t}nt
|t rtj!}t
|tj	sJ |S )	Nr   )TorchFunctionModer
   )GenericContextWrappingVariable)TorchFunctionModeVariable)is_forbidden_context_manager	__enter____exit__)"torch.overridesr   variables.ctx_managerr   variables.torch_functionr   variables.user_definedr   r   r   
issubclass is_supported_torch_function_modehasattrr   nnModuleUnspecializedNNModuleVariabler^   collectionsOrderedDictUserDefinedDictVariablerJ   r   UserDefinedSetVariabler   UserDefinedTupleVariabler9   UserDefinedListVariabler   MutableMappingVariabler   r#   r   "UserDefinedExceptionObjectVariable)rL   r   r   r   r   r   r   r4   r4   r5   get_variable_cls  sJ   



zSideEffects.get_variable_clsbase_cls_vtcls_vt	init_argsc                 C   sv   |j }t|tjjrt|}|S t|tjr|j	}nt|tj
r$|j }ntd| tj
|js4J ||}|S )NzUnexpected base_cls_vt )r   r   r   r   r   r   rV   r   BuiltinVariablefnUserDefinedClassVariableRuntimeErroris_supported_new_method__new__)rL   r   r   r   r   r   base_clsr4   r4   r5   get_example_value  s   

zSideEffects.get_example_valuec           	      C   sZ   |j }|j}| |}| |||}|||j ||t|d}|| jt|< | j| |S )aM  
        Creates a UserDefinedObjectVariable (or its subclass) variable tracker
        and mark it for attribute mutation tracking.

        Also records the variable trackers to call __new__ method on
        reconstruction. Roughly, the reconstruction looks like this
            base_cls_vt.__new__(user_cls, *init_args)
        )r   r   r   r}   )	r   r   r   r   r   r=   rd   r?   r   )	rL   r   r   r   r   r   r   r   r   r4   r4   r5   track_new_user_defined_object  s   
z)SideEffects.track_new_user_defined_objectc                 C   s2   t  }tjt d}|| jt|< | j| |S )Nr}   )r   r   r   r   r=   rd   r?   r   )rL   r   r   r4   r4   r5   track_cell_new  s   zSideEffects.track_cell_newr   cellcontentsc                 C   s<   t j|d u rd nt ||d}|| jt|< | j| |S )N)r}   r   r   )r   r   r   r=   rd   r?   r   )rL   r   r   r   r   r4   r4   r5   track_cell_existing  s   zSideEffects.track_cell_existingc                 C   s.   t jt |d}|| jt|< | j| |S )N)r}   r   )r   NewGlobalVariabler   r=   rd   r?   r   )rL   r   rc   r   r4   r4   r5   track_global_existing,  s   z!SideEffects.track_global_existingctxargsc                 C   s$   t |tjsJ | j||f d S r+   )rV   r   r	   rA   r   )rL   r   r   r4   r4   r5   track_save_for_backward5  s   z#SideEffects.track_save_for_backwardc                 C   sJ   |j D ]}t|}|j| }|| jvr"t|tjtjfr"| || qd S r+   )r?   rd   r=   rV   r   TensorVariableSymNodeVariabler   )rL   rU   
other_itemother_idother_variabler4   r4   r5   -track_runahead_tensor_and_symvar_side_effects;  s   

z9SideEffects.track_runahead_tensor_and_symvar_side_effectstxr%   c                    s   t  t  dtdd ffdddtdtffdd dd j D }g }|}|d urA||j|jg |j}|d us1t	|||j
jjg   fd	d
j D _ fdd
j D _d S )NrO   r*   c                    sP   | v rd S  |  t| jtr  |  | jv r&tj|   d S d S r+   )rP   rV   r}   r   r>   r"   visitrO   )live_new_objectsrL   r   visitedr4   r5   r   N  s   


z0SideEffects.prune_dead_object_new.<locals>.visitc                    s   t | jtr
|  v S dS NTrV   r}   r   r   )r   r4   r5   is_live^  s   z2SideEffects.prune_dead_object_new.<locals>.is_livec                 S   s   g | ]
}t |jts|qS r4   r   r`   rO   r4   r4   r5   
<listcomp>c  s    
z5SideEffects.prune_dead_object_new.<locals>.<listcomp>c                    s   i | ]\}} |r||qS r4   r4   r_   r   r4   r5   ra     
    z5SideEffects.prune_dead_object_new.<locals>.<dictcomp>c                    s   i | ]\}} |r||qS r4   r4   r_   r   r4   r5   ra     r   )rJ   r"   ri   r=   valuesr:   stacksymbolic_localsparentr   rk   backward_staterB   r/   r>   )rL   r   pre_existing_varsinit_live_varscur_txr4   )r   r   rL   r   r   r5   prune_dead_object_newI  s:   	

z!SideEffects.prune_dead_object_newc                 C   s`   || j v rd S | | t|jtrd|j_|jr*t|tjr,t|tj	s.d| _
d S d S d S d S r   )rK   r~   rV   r}   r    r   r   r   ConstDictVariableSetVariablerH   rQ   r4   r4   r5   mutation  s   




zSideEffects.mutationc                 C   s   | j S r+   )rH   rL   r4   r4   r5   has_existing_dict_mutation  s   z&SideEffects.has_existing_dict_mutationc                    s    fdd j  D S )Nc                    s   g | ]	}  |r|qS r4   )r   r   r	  r4   r5   r     s    z2SideEffects._get_modified_vars.<locals>.<listcomp>)r=   r   r	  r4   r	  r5   _get_modified_vars  s   zSideEffects._get_modified_varscgc                    s  |   D ]ɉtjtsjd usJ qttjrJjd u r>  fdd  	t
dd   t j _qjd u rItj_qttjrottjjrn dd      t j _qttjr~tdddg d	 qttjrd fdd} | n	  fdd jjd usJ  jj jD ]} | q 	t
dtj d   t j _q| jD ]&\}} |j  d |D ]} | q 	g tt|td qd S )Nc                           tjdS )N	make_cellload_import_fromr   __name__r4   r  r4   r5   <lambda>      z3SideEffects.codegen_save_tempvars.<locals>.<lambda>r   Fallow_cachez<AutogradFunctionContextVariable escaped Dynamo-traced regionru   zAWe cannot reconstruct a torch.autograd.Function's context object.rw   r*   c                      s.   j d usJ  j    dg d S )Nr   )r   extend_outputcreate_load_attrr4   r  rO   r4   r5   load_new_method  s   
z:SideEffects.codegen_save_tempvars.<locals>.load_new_methodc                      r  )Nr   r  r4   r  r4   r5   r    r  r
   rA   POP_TOPr*   N)r  rV   r}   r   r   r   r   
local_nameadd_push_nullr  r   	add_cacher   tempvarsr   r   torch_functionTensorWithTFOverrideVariable	clear_tosr	   r   r   r   r   lenrA   load_methodr   r   )rL   r  r  argr   r   r4   r  r5   codegen_save_tempvars  sr   




	







z!SideEffects.codegen_save_tempvarstensorhookhandlec                 C   s   t |tjsJ t |tjsJ t |tjr| sJ ttj|s$J t	| j
 }|| j
v r9|d7 }|| j
v s0||||f| j
|< |jrGJ ||_d S )Nr
   )rV   r   r   r"   RemovableHandleVariable
is_mutabler   r   Tensorr$  rB   rX   idx)rL   r(  r)  r*  r   r.  r4   r4   r5   register_hook  s   




zSideEffects.register_hookr.  c                 C   s   | j |= d S r+   )rB   )rL   r.  r4   r4   r5   remove_hook  r   zSideEffects.remove_hookc                    sh   | j  D ],\}}jsJ dd fdd} |  |  tdd  | qd S )Nz4Hooks on non input tensors NYI - should not get herer*   c                      s        g d S r+   )r  r  r4   r  r   r(  r4   r5   gen_fnD  s   z)SideEffects.codegen_hooks.<locals>.gen_fnr
   Fr  )rB   r   r   r  r  r   r  )rL   r  r)  r*  r2  r4   r1  r5   codegen_hooks  s   #
zSideEffects.codegen_hooksvariables.ListVariablec                 C   s.   ddl m} | jd u rtjg | d| _| jS )Nr
   )r!   r   )variables.baser!   rI   r   r&   )rL   r!   r4   r4   r5   get_ca_final_callbacks_varP  s   
z&SideEffects.get_ca_final_callbacks_varc              	      s  g }|   D ]ottjr5 dd  j   d  d tdddg |tdg qttj	j
rtjtjrIjjd u sRtddd	d
gd  j  d  dd |g tdtd  j  d |g tdtd qttjr̈ rˈ j  d  dd jr j  d |g tdtd jr|g tdtd qttjjr$  fdd  dd tj } jd  |f7  <  td|d   fdd  j  tdtjd  dd  td qttjrKjd urK| j v rJ| !} | | "jg q| #rttj$r| %j&ri }t'j(j)D ]} j*j+, ||< qezt-jj./t0j1}W n t2y   t-jj./t3}Y nw  td|dtd|d dg  j  td|d dg  j&dd  td|d dg t4t'|d}|g |td nUttj5r-| %j6r-i }t7j(j)D ]} j*j+, ||< q j  td|d dg  j6dd  td|d dg t4t7|d}|g |td t8| j 9i : D ]\}}	ttj;rc j*j+<|  |	 tjt=sWJ |td |dg q8t|	tj>rtj?t@rtAtBd!d |r j*j+<|  j |td"|dg q8ttjCrɈD|rɈ  fd#d  j  t|  |	 |g tEd$dtd q8ttjCrF r Gd%d&  d'  j  t|  |	 |g td$td q8 j*j+<|  |	   |td(|dg q8qttjHrDtIj/D ]}
  fd)d  j  dd  J  q'qttjKrqd- fd+d,} |  LjMN  |g tEddtd qtOt-t8|D ]} | q{d S ).NFr  BUILD_SLICE   )r&  STORE_SUBSCRz1Side effect on existing deque with limited maxlenru   rv   z*Don't use a deque with `maxlen` specified.rw   r:   r
   r  r.   r   updatec                      r  )Nget_torch_function_mode_stackr  r4   r  r4   r5   r        z4SideEffects.codegen_update_mutated.<locals>.<lambda>co_varnames
STORE_FAST)argvalc                      r  )Nset_torch_function_mode_stackr  r4   r  r4   r5   r    r<  
BUILD_LIST
LOAD_CONSTr)   r(   r'   )varname_mapr8   r7   STORE_GLOBALr   DELETE_ATTRc                      r  )N object_setattr_ignore_descriptorr  r4   r  r4   r5   r  V  r<     builtinsr   __setattr__
STORE_ATTRc                      r  )N	iter_nextr  r4   r  r4   r5   r  x  r  r*   c                      s    j   d d S )Nsetstate)r   r   r4   r  r4   r5   r2       
z2SideEffects.codegen_update_mutated.<locals>.gen_fnr  )Pr  rV   r   r&   r   r  create_load_constr   r   listsDequeVariablemaxlenConstantVariabler   r   r%  r   r  has_new_itemsshould_reconstruct_allr!  TorchFunctionModeStackVariabler  call_functionget_prev_stack_var_namecode_optionsappend_outputforeachsymbolic_stackr$  r   r  r>   r   create_store_derefr   r   r   _dict_vtr6   __code__r=  r   rk   new_varr,   r-   indexr   r   
ValueErrorr^   r   r   _list_vtr;   reversedr   r/   r   update_co_namesr   r   r}   r   r   getattrr   should_skip_descriptor_setterr   needs_slow_setattrr  ListIteratorVariablerangepop_topRandomVariable
wrap_staterandomgetstater   )rL   r  suffixesr   contents_varrC  r)   dict_update_instslist_update_instsr   _r2  suffixr4   r  r5   codegen_update_mutatedZ  s  






	










	







6


z"SideEffects.codegen_update_mutatedc                 C   s*   t t| j| j p| jp| jp| j S r+   )anymapr   r=   r   rB   rA   r	  r4   r4   r5   is_empty  s   zSideEffects.is_emptyc                 C   s   | j   | j  d S r+   )r?   r.   r=   r	  r4   r4   r5   r.     rM  zSideEffects.clear)NNNNN)r*   r<   )FF)rU   r<   r*   N)r   r%   r*   N)r*   r4  r  )Fr  
__module____qualname____doc__r^   r   r"   __annotations__r   r9   r   r   r   r	   rD   rR   rT   r   ri   rW   r]   rb   rf   rh   rp   rs   rt   r~   r   r   r   r   r   r   staticmethodr,   r   r   r   r   r   r    r   track_mutabler   r   r   r   r   r   r   r   r   r   r   r   r  r  r
  r  r   r'  r/  r0  r3  r6  ru  rx  r.   __classcell__r4   r4   rM   r5   r<   S   sz  
 
.








'



	


E_

6
  ;r<   r   r%   )NNNc                 c   sH    | j jjsJ | j jj}zd| j j_d V  W || j j_d S || j j_w r   )rk   rl   rm   rn   r   orig_valr4   r4   r5   rn     s   

rn   c                 c   :    | j jj}zd| j j_d V  W || j j_d S || j j_w r   )rk   rl   rr   r  r4   r4   r5   2allow_externally_visible_side_effects_in_subtracer     

r  c                 c   r  r   )rk   rl   rt   r  r4   r4   r5   "disallow_side_effects_in_generator  r  r  )Cr{  r   
contextlibr   r   rE   collections.abcr   r   typesr   typingr   r   r   torch.nnr   r|   r	   ru   r   r   r   bytecode_transformationr   r   r   r   codegenr   excr   r   r   r   r   r   r   r   r   r   r5  r   r   r   r   r    r!   r"   r   r#   torch._dynamo.output_graphr$   torch._dynamo.symbolic_convertr%   torch._dynamo.variables.listsr&   r^   r   r6   r9   r;   r<   contextmanagerrn   r  r  r4   r4   r4   r5   <module>   sx    $	


        U