o "j(,@sddlZddlmZddlmZddlmZddlmZddl m Z m Z gZ Gdd d Z Gd d d ZGd d d ejZiZ dddZddZddZddZdS)N)Sequence)core)Program)gast) ORIGI_INFOunwrapc@s2eZdZdZdZd ddZddZedd ZdS) Locationz. Location information of source code. filepathlineno col_offsetNcC||_||_||_dSNr )selfr r r ra/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/dy2static/origin_info.py__init__& zLocation.__init__cCsd|jd|jd|jS)Nz location: :r rrrr__str__+szLocation.__str__cCs |j|jfSr)r r rrrr line_location.s zLocation.line_locationr) __name__ __module__ __qualname____doc__ __slots__rrpropertyrrrrrr s r c@s4eZdZdZdZddZddZddZd d Zd S) OriginInfoz. Original information of source code. location function_name source_codecCrrr )rr!r"r#rrrr>rzOriginInfo.__init__cCsd|j|j|jS)Nz&{} source_code: {} in function {} )formatr!r#r"rrrrrCs zOriginInfo.__str__cCs&d}d|jj|jj|j||jS)Nz(* user code *)z$ File "{}", line {}, in {} {} {})r$r!r r r"r#lstrip)rZflag_for_origin_inforrrformated_messageHszOriginInfo.formated_messagecCs|jj|jj|j|jfSr)r!r r r"r#r%rrrras_frameRs zOriginInfo.as_frameN) rrrrrrrr&r'rrrrr3s rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)OriginInfoAttacherzZ Attach original source information to AST node according corresponding function. cCs6||_t||_t|j|_t|j|_g|_dSr) rootrfuncinspect getsourcefiler getsourcer# current_func)rr)r*rrrr`s   zOriginInfoAttacher.__init__cCsXt|j\}}|d}t|t||_dd|D|_|d|_||j dS)NrcSsg|]}|dqS) )strip).0linerrr ksz0OriginInfoAttacher.transform..r) r+getsourcelinesr*lenr%r source_lines lineno_offsetvisitr))rr6Z begin_linenoZ begin_linerrr transformgs  zOriginInfoAttacher.transformcCsPt|tjr |j|t|dr||||t|tjr&|j|S)Nr ) isinstancer FunctionDefr.appendhasattr_attach_origin_info generic_visitpoprnoderrrr8os       zOriginInfoAttacher.visitcCsxt|tjsJt|dsJ||}||}t|j||}|jdj }|j |j d}t |||}t |t|dS)Nr r)r:rASTr= _abs_lineno_abs_col_offsetr r r.namer6r rsetattrr)rrBr r loc func_nameZ code_lineZ origin_inforrrr>zs    z&OriginInfoAttacher._attach_origin_infocCs |j|jSr)r7r rArrrrEs zOriginInfoAttacher._abs_linenocCs |j|jSr)r rArrrrFs z"OriginInfoAttacher._abs_col_offsetN) rrrrrr9r8r>rErFrrrrr([s  r(Tc Csi}t|}t|}t||}t||D]P\}}t|t|ks.Jdt|t|t|t d}t|t d} |dusB| durCq| j j } | | } | durb| j j |j j krYq| j j|j jkrbq||| <qt||rptS|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)r+r-rparseattach_origin_infoast_walktyper$getattrrr!rgetr r global_origin_info_mapupdate) transformed_nodeZ static_func is_globalZorigin_info_mapZ static_source static_nodet_nodes_nodeZ dygraph_infoZ static_infoZ static_locZexist_origin_inforrr!create_and_update_origin_info_maps@         rYcCst||}||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. )r(r9)Zast_noder*resolverrrrrMs rMc csNdd}||}||}|rt|t|ksJ|}|}t|t|kr._as_listrKN)r5r@rOr:rLoadParamr$_fieldsrPrDr<r[tuplezip) rTrVr]Ztransformed_node_listZstatic_node_listrWrXfieldZ t_node_childZ s_node_childZd_itemZs_itemrrrrNsF             rNcCst|tsJdd}tj}|}|jD](}t|jD] \}}||r<| |}||}z| ||WqYqqq|S)ze Replaces op callstack information about transformed static code with original dygraph code. c Sst|ddks Jtdt|dD]Z}||dd}|ddd}t|ddd}|dd d}||dd}t||}t|j}|rV| \}}}}d |d |d |||<d |||d<q|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(): r ,rCrNz File "z", line z, in z ) r5ranger%splitintr rRrQrr') callstackiZ file_liner r funcnamecoderIZdygraph_func_inforrrget_new_op_callstacks  zBupdate_op_callstack_with_origin_info..get_new_op_callstack) r:rrZop_proto_and_checker_makerZkOpCreationCallstackAttrNameblocks enumerateopsZhas_attrattrZ _set_attr)programrpZop_makerZcallstack_var_nameblockrmoprlrrr$update_op_callstack_with_origin_infos #    rx)T)r+collections.abcrZ paddle.baserZpaddle.base.frameworkrZ paddle.utilsrutilsrr__all__r rNodeTransformerr(rRrYrMrNrxrrrrs     (; 6 4