`ir5 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 dlZ G d d Ze G d d
Ze G d d Zd
efdZd Zd dlmZ ed Zd Zd ZdS ) N) innermost_fn)_debug_get_cache_entry_list)CodeType)ListCallableDictUnionSet) dataclassc e Zd ZU i Zeed f ed< e Ze e ed< e
defd Z e
defd Z e
d
de
defd Zdefd Zd
Zej e
d Zd S ) CodeProxy instancesused_instancesnamec d}| }| d r'| d d } | d| }|t j vrn|dz
} |S )Nr :T_ )endswithr
r )r inew_names g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/depyf/explain/utils.py get_new_namezCodeProxy.get_new_name si
S ! ! 9D
"==Q==9#666Q
c X t | }d t j |< |S N)r
r r )r r s r consume_new_namezCodeProxy.consume_new_name s' ))$//(, H%r Fcodec ddl m} t | d r| j } | j d s| j d r.t
| j }| j }n+t
| }|s
|| | }nd}t | }||_ d|j d|j d |_
|t j |j < |S )
Nr )decompile_ensure__code__transformed_code___transformed_code_ z
z
```python
z
```
) depyf.utilsr! hasattrr" co_name
startswithopen co_filenamereadr
r r raw_coder r )r r skip_decompiler! srcr selfs r decompile_with_namez CodeProxy.decompile_with_name# s
000000 4 $ $ !=D < " "#6 7 7 4<;R;RSh;i;i t'((--//C|HH --d33H!
&&tX66~~ I *. DI& r c r d d | D | _ d S )Nr% c N g | ]"}| d k d|z dz #S )r% z
)strip).0lines r
z&CodeProxy.__init__..A s4 T T TD QSASASTD[4
ASASASr )join
splitlinesr- )r0 r s r __init__zCodeProxy.__init__> s8 T TDOO,=,= T T TV V
r c X t j | j | j S r )r
r addr r0 s r __str__zCodeProxy.__str__C s# $$TY///yr c # V K t t _ t j V d S r )setr
r r r recordzCodeProxy.recordG s) $'55 &&&&&&r N)F)__name__
__module__ __qualname__r r str__annotations__rA r r
staticmethodr r r r1 r; r?
contextlibcontextmanagerrC rB r r r
r
s! (*ItC$%*** "suuNCH$$$
3
\
s \
( # \4VS V V V V
' ' \ ' ' 'r r
c ~ e Zd ZU eed< eed< ee ed< eed< eed< eed< e edf ed< d Z
d
Z d S ) CacheResult
original_codetransformed_codeguardcompiled_subgraphcompiled_subgraph_proxytransformed_code_proxyDynamoOptimizationResultreferenced_global_functionsc v || _ d}t |d rdnd}t || } t t j j dd pDt t j j dd p$t t j j j dd }|J t || }n# t $ r Y nw xY w|s%|j }|j
j } d |j D }
ndfdg }|j
} | | t | d r'| D ]} t | | |j t# } g }
nLt# |j } t' |j }
|| _ d t- | |
D | _ |j }
d
|
j D }t5 | d k sJ |rNt7 ||d | _ t: | j |d d
| _ nd | _ d | _ |
| _ t: | j d | _! d |
j D }i | _" |D ]'}tG || j
|| | j" |< (d S )NF
guard_managercheck_fnGuardManagerWrapper GuardManagerc g | ] }|j
S rB )
cell_contents)r6 xs r r8 z(CacheResult.__init__..r s Q Q Q!ao Q Q Qr c | D ]A}t |t j j j j rsdn1t || B| D ]} || d S )NT) get_leaf_guards
isinstancetorch_C_dynamoguardsNO_TENSOR_ALIASINGappend_guard_codeget_child_managers)rootans
leaf_guardchildtensor_aliasing_guard_seenvisits r rm z#CacheResult.__init__..visitw s "&"6"6"8"8 7 7J!*eh.>.E.XYY % 9 %9= 6 6 $%j#6666 ! 4 4 6 6 & &EE%%%%%& &r get_epilogue_lambda_guardsc i | ]\ }}|| S rB rB r6 r values r
z(CacheResult.__init__.. s [[[uu[[[r c < g | ]}| d |S )
__compiledr) r6 r s r r8 z(CacheResult.__init__.. s@ M M Mdool.K.K M M M Mr r r T)r. ztransformed_code:c < g | ]}| d |S )__resumeru rv s r r8 z(CacheResult.__init__.. s@ K K Kdooj.I.IK K K Kr )rN
function_namemodule)$rN r' getattrra rc rd rb r` Exception
code_partsr" co_freevars __closure__rh rn rf closure_varstuplekeyslistvaluesrP zipfreevarsr co_nameslenr rQ r
r1 rR rO rS rU rT )r0 rN rz cache cpp_guard attr_namerW klass guard_codes
freevar_namesfreevar_valuesrh lambda_guardr compiled_subgraphs
resume_fnsr rl rm s @@r r; zCacheResult.__init__X sX *
(/uo'F'FVOOJ y11
EM 02GNN G
,ndCCG(/ FF
$$$"=%88II D $ K (2K ) 2 >M Q Q}7P Q Q QNN */ &
&
&
&
&
&
& K %D
E$ $ $ $t 9::
A$($C$C$E$E A AL%lK@@@@ )1 %
!# %m&@&E&E&G&G H H
!%m&@&G&G&I&I!J!J
[[M>8Z8Z[[[
z M M !] M M M%&&!++++ 0%1&9KA9N2O%P%PD "+4+H+H&(:1(=d ,I ,T ,TD ( ( &*D "+/D ( $ &/&C&C !#6'8 '8#K K !]K K K
+-( D5M $Tl 3 " 6 6 6 D ,T 2 2 s A9B+ +
B87B8c | j t | j | j t | j ndd | j D dS )Nz"No compiled subgraph."c > i | ]\ }}|| S rB to_data)r6 r fns r rr z'CacheResult.to_data.. s? ,@ ,@ ,@(t bjjll,@ ,@ ,@r )rP rO rQ rU )rP rG rS rR rU itemsr> s r r zCacheResult.to_data s Z #+!- !- 261M1Y "%,". ". "._x,@ ,@6<<>>,@ ,@ ,@
A A Ar N) rD rE rF r rH r rG r r
r r; r rB r r rM rM N s 9 &&&& %%%%!%c+E&E!FFFFW W W r A A A A Ar rM c b e Zd ZU eed< eed< eed< eed< ee ed< d
dZ
d Z d Z dS ) rT ry rz rN source_code_proxytransformed_code_entriesNc | _ |
j | _ n|| _ | _ t }fd|D | _ t | j | j | _ d S )Nc 2 g | ]}t | S rB )rM )r6 r rz rN s r r8 z5DynamoOptimizationResult.__init__.. s8 )K )K )K:?K
vu 5 5)K )K )Kr ) rN r( ry rz r r r
r1 r )r0 rN ry rz cachess ` ` r r; z!DynamoOptimizationResult.__init__ s * !.!6D !.D ,];;)K )K )K )K )KCI)K )K )K%!*!>!> 2"4 "4 r c ^ | j t | j d | j D d}|S )Nc 6 g | ]}| S rB r )r6 entrys r r8 z4DynamoOptimizationResult.to_data.. s3 )L )L )L$)
)L )L )Lr )ry source_coder )ry rG r r )r0 datas r r z DynamoOptimizationResult.to_data sT ! / & ( ()L )L-1-J)L )L )L M M r c | j j }| d ddd }| }| j }|j |j z }|j d | }dd d |D z dz }|d |z z
}|d
z
}d }| j
D ]} d d | j D }
|
rd
|
z }
dg} | j
D ] } | dd| dz | d d | }
| j j d r!| j j dd }nt$ d }|d|
z d| dz |
z z
}| j <| j j }|dz
}|d| dz
}|dz
}|dz
}|dz
}|d z
}|d z
}|d| d z
}|dt+ | j j z dz z
}| j D ] \ }}| |z } |d d | d z d!z d"| j j d#d | d$z z
}|d%t+ | j j z dz z
}|d&d"| j j d#d | d$z z
}||z d'| j d(z S ))Nr zdef zdef transformed_r z__local_dict = {z, c g | ]
}d | d| S )"z": rB rv s r r8 z3DynamoOptimizationResult.to_src.. s) AAAt*4**D**AAAr }z
z
__global_dict = globals()r% c , g | ]\ }}|d v | d| dS )) __builtins__z = '''z'''
rB rp s r r8 z3DynamoOptimizationResult.to_src.. sZ G G G eim vF jF jF4 ; ;u ; ; ; jF jF jFr zD# Note: the following variables are used inside the guard function.
z __guard_hit = True
z z __guard_hit = __guard_hit and r4 z return __guard_hit
r$ __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 ============#
)r r- r: replacer5 rN co_argcountco_kwonlyargcount co_varnamesr9 r r r rP appendrS r r) r
r rR remove_indentationrU to_srcry )r0 r- signaturer code_objnormal_arg_count arg_namesarg_dictadditional_coder r guard_linesr] rP guard_func_name
subgraph_namer funcs r r z DynamoOptimizationResult.to_src s5 )2 ''))!,44 &+ + %#/(2LL ():*:):; % IIAAyAAA B BCEHI )) << 2 & Z & ZE ww G Gu~OcOcOeOe G G G H HH
nbemm;"C"K"KLacm"n"n"+"<" ? ? @BF G G
U 6 ;UUdii >R>RUUUV V % L(: L L LM Mr )NN)
rD rE rF rG rH dictr r
r rM r; r r rB r r rT rT s LLL "; //// 4 4 4 4 BM BM BM BM BMr rT r c | }t |d t |d z
d fd|D S )Nr r% c * g | ]}|d dz S )Nr4 rB )r6 r7 indents r r8 z&remove_indentation..# s& ;;;TDMD(;;;r )r: r lstripr9 )r linesr s @r r r sc OO E
q]]Sq !2!2 3 3
3F
77;;;;U;;; < < 1 > 5> 5> A c t | 5 dd l}d} |t | z }|j | s>t |d 5 }| | d d d n # 1 swxY w Y n.t | }|| k rn|dz
}|cd d d S # 1 swxY w Y d S )Nr Twr )r r rG pathexistsr* writer, )r/ r r count new_filepathf
existing_codes r write_code_to_file_templater 9 s[ m $ $
(3u::5L7>>,//
,,, !GGCLLL! ! ! ! ! ! ! ! ! ! ! ! ! ! ! . . 3 3 5 5M ## QJE
s6 AB>A6* B>6A: :B>=A: >3B>>CCc | dd l } ddlm} ddlm} t || dz
}dt | z S )Nr )_unique_id_counter)copyr __compiled_fn_)ra %torch._dynamo.bytecode_transformationr r nextrG )ra r r
current_counts r get_current_compiled_fn_namer K sa LLLHHHHHH 01122Q6M c- 0 0 00r )ra torch._dynamo.eval_framer r inspectdistypesr typingr r r r r
dataclassesr rJ r
rM rT rG r rf rK r r r rB r r r s 1 1 1 1 1 1 @ @ @ @ @ @
3 3 3 3 3 3 3 3 3 3 3 3 3 3 ! ! ! ! ! ! ?' ?' ?' ?' ?' ?' ?' ?'D kA kA kA kA kA kA kA kA\ _M _M _M _M _M _M _M _MD=S = = = =
& % % % % %
$1 1 1 1 1r