x-jUL<ddlZddlmZddlZddlmZddlmZmZddl m Z  ddlm Z nQ#e $rIddl Z ejedzZe jejed ddlm Z YnwxYwGd d ZGd d ZdifdZdS)N)path) framework_pb2)core unique_name) OpProtoHolder) pass_desc_pb2z /../../baseprotocBeZdZgZdifdZdZdZdZdZdZ dZ d S) RegisterPassHelpercn||_||_||_tj|dSN) _pass_type _pass_pairs _input_specsr _register_helpersappend)self pass_pairs pass_type input_specss Y/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/incubate/passes/ir.py__init__zRegisterPassHelper.__init__%s6#%',33D99999cg}tj|}|jD]}|j|}t |t jjr:| t ||j |j ut |t jr(| t j|| t |dg|S)N)inspectgetfullargspecargsrget isinstancepaddlestatic InputSpecrPassDesc VarHelpershapedtype ParamAttr)rfuncr arg_specsarg_name input_specs r_get_args_from_funcz&RegisterPassHelper._get_args_from_func+s*400 ! @ @H*..x88J*fm&=>> @ && *"2J4D J(899 @ F,X667777 H..x">>???? rc|D]}tj|j}g}|jD]}|jdvrh|}t|dkrA|dd}||j}||krs| ||D]}|j |dS)N) op_namescope op_callstack op_devicerr) rget_op_attrs_default_valuetypeencodeattrsname ListFieldslenr!rremove) ropsop_desc default_attrs remove_attrsattrattr_list_fields attr_valuedefault_attr_values r_prune_program_descz&RegisterPassHelper._prune_program_desc<s + +G ; ##%%ML  * *9% (,'8'8$+,,11%)__%6%6r%:2%> -:->->ty-I-I*-;;$##D))))$ + + $$T**** ++ + +rc>g}tj}tj}tj||5||}||||}t |ttfs|g}|D]}t |tj rl| } t| dkrtd|jd| D]} || || dddn #1swxYwY|j} t'| D]N} || | O||||jfS)N Operator 'z;' has multiple outputs, please specify one output variable.)r#r$Program program_guardr/extendr"listtupler&OpHelperOutputsr: ValueError_typevaluesr current_blockdescrangeop_sizeaddParseFromStringopserialize_to_stringrDr<) rr+r<varsprogramstartup_programr outsoutop_outsop_out block_descis r_func_to_program_descz(RegisterPassHelper._func_to_program_descUs<-'')) -//11 ] ( (/ B B % %++D11D KK   4;DdT5M22 v % %c8#455 %!kkmmG7||q(((oooo#*.."2"2,, F++++,KK$$$$ %  % % % % % % % % % % % % % % %"**,,1 z))++,, N NA GGII % %jmmA&6&6&J&J&L&L M M M M   %%%W**,,000sC/EE"Ec fd t||D]1\}}|j}|j|_|j|_|j}|j|jvr|}|d |j d|j _tj j|j _|j j|jt&jjj|_t(j|jkr)|jD]}  || g3dS)Ncv|D]4}|jr||j||j5dSr) _conditionr _elements) conditionselementselement_add_element_conditionss rrkzNRegisterPassHelper._convert_vars_to_pass_desc.._add_element_conditionsqs[# G G%:%%g&8999'' G4EFFFF G Grr(r )zipvar_mapsrVr8 pattern_var replace_varvar_attr_conditionsrAttr_to_pass_desc_attrr@condition_valuerAttrTypeLONGSr5longsrJr(r r& ConditionTypekEQr' __class___attrsrQ) rpatternsreplacesrSpatternreplacevar_maprh conditionr@rks @r_convert_vars_to_pass_descz-RegisterPassHelper._convert_vars_to_pass_descpsP G G G G G !$Hh 7 7 @ @ GWm''))G"),G "),G 1J|t000&NN,,  W%%88HHH13 ).1>1G1M ).)/66w}EEE!.!7!E!I !W%666#N1133@@D++J????# @ @rc|D]}t|tjr|jD]}|j}t j|r ||}|j }| |j | |j |j |j|j dSr)r"r&rMrzrQ_mappedr isfunction op_attr_mapsrVrr pattern_attr replace_attr _operation operationCopyFrom)rr{r|rSr~r@mappedattr_maps r_convert_ops_to_pass_descz,RegisterPassHelper._convert_ops_to_pass_descs G GG'8#455 G#N1133 G GD!\F)&112!'!1!1#04466H--h.CDDD++H,ABBB(4 *33F4EFFF G Grctj}|rtjtj}|j|_|jD]\}}|j }| ||j \}}| ||j \}} | |||||| ||rtj|Sr)r#in_dynamic_mode enable_staticr MultiPassDescrrr pass_descsrVrcr}r~rrdisable_staticSerializeToString) rswitch_static_modemulti_pass_descr}r~ pass_desc pattern_vars pattern_ops replace_vars replace_opss rSerializeMultiPassDescz)RegisterPassHelper.SerializeMultiPassDescs#355  #  " " "'577$(O!!% 0 P P GW'26688I(,(B(B*)) %L+)-(B(B*)) %L+  + +lI     * *; Y O O O O  $  ! # # #00222rN) __name__ __module__ __qualname__rrr/rDrcrrrrrr r "s-/R:::: "+++21116@@@4 G G G33333rr ceZdZGddZGddejjZGddZeZ dS)r&c^eZdZddZdZdZdZddZdZdZ d Z d Z d Z d Z ddZdS)PassDesc.AttrHelperNcv||_||_d|_||_g|_d|_d|_d|_dSr)_obj_name_operation_type_element_indexrgrrfr)robjr8 element_indexs rrzPassDesc.AttrHelper.__init__s@DIDJ#'D "/D DN"DO"DODLLLrct|j|j|}|j||S)Nr)r& AttrHelperrrrgr)rindexrjs r __getitem__zPassDesc.AttrHelper.__getitem__sD)) 4:U*G N ! !' * * *Nrc^t|jtjr-tjjj|_|jj|_ n,tjjj |_|jj |_ |j |_|j |j|_|j|j|_dSdSr)r"rr&r'r RoleType kVariableroler8var_name kOperator_indexop_indexrrrrr)rpass_desc_attrs rrrz&PassDesc.AttrHelper._to_pass_desc_attrs$)X%788 ;&3&<&E&O#*.).''&3&<&E&O#*.)*:'"&*N #/+/+?("./3/B,,,/.rcd|_t|trtjj|_||_dStd)Nr z"Unimplemented transform operation.) r8r"intrrtINTr5rbNotImplementedError)rvalue op_desc_attrs r_to_op_desc_attrz$PassDesc.AttrHelper._to_op_desc_attrsE "L %%% P$1$:$> !!& )*NOOOrct|j|j|j}|j|| ||_|Stj }||_ t|tjr| |j n|||j||_|j|_|Sr)r&rrrrrgrrr Operationr5r"rrr@rrr)rr5rr@rs r_clone_with_operationz)PassDesc.AttrHelper._clone_with_operations&& 4:t':D N ! !$ ' ' '}'+$ %.88::I!IN%!455 >((8888%%eY_==='DO#'#7D KrcV|tjjj|Sr)rr r& OperationTypekSubrrs r__sub__zPassDesc.AttrHelper.__sub__(--&495 rcV|tjjj|Sr)rr r&rkAddrs r__add__zPassDesc.AttrHelper.__add__rrcV|tjjj|Sr)rr r&rkModrs rModzPassDesc.AttrHelper.ModrrcT|tjjjSr)rr r&rkSizers rSizezPassDesc.AttrHelper.Sizes&--&4: rc~tj}||j||_t |tjr||jn| ||j |j r|j |j ||_dSr)r r& AttrConditionrrr@r5r"rcondition_attrrrsrrrrf)rr5rrs r_set_with_conditionz'PassDesc.AttrHelper._set_with_conditions%.<<>>I  # #IN 3 3 3!IN%!455 H(()ABBBB%%eY-FGGG >#,,T_==='DOOOrcZ|tjjj|dSr)rr r&rwrxrs rEQzPassDesc.AttrHelper.EQ s2  $ $&48%     rrczt|grtdd}fd}||n||_dS)Nz#Only mapped one of which var or op.c td)Nz'Mapping to variable is not implemented.)r)rs r mapped_varz5PassDesc.AttrHelper.MappedPattern..mapped_vars)=rcfd|D}t|krtdddt|S)Nc*g|]}|jk |SrrP).0orXs r zHPassDesc.AttrHelper.MappedPattern..mapped_op..s ???QBqrzIndex '' of operator 'z' is incorrect.r)r:rOr&r)rr<rrr8rXs r mapped_opz4PassDesc.AttrHelper.MappedPattern..mapped_ops????+???s88u$$$K%KKKKK **JM+r)allrOr)rvarrXrr8rrrs ```` r MappedPatternz!PassDesc.AttrHelper.MappedPatternsvC9~~ H !FGGG            ), 99DLLLrr)NNrNN)rrrrrrrrrrrrrrrrrrrrrs        C C C P P P    $                ( ( (    HL D D D D D Drrc eZdZdZdZdZdS)PassDesc.VarHelperctj}tjj|i||_i|_dSr)r#r$default_main_programrRdata_varrz)rr kwargsblocks rrzPassDesc.VarHelper.__init__)sEM6688FFHHE *D;F;;DIDKKKrc,t|j|Sr)getattrr)rr8s r __getattr__zPassDesc.VarHelper.__getattr__.s49d++ +rc|j|}|%t||}||j|<|Srrzr!r&rrr8r@s rrqzPassDesc.VarHelper.Attr1@;??4((D|**466$( D!KrN)rrrrrrqrrrr'r(sA     , , ,     rr'cNeZdZd dZddZdZdZdZdZd Z d Z d Z d Z dS)PassDesc.OpHelperTc t|tsJdt|d}|dd|jDz }|dz }d}|dd|jDz }|dz }d}|dd|jDz }|dz }|d|jd |d|d }|S) NzCskip_op_callstack parameter's type is error, expect bool, received {z, c"g|] \}}|d| S=rrkvs rrz7PassDesc.OpHelper._to_readable_code..?s&>>>1A>>>r}c"g|] \}}|d| Srrrs rrz7PassDesc.OpHelper._to_readable_code..Es&===1A===rc"g|] \}}|d| Srrrs rrz7PassDesc.OpHelper._to_readable_code..Js&#O#O#O41aqJJ1JJ#O#O#Orz = z(inputs=)) r"boolr5join_outputsitems_inputsrzrP)rskip_op_callstack outputs_str inputs_str attrs_strop_strs r_to_readable_codez#PassDesc.OpHelper._to_readable_code9sA/66  oVZ[lVmVmoo  6K 499>> (;(;(=(=>>> K 3 KJ $))== (:(:(<(<=== J # JI #O#O4;;L;L;N;N#O#O#OPP PI  I#VV VVJVV)VVVFMrNc||_dSrr)rr5s rrzPassDesc.OpHelper.__init__Ps DJJJrcbt|}||Sr)r&rMInit)rr8rXs rrzPassDesc.OpHelper.__getattr__Ss'""4((B GGIIIIrcpt|dkrtd|D]P\}}|j|}|td|jd|dt |ttfr/t|dkrtd|d|jdn|g}|D]}t |tj rl| }t|d krtd |jd | D]}| ||||j|d |DRt"j} |jD]c\} } t-j|j} | | | |j| | gd|S)Nrz6Each input argument needs to specify a parameter name.rGz' does not have input named ''.zInput 'rz' cannot be empty.rFz!The size of outputs of operator 'z5' is not equal 1, please specify one output variable.cg|] }|j Srr8rrbs rrz.PassDesc.OpHelper.__call__..us.H.H.H!qv.H.H.Hrr)r:rOrrr!rPr"rKrLr&rMrNrQrJr_desc set_inputr#r$rrRrrgenerate create_var set_output) rr rin_namein_argsop_inputin_argr_r`rout_name op_outputop_output_names r__call__zPassDesc.OpHelper.__call__Xsq4yy1}} L%+LLNN J J <++G44#$YTZYYgYYYge}55(7||q(((\g\\dj\\\) 'iG% 0 0F!&(*;<< 0"(.."2"2w<<1,,",!HFL!H!H!H##'.nn&6&644F$OOF33334!//// $$W.H.Hx.H.H.HIIIIM6688FFHHE'+}':':'<'< B B#)!,!5dj!A!A  !1!1~!1!F!FGGG %%h0@AAAAKrchtj}t jj|j|_ |j td|jdt|j |_ |j|_|j|ji|_d|j jD|_d|j jD|_|j |dS)Nz)type object 'OpHelper' has no attribute ''ci|] }|jg Srrrs r z*PassDesc.OpHelper.Init..sCCC1AFBCCCrci|] }|jg Srr)rrs rr'z*PassDesc.OpHelper.Init..sEEEAQVREEEr)r#r$rrRrinstance op_proto_mapr!rP_protoAttributeErrorr:r<rrS append_oprset_typerzinputsroutputsrr)rrs rrzPassDesc.OpHelper.Init}sM6688FFHHE'022?CCDJOODK{"$M MMMei..DK--//DJ J   + + +DKCC 0BCCCDLEE1DEEEDM I  T " " " " "rc|j|}|%t||}||j|<|Srrrs rrqzPassDesc.OpHelper.Attrrrct|tjr|||_dS|j||dSr)r"r&rrqrr _set_attr)rr8rs rSetAttrzPassDesc.OpHelper.SetAttrsM%!455 2*/ $''' $$T511111rct|j|}|td|jd|d|S)NrGz' does not have output named 'r)rr!rOrP)rr8outputs rOutputzPassDesc.OpHelper.OutputsM]&&t,,F~ SSS4SSSMrc|jSr)rrs rrNzPassDesc.OpHelper.Outputss = rc |D]C\}}||j|"|j||jgDdSr)rr remove_outputrr8)rrparamargs r SetOutputszPassDesc.OpHelper.SetOutputssd$llnn = = s;J,,U3333J))%#(<<<<  = =r)Tr) rrrrrrr#rrqr4r7rNr=rrrrMr8s    .        # # # J # # #    2 2 2     ! ! ! = = = = =rrMN) rrrrr#r$Variabler'rMOPrrrr&r&ssDsDsDsDsDsDsDsDjFM* q=q=q=q=q=q=q=q=f BBBrr&cZdfd}tj|r ||S|S)a The function decorator of Register Pass. Decorator @RegisterPass handles the function and register it into a core.Pass instance. Use name of function as Pass type. Args: function (callable): The function with return of callable pair(s) that represents the pattern subgraph and the replace subgraph. input_specs (dict[str, InputSpec]): Dict of InputSpec to specific the shape/dtype information of Tensor. Some operators limit the shape and dtype of datas when create subgraph with Paddle APIs. So user need specify InputSpec of data to ensure create a correctly subgraph. Of course, this argument is not limited to matching subgraph. The default is dict(). Returns: callables: Callable pair(s). Examples: .. code-block:: python >>> import paddle >>> from paddle.incubate.passes.ir import RegisterPass >>> @RegisterPass >>> def multi_add_to_addn(): ... def pattern(x, y, z): ... return paddle.add(paddle.add(x, y), z) ... def replace(x, y, z): ... return paddle.add_n([x, y, z]) ... return pattern, replace ct|ttfr._is_pass_pairsP j4- 0 0 :!##s7-z::;; 4urc|j}tj|}t|jdkrt dt|jdkrp|}|r|g}n,t t|stdt||}tj ||j |S)Nrz2Pass function with parameter is not supported now.z;Return value of Pass function must be (callable, callable).) rr signaturer: parametersrrrCrOr r register_passr) python_funcrrGrhelperrErs r decoratedzRegisterPass..decorateds( %k22 y# $ $q ( (%D % & &! + +$J}Z(( (\ ]J7788  Q( I{KKF  y&*G H H Hr)rr)functionrrLrEs ` @r RegisterPassrNs]B&(###y""" r)rosrr#paddle.base.protorbaserrbase.frameworkrr ModuleNotFoundErrorsysdirname__file__ base_pathrrr r&rNrrrrXs| ++++++%%%%%%%%++++++0///////000JJJ X&&6IHOOIDIi11222//////// 0M3M3M3M3M3M3M3M3`yyyyyyyyxB>>>>>>s+A A98A9