|jc#UddlmZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z m Z mZmZddlmZddlZddlmZejejzZded <ed Zed ZGd d Zd*dZd+dZd,dZdZdZdZ eddddddd-d$Z!e d.ddddddd/d'Z! d.ddddddd0d)Z!dS)1) annotationsN)MappingSequence)partialwraps)AnyCallableTypeVaroverload) ParamSpec)_C_opsintHAS_VAR_ARGS_OR_KWARGSP1R1ceZdZddZdZdS) MissingArgumentfnCallable[P1, R1]namestrc"||_||_dSN)rr)selfrrs g/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/static/python_op.py__init__zMissingArgument.__init__%s c2d|jd|jjdS)Nz)rr__name__)rs r__repr__zMissingArgument.__repr__)s"TtyTTAQTTTTrN)rrrr)r __module__ __qualname__rr!rrrr$s<UUUUUrrrr parameterinspect.Parameterc|jtjjurdS|jtjjuriS|jtjjurt||jS|jS)Nr$) kindinspect ParameterVAR_POSITIONAL VAR_KEYWORDdefaultemptyrr)rr%s rextract_defaultr/-sa~*999r 7,8 8 8  g/5 5 5r9>222  rreturntuplecfdtjjD}t |D]&\}}t |t s ||d}n't|S)Nc0g|]}t|Sr$)r/).0paramrs r z*get_fn_defaults_params..8s3  E""r)r) signature parametersvalues enumerate isinstancerr1)rfn_defaults_paramsir-s` rget_fn_defaults_paramsr>7s&r**5<<>> 233 7'?33 !3ABB!7  E  # $ $$rct|tjs Jd|j}|jt z}|j|jzt|jtj zzt|jtj zz}tj dkrltj|dd|j|j||j|j|j|j|j|j|j|j|j|j|j|j}nvtj|dd|j|j||j|j|j|j|j|j|jtj dkr|jn|j|j|j}t=|}tj||j|j ||j!}|j |_ |j"|_"|j#|_#d|_$|S)NzOnly support regular function) r)r@ )%r;types FunctionType__code__co_flagsr co_argcountco_kwonlyargcountboolr) CO_VARARGSCO_VARKEYWORDSsys version_infoCodeType co_nlocals co_stacksizeco_code co_constsco_names co_varnames co_filenameco_name co_qualnameco_firstlineno co_linetableco_exceptiontable co_freevars co_cellvars co_lnotabr> __globals__r __closure____doc____annotations____kwdefaults__)rcoderFargcountnew_coder<new_fns r$eliminate_positional_or_keyword_onlyrgCs b%, - -NN/NNN - ;DM%;$;;H    ! t}w11 2 2 3 t}w55 6 6 7  7"">  O    L N M     L        "    %  *>  O    L N M     L  7**       %  *033      FkFOZFN/F F Mrctj|}|j|i|}||j}i}i}|D]/\} } t | tjj r| || <*| || <0t| } t|}t|}| t|fi|t|fi|t||fSr)r)r7bindapply_defaults argumentsitemsr;paddlepirValuelistkeysrgrr9) r infer_metaargskwargssig bound_argsparamsmutable_params const_paramskvmutable_arg_namess rbind_constantsr}s  B  C4*6**J  !FNL   1 a) * * !N1  LOO^002233 -b 1 1B5jAAJ##l## ++l++ ^ " " $ $%%  rcfd}|S)Nctjjj5|i|cdddS#1swxYwYdSr)rmbasedygraphguard)rsrtrs rdynamic_mode_fnz,run_in_dynamic_mode..dynamic_mode_fns [ % + + - - ' '2t&v&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's?AAr$)rrs` rrun_in_dynamic_moders$''''' rct|ttttt frt |St|ttfr t |S#t$rYnwxYwt|ttfr; t dgtd|DRS#t$rYnwxYwt|tr\ ttd|D}t dg|RS#t$rYnwxYw t |S#t$rt|cYSwxYw)Nrc34K|]}t|VdSr custom_hash)r4items r zcustom_hash..s*"E"E;t#4#4"E"E"E"E"E"Erc3XK|]%\}}t|t|fV&dSrr)r4rzr{s rrzcustom_hash..s7PPDAq A A7PPPPPPr)r;rfloatrrIbyteshashr1 frozenset TypeErrorrsetrsortedrlid)obj items_hasheds rrrs#UCu566Cyy#y)** 99     D ##'' FU"E"E"E"E"EEEFFGG G    D #w  PPCIIKKPPPPPL*\**++ +    D Cyy #wwsIA** A76A7)C C C'A D22 D?>D?EE.-E.)rrr input_names output_names inplace_mapr str | NonerrCallable[..., Any] | Nonerlist[str] | Nonerrdict[str, str] | NonecdSrr$rrrrrrrs r register_oprs srNone.Callable[[Callable[P1, R1]], Callable[P1, R1]]cdSrr$rs rrrs 69SrCallable[P1, R1] | Nonectdtdtddfd }||S||S) Nz(Currently, input_names must be provided.z)Currently, output_names must be provided.z'Currently, infer_meta must be provided.real_fnrr0c pjtjjjt dfd }|S) NrsP1.argsrt P1.kwargsr0rc tjr|i|St g|Ri|\}}}}}t|t ks+Jdt|dt dt |}t j| d| |t|d pid}t dkr|dn|S) NzNumber of mutable arguments (z,) does not match the number of input names (z)._)infer_meta_fn_ptrfn_ptr)rrrattrsrrr)rmin_dynamic_moder}lenrr _run_python_opr)rsrtr|bound_constants_fnbound_constants_infer_meta mutable_argsryconst_params_hashoutrrrrop_namerrs r wrapped_fnz5register_op.._register_op..wrapped_fnsI%'' 0w////w DTDDDVDD  !"*())S-=-====C4E0F0FCC.1+.>.>CCC>== !,L 9 9 '55"355'))C12DEE(-2   C!..!333q66 ._register_opsw*'*   " w = = = = = = = = = =  # " =@r)rrr0r) ValueError)rrrrrrrrs ````` rrrsCDDDDEEEBCCC''''''''''T z <  r)rrr%r&)rrr0r1)rrr0r)rrrrrrrrrrrrrr0rr)rrrrrrrrrrrrrr0r) rrrrrrrrrrrrr)" __future__rr)rLrCcollections.abcrr functoolsrrtypingrr r r typing_extensionsr rmr rJrKrrarrrr/r>rgr}rrrr$rrrsQ#"""""" --------$$$$$$$$333333333333'''''' %073IIIIIIYt__ WT]]UUUUUUUU % % % %DDDDN6%%%P ,0$(%))-         9,0$(%))- 9 9 9 9 9  9#'>,0$(%))->>>>>>>>r