o
=)ir5 @ s d dl Z d dlmZ d dlmZ d dlZd dlZd dlmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ d dlZG dd dZeG d d
d
ZeG d d d Zd
efddZdd Zd dlmZ edd Zdd Zdd ZdS ) N) innermost_fn)_debug_get_cache_entry_list)CodeType)ListCallableDictUnionSet) dataclassc @ s e Zd ZU i Zeed f ed< e Ze e ed< e
defddZ e
defddZ e
dd e
defd
d Zd efd d
Zdd Zeje
dd ZdS ) CodeProxy instancesused_instancesnamec C sL d}| }| dr$| d d } | d| }|tjvr |S |d7 }q|S )Nr :T_ )endswithr r )r inew_name r _/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/depyf/explain/utils.py get_new_name s
zCodeProxy.get_new_namec C s t | }d t j|< |S N)r r r )r r r r r consume_new_name s
zCodeProxy.consume_new_nameFcodec C s ddl m} t| dr| j} | jds| jdr%t| j }| j}nt
|}|s2|| |}nd}t |}||_ d|j d|j d |_
|t j|j < |S )
Nr )decompile_ensure__code__Ztransformed_code___transformed_code_ z
z
```python
z
```
)Z depyf.utilsr hasattrr co_name
startswithopen co_filenamereadr r r raw_coder r )r r skip_decompiler srcr selfr r r decompile_with_name# s&
z CodeProxy.decompile_with_namec C s d dd | D | _d S )Nr c S s$ g | ]}| d krd| d qS )r z
)strip.0liner r r
A s $ z&CodeProxy.__init__..)join
splitlinesr& )r) r r r r __init__> s
zCodeProxy.__init__c C s t j| j | jS r )r r
addr r) r r r __str__C s zCodeProxy.__str__c c s t t_tjV d S r )setr r
r r r r recordG s zCodeProxy.recordN)F)__name__
__module__ __qualname__r r str__annotations__r7 r
r staticmethodr r r r* r3 r6
contextlibcontextmanagerr8 r r r r r s
r c @ sb e Zd ZU eed< eed< ee ed< eed< eed< eed< e edf ed< d d
Z
d d Z d
S ) CacheResult
original_codetransformed_codeguardcompiled_subgraphcompiled_subgraph_proxytransformed_code_proxyDynamoOptimizationResultreferenced_global_functionsc s || _ d}t|dr dnd}t||}z&ttjjdd p,ttjjdd p,ttjjjdd }|d us3J t||}W n tyB Y nw |sU|j }|j
j } dd |j D }
n=d fdd g }|j
} | | t| d
ry| D ]} t| | qq|jd u rt } g }
nt|j } t|j }
|| _d d t| |
D | _|j}
d
d |
jD }t|dksJ |rt||d | _ t j | j |d dd| _ nd | _ d | _ |
| _ t | j d| _!dd |
jD }i | _"|D ]}t#|| j
||d| j"|< qd S )NF
guard_managerZcheck_fnZGuardManagerWrapperZ GuardManagerc S s g | ]}|j qS r )
cell_contents)r. xr r r r0 r s z(CacheResult.__init__..c sR | D ]}t|tjjjjr sd nqt|| q| D ]}|| q d S )NT) Zget_leaf_guards
isinstancetorch_C_dynamoguardsZNO_TENSOR_ALIASINGappend_guard_codeZget_child_managers)rootansZ
leaf_guardchildZtensor_aliasing_guard_seenvisitr r rW w s z#CacheResult.__init__..visitget_epilogue_lambda_guardsc S s i | ]\}}||qS r r r. r valuer r r
s z(CacheResult.__init__..c S g | ] }| d r|qS )Z
__compiledr" r. r r r r r0
r r T)r' ztransformed_code:c S r\ )Z__resumer] r^ r r r r0 r_ )rB
function_namemodule)$rB r getattrrN rP rQ rO rM ExceptionZ
code_partsr co_freevars __closure__rS rX rR Z closure_varstuplekeyslistvaluesrD zipfreevarsr co_nameslenr rE r r* rF rC rG rI rH )r) rB ra cacheZ cpp_guard attr_namerJ klassZ guard_codesZ
freevar_namesZfreevar_valuesrS Z lambda_guardr Zcompiled_subgraphsZ
resume_fnsr r rV r r3 X sz
zCacheResult.__init__c C s: | j t| j| jd urt| jnddd | j D dS )Nz"No compiled subgraph."c S s i | ] \}}|| qS r to_data)r. r fnr r r r[ s
z'CacheResult.to_data..)rD rC rE rI )rD r< rG rF rI itemsr5 r r r rr s
zCacheResult.to_dataN) r9 r: r; r r= r r< r r r r3 rr r r r r rA N s
YrA c @ sT e Zd ZU eed< eed< eed< eed< ee ed< d
ddZ
d d
Z d d Z dS )rH r` ra rB source_code_proxytransformed_code_entriesNc sX | _ |d u r j| _n|| _ | _t} fdd|D | _t| j | j| _d S )Nc s g | ]}t |qS r )rA )r. rn ra rB r r r0 s z5DynamoOptimizationResult.__init__..) rB r! r` ra r rv r r* ru )r) rB r` ra cachesr rw r r3 s
z!DynamoOptimizationResult.__init__c C s$ | j t| jdd | jD d}|S )Nc S s g | ]}| qS r rq )r. entryr r r r0 s z4DynamoOptimizationResult.to_data..)r` Z source_coderv )r` r< ru rv )r) datar r r rr s z DynamoOptimizationResult.to_datac C sN | j j}| d ddd}| }| j}|j|j }|jd | }dd dd |D d }|d
| 7 }|d 7 }d }| j
D ]} d d
d | j D }
|
rUd|
}
dg} | j
D ]
} | dd| d q[| d d | }
| jjdr| jjdd}ntd}|d|
d| d |
7 }| jd ur| jj}|d7 }|d| d7 }|d7 }|d7 }|d 7 }|d 7 }|d 7 }|d| d 7 }|dt| jj d 7 }| j D ]
\}}| | }q|d
d | d! d" d#| jj d$d | d% 7 }q@|d&t| j j d 7 }|d'd#| j j d$d | d% 7 }|| d(| j d) S )*Nr zdef zdef transformed_r z__local_dict = {z, c S s g | ]
}d | d| qS )"z": r r^ r r r r0 z3DynamoOptimizationResult.to_src..}z
z
__global_dict = globals()r c S s( g | ]\}}|d vr| d| dqS )) __builtins__z = '''z'''
r rY r r r r0 s ( zD# Note: the following variables are used inside the guard function.
z __guard_hit = True
z z __guard_hit = __guard_hit and r+ z return __guard_hit
r Z__guard_zguard:z (L, G, **___kwargs_ignored):
z*# Note: please refer to the graph code in z*.py.
zP# Captured Graph: Dynamo generated graph (debuggable when using eager backend).
z?# Joint graph: joint forward+backward graph from aot autograd.
z\# Forward graph: forward graph from aot autograd (debuggable when using aot_eager backend).
z^# Backward graph: backward graph from aot autograd (debuggable when using aot_eager backend).
z;# AFTER XXX: graph processed by inductor (not debuggable).
z (*args, **kwargs):
pass
zif z (__local_dict, __global_dict):
z zreturn ()z
# Note: if there is a transformed version below, this function might well not be executed directly. Please check the transformed version if possible.
z
# Note: this function might well not be executed directly. It might well be transformed again, i.e. adding one more guards and transformed code.
z
#============ end of z ============#
)ru r& r2 replacer, rB co_argcountco_kwonlyargcount co_varnamesr1 rv rk rt rD appendrG r r" r r rF remove_indentationrI to_srcr` )r) r& signaturer Zcode_objZnormal_arg_count arg_namesZarg_dictZadditional_codery rk Z guard_linesrL rD Zguard_func_nameZ
subgraph_namer funcr r r r s
z DynamoOptimizationResult.to_src)NN)
r9 r: r; r< r= dictr r r rA r3 rr r r r r r rH s
rH r c s< | }t|d t|d d fdd|D S )Nr r c s g | ]
}| d d qS )Nr+ r r- indentr r r0 # r| z&remove_indentation..)r2 rm lstripr1 )r linesr r r r s r c C s$ | D ] }| }|| qd S r )Zverbose_code_partsr, r )rD rT Z verbose_strr r r rR % s rR )r@ c c s^ | d }ddl m} dd l}||}z|
d V W d W d S 1 s&w Y W d S w )Nz.lockr )FileLock)Zfilelockr os)
path_templateZ lock_pathr r lockr r r lock_on_file, s r c C s t |H dd l}d} |t| }|j|s3t|d
}|| W d n1 s-w Y nt| }|| kr>n|d7 }q |W d S 1 sOw Y d S )Nr Twr )r r r< pathexistsr# writer% )r( r r countZ new_filepathfZ
existing_coder r r write_code_to_file_template9 s"
$r c C s< dd l } ddlm} ddlm} t||d }dt| S )Nr )_unique_id_counter)copyr Z__compiled_fn_)rN Z%torch._dynamo.bytecode_transformationr r nextr< )rN r r Z
current_countr r r get_current_compiled_fn_nameK s
r )rN Ztorch._dynamo.eval_framer r inspectdistypesr typingr r r r r dataclassesr
r? r rA rH r< r rR r@ r r r r r r r s* Bnb