x-j+ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z gZ Gdd Z Gd d ZGd d e jZiZ ddZdZdZdZdS)N)Sequence)core) use_pir_api)gast) ORIGIN_INFOc:eZdZdZdZddZdZedZdS)Locationz. Location information of source code. filepathlineno col_offsetNc0||_||_||_dSNr )selfr r rs `/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/jit/dy2static/origin_info.py__init__zLocation.__init__'s   $c6d|jd|jd|jS)Nz location: :r rs r__str__zLocation.__str__,s&KDMKKDKKK$/KKKrc|j|jfSr)r r rs r line_locationzLocation.line_location/s t{++rr) __name__ __module__ __qualname____doc__ __slots__rrpropertyrrrr r sfI %%%% LLL,,X,,,rr c.eZdZdZdZdZdZdZdZdS) OriginInfoz. Original information of source code. location function_name source_codec0||_||_||_dSrr$)rr%r&r's rrzOriginInfo.__init__?s  *&rc6|jd|jd|jdS)Nz source_code: z in function z )r%r'r&rs rrzOriginInfo.__str__Ds*-ii1AiiQUQciiiirc d}d|jjd|jjd|jd|d|j S)Nz(* user code *)z File "", line , in  z r%r r r&r'lstrip)rflag_for_origin_infos rformatted_messagezOriginInfo.formatted_messageGs0bDM2bbDMz0OriginInfoAttacher.transform..ds$GGG$TZZ--GGGrr) r8getsourcelinesr:lenr/r source_lines lineno_offsetvisitr7)rrG begin_lineno begin_lines r transformzOriginInfoAttacher.transform`s%,%;DI%F%F" l!!_ j//C 0A0A0C0C,D,DDGG,GGG)A- 49rcJt|tjr|j|t |dd||||t|tjr|j|S)Nr ) isinstancer FunctionDefr=appendgetattr_attach_origin_info generic_visitpoprnodes rrIzOriginInfoAttacher.visiths dD, - - +   $ $T * * * 44 ( ( 4  $ $T * * * 4   dD, - - $   ! ! # # # rc~t|tjsJt|dsJ||}||}t |j||}|jdj }|j |j dz }t|||}t|t|dS)Nr r)rNrASThasattr _abs_lineno_abs_col_offsetr r r=namerGr r#setattrr)rrVr rloc func_name code_line origin_infos rrRz&OriginInfoAttacher._attach_origin_infoss$)))))tX&&&&&!!$''))$// t}fj99%b). %dkAo6  i;; k;/////rc |j|jzSr)rHr rUs rr[zOriginInfoAttacher._abs_linenos!DK//rc |j|jzSr)rrUs rr\z"OriginInfoAttacher._abs_col_offsets00rN) rrrrrrLrIrRr[r\r!rrr5r5Tsx    0 0 000011111rr5Tci}tjtj|}t j|}t ||}t||D]\}}t|t|ks+Jdt|dt|dt|td}t|td} || | j j } | | } | 6| j j|j jkr| j j|j jkr||| <t ||rt S|S)a Creates a original information map between transformed static function and original dygraph function. Args: transformed_node(gast.AST): The AST node of transformed dygraph function with attached source information of original dygraph function. static_func(Callable): The static function transformed by dygraph function corresponding to transformed_node. Returns: The original information map. @The node types should be the same, but received type(t_node) is , and type(s_node) is .N)textwrapdedentr8r<rparseattach_origin_infoast_walktyperQrr%rgetr rglobal_origin_info_mapupdate) transformed_node static_func is_globalorigin_info_map static_source static_nodet_nodes_node dygraph_info static_info static_locexist_origin_infos r!create_and_update_origin_info_mapr~sOOG$5k$B$BCCM*]++K$[+>>K"#3[AA33F||tF||+++ CtTZ|| C Cswx~ss C C C,++v{D99 fk488  ;#6  )7 +// ;;  (!*1(/00!*5(344&2 ##!!/222&%% rcNt||}||S)ap Attach original source information to AST node according corresponding function. Args: ast_node(gast.AST): The AST node to attach original source information. func(Callable): The corresponding function of ast_node. Parse the original information from this function. Returns: An AST node attached original source information. )r5rL)ast_noder:resolvers rrlrls*"(D11H  Orc#nKd}||}||}|rt|t|ksJ|}|}t|t|krMt|tjtjfs&t|tjtjfrt|t|ks+Jdt|dt|d||fV|jD]}t||}t||} t|tj r+| || | ht|ttfr|t|t| ksJt|| D]I\} } t| tj r*| | | | J|dSdS)a Recursively yield all descendant nodes in the trees starting at transformed_node and static_node (including itself) in parallel. NOTE(liym27): Function ast.walk is not used because it yield all descendant nodes in no specified order. cX|gSt|trt|n|gSr)rNrlist)xs r_as_listzast_walk.._as_lists- 9I$Q11:tAwwws:rrfrgrhN)rFrTrnrNrLoadParam_fieldsrQrYrPrtuplezip) rrrwrtransformed_node_liststatic_node_listrxryfield t_node_child s_node_childd_items_items rrmrmsk;;; %H%566x ,, 8())S1A-B-BBBBB&**,,!%%'' <<4<< ' '&49dj"9:: jDJ/?? F||tF||+++ CtTZ|| C Cswx~ss C C C,++fn^ 8 8E"6511L"6511L,11 8%,,\::: '' 5555L4-88 8<((C ,=,=====&), &E&E88NFF!&$(338-44V<<<(//777? 88888rc d} fd tj}|}trO|} |}|D],}||r||j|_-nx|jD]p}t|j D]Y\}}||r?| |} || } | || R#YVxYwZq|S)ze Replaces op callstack information about transformed static code with original dygraph code. cPt|dzdksJtdt|dD]}||dd}|ddd}t |ddd}|dd d}||dzd}t ||}t |j}|r| \}}}}d |d |d |||<d |||dz<|S)a An example of callstack: File "path1/to/file.py", line 10, in func_1 y = paddle.tensor.fill_constant(x, shape=[1], dtype="int32") File "path2/to/file.py", line 740, in fill_constant stop_gradient=True) File "path3/to/file.py", line 43, in append_op return self.main_program.current_block().append_op(*args, **kwargs) File "path4/to/file.py", line 2811, in append_op attrs=kwargs.get("attrs", None)) File "path5/to/file.py", line 1919, in __init__ for frame in traceback.extract_stack(): rr-,rXrNz File "r+r,z ) rFranger/splitintr rprorr3) callstacki file_liner r funcnamecoder_dygraph_func_infos rget_new_op_callstackzBupdate_op_callstack_with_origin_info..get_new_op_callstacksG 9~~!Q&&&&q#i..!,, - -A!! ++C0066s;;I |AbD)H1abb)**F |ABB'HQU#**3//D8V,,C 6 : :3;L M M   P3D3M3M3O3O0&(DOhOOOOXOOIaL,d}}Ia!e  rcg}|jD]<}|||D]}||z }=|Sr)opsrPblocks)blockrop sub_blockget_all_pir_block_opss rrzCupdate_op_callstack_with_origin_info..get_all_pir_block_ops(sa) 8 8B JJrNNNYY[[ 8 8 ,,Y777 8 r) rop_proto_and_checker_makerkOpCreationCallstackAttrNamer global_blockhas_attrrr enumeraterattr _set_attr) programrop_makercallstack_var_namerrrrrrrs @r$update_op_callstack_with_origin_infors\ !!!F.H!>>@@}}++-- ##L11 B BB{{-.. B33BLAA  B^  E"59--  2;;122  "(: ; ;I 4 4Y ? ?I %7CCCC   Ns C66C:)T)r8ricollections.abcr paddle.baserpaddle.frameworkr paddle.utilsrutilsr__all__r r#NodeTransformerr5rpr~rlrmrr!rrrsV$$$$$$(((((( ,,,,,,,,0        @0101010101-010101f.21111h    /8/8/8dGGGGGr