ϑiRMSSKJr SSKrSSKrSSKrSSKrSSKJr SSK J r SSK J r SSK JrJr SrS rS rS rS rS rSrSrSrSrSrSrSrSrSrSrSr Sr!"SS\ RD5r#Sr$Sr%Sr&Sr'S r(S!r)S"r*S#r+S$r,\ RZ"5S%4S&jr.S'r/S(r0S)r1"S*S+5r2"S,S-\ RD5r3g).) annotationsN) unique_name)ast_to_source_code)gast)PADDLE_MODULE_PREFIXis_api_in_module_helperget_argsset_args__argstrue_fnfalse_fn__for_loop_var_index__for_loop_iter_tuple__for_loop_iter_target__for_loop_iter_iterator__for_loop_iter_tuple_index__for_loop_var_len__for_loop_iter_var__for_loop_iter_zipwhile_condition while_bodyfor_loop_condition for_loop_bodycFURU5$![a gf=fN)index ValueError) array_listitems g/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/jit/dy2static/transformers/utils.py index_in_listr#3s*%% s   c$\rSrSrSrSrSrSrg)BaseNodeVisitor;z Implement customized NodeVisitor inherited from gast.NodeVisitor. Ancestor nodes are traced to easily support more operations of currently visited node. c/UlgNancestor_nodesselfs r"__init__BaseNodeVisitor.__init__Bs  cURRU5 SURR-n[ XUR 5nU"U5nURR 5 U$)z Visit a node.visit_)r*append __class____name__getattr generic_visitpop)r,nodemethodvisitorrets r"visitBaseNodeVisitor.visitEs\ ""4(DNN333$(:(:;dm ! r/r)N)r4 __module__ __qualname____firstlineno____doc__r-r<__static_attributes__r/r"r%r%;s !r/r%cXUSUS3n[R"U5RS$)Nz = _jst.UndefinedVar('z')r)rparsebody)name func_codes r"create_undefined_varrIPs/&.tfB7I ::i % %a ((r/c[U[5(deUSU3n[R"U5RS$)z) Create a assign stmt for name = value . z = r) isinstanceboolrrErF)rGvaluer8s r"create_bool_noderNUsA eT " "" "V3ug D ::d   ##r/c0n[R"U5H$n[R"U5HnX!U'M M& U$r()rwalkiter_child_nodes)root to_parentr8childs r"get_parent_mappingrU^s>*,I $**40E#e 1  r/c U(dgUVs/sH$nSRURSS55PM& nnSRSRU55$s snf)z Return "('x', 'y')" for [x, y] Nonez'{}''z\'z({}, ),)formatreplacejoin)name_idsrG names_strs r"create_name_strr_fsP EMNXTt||C78XIN ??388I. //Os+Ac [R"U5nU(a-UR[R"[ U5S95 O#UR[R"SS95 [R "UUU/SS/S9nU$)z\ Wrapper all statements of nodes into one ast.FunctionDef, which can be called by ast.Call. )rMN)rGargsrFdecorator_listreturns type_comment type_params)copyr2rReturngenerate_name_node FunctionDef)nodesrG input_argsreturn_name_ids func_def_nodes r"create_function_def_nodernqsr IIe E T[['9/'JKL T[[t,-$$   M r/cp[U[R"5S9n[R"U/USS9nX#4$)zL Creates a `gast.Assign` node by given name_id as target and node as value. )ctxN)targetsrMrd)rhrStoreAssign)rGr8rq assign_nodes r"create_assign_noderus:!4::<8G++ K  r/cSn[U[[45(de[U5nU(dU"5$U/:XaSnO[ US5nSnUR [ R"[5USRU5S9n[R"[R"U55RS$)zg Create get_args function as follows: def get_args_0(): nonlocal x, y return x, y cS[R"[5S3n[R"[ R "U55RS$)N def z(): return r)rgenerateGET_ARGS_FUNC_PREFIXrrEtextwrapdedentrFfunc_defs r" empty_node(create_get_args_node..empty_nodesJ  ! !"6 7 89  zz(//(3499!<.empty_nodesQ  ! !"6 7 8)E  zz(//(3499!<,create_nonlocal_stmt_nodes..3aSU;a[R"UTSSS9$[R"U5RSR$)Nr)idrp annotationrdr)rNamerErFrM)rGrps r"create_node_for_name0generate_name_node..create_node_for_namesF d?99ST zz$$$Q'---r/r)eltsrp) rKstrrrset TypeErrortypelenrTuple)r]rpgen_tuple_if_singlername_id gast_names name_nodes ` r"rhrhs(C  : huc 2 3 3B4>BR S  .@HHxG&w/xJH :!$7qM  JJJC8  IsB"c[U[R5(dS5e[U5R 5$)Nz3Input non-Attribute node to get attribute full name)rKr Attributerstripr8s r"get_attribute_full_namers9 dDNN + += + d # ) ) ++r/c[U[R5(dS5eURn[U[R5(a-URn[U[R5(aM-[ U5R 5nSSKnSSKJs J n SSKJ n [UUUS.n[X75n[X5$![a gf=f)Nz(Input non-Call node for is_api_in_moduler) to_tensor)nppaddle_jstrF)rKrCallfuncrrrpaddle.jit.dy2staticjit dy2staticrrevalr Exception) r8 module_prefix func_nodefunc_strrrrglobalsfns r"is_api_in_modulers dDII & &2 &  I Y * *NN  Y * *"),224H++$"  ( $&r99 s3C CCc"[U[5$r()rrrs r" is_paddle_apir3s D"6 77r/cV\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rg) NameScopei7c[5Ul[5Ul[5UlSUl[5Ul[5Ul[5Ulg)a A NameScope is a object which manager all the variable names. only FunctionDef and Controlflow node will have a namescope property. type can be "function" and "controlflow" we don't analyze the read only variable because they don't affect the analysis. N)rr nonlocalsrafatherw_varscreated push_pop_varsr+s r"r-NameScope.__init__8sDu E  e u !Ur/cXlgr()r)r,rs r" set_fatherNameScope.set_fatherKs r/cURUR- UR- UR- n[ [ SU55$)zGvars existing in current scope. they must not contain qualified names. c SU;$rrC)xs r"r(NameScope.existed_vars..SsCqLr/)rrrrarr)r, local_varss r" existed_varsNameScope.existed_varsNs;[[4<</$..@499L 60*=>>r/cUR$r()rr+s r" created_varsNameScope.created_varsUs ||r/cUR$r()rr+s r" modified_varsNameScope.modified_varsXs {{r/c /nURHbnURU5(a8URU5(a"[R"SUSUSUS35 MQUR U5 Md [ U5$)z At present, we do not support global append, such as import numpy as np a = [] def func(): a.append() # global names `a`, we will raise a warning. p.append(a, 1) # global names `np`, we will raise a warning. zFind variable `z$` defined in global scope and call `z .append() or zJ.pop()`, which will be ignored and never be transferred into tensor array.)r_is_simple_name is_global_varwarningswarnr2r)r,non_global_push_pop_namesvars r"variadic_length_varsNameScope.variadic_length_vars\s%'!%%C##C((T-?-?-D-D %cU+""%mC59%&*005&,--r/cpURnURRU-4nX RU- S.$)N)globalnonlocal)rr global_vars)r, valid_namestmps r"control_flow_varsNameScope.control_flow_varsss4kk {{&&46;;+<==r/cSU;dSU;agg)NrrFTrCr,rGs r"rNameScope._is_simple_namexs $;#+r/cURU5(dS5eUnUb>XR;agXRUR-;agURnUbM>g)z Return whether the name is a var created in global scope. Search from bottom to top. If it is not created or modified, it means global vars; otherwise, it means local vars. Only valid after FunctionNameLivenessAnalysis visitor. z5is_global_var accept a simple name, but get `{name}`.TF)rrrrr)r,rGancestors r"rNameScope.is_global_var}so##D)) C )"'''**X__<=H " r/c.URU5(+$r()rrs r" is_local_varNameScope.is_local_vars%%d+++r/c:U=RUR-slU=RUR-slU=RUR-slU=RUR-slU=RUR-slgr()rrrarr)r, name_scopes r" merge_fromNameScope.merge_fromsg ***  *... Z__$  z(((  j666r/)rarrrrrrN)r4r>r?r@r-rrrrrrrrrrrBrCr/r"rr7s9#&?..>  &,7r/rc\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrSrSrSrSrSrSrSrSrSrSrg)FunctionNameLivenessAnalysisiaanalyze the liveness of a function. every variables stored in this scope will be collected, in addition with global/nonlocal information and push_pop information. 1. global variable is stored in node.var_globals. 2. nonlocal variable is stored in node.var_nonlocals. 3. arguments is stored in node.var_args. 4. if a variable's push and pop attribute is called, it will be collected in push_pop_vars. They are used for transformation to tensor_array. NOTE: push_pop_vars **may not** in w_vars. a.push(0) don't modify the variable a, but the content of a. For example: def func(*args, **kargs): a = 12 global i,j nonlocal x,y print(a) i = k b = [] c = [1,2,3] for m in range(10): q = 12 b.push(1) c.pop() After this visitor we have: # node is the FunctionDef node with name: "func" node.pd_scope = NameScope( globals = ['i', 'j'], nonlocals = ['x', 'y'], args = ['args', 'kargs'], wr_vars = ['a', 'i', 'q', 'm', 'c', 'b'] push_pop_vars = ['b', 'c'] ) c4/UlURU5 gr()scope_node_stackr<)r, root_nodes r"r-%FunctionNameLivenessAnalysis.__init__s " 9r/c"[5Ulgr()rpd_scoper,r8s r"_reset_name_scope.FunctionNameLivenessAnalysis._reset_name_scopes !  r/cZ[US5(d[5UlUR$)Nr )hasattrrr r s r"_get_name_scope,FunctionNameLivenessAnalysis._get_name_scopes"tZ((%KDM}}r/c>URURS5$r)rrr+s r"_current_name_scope0FunctionNameLivenessAnalysis._current_name_scopes##D$9$9"$=>>r/cr[UR5S:XagURURS5$)Nr)rrrr+s r"_father_name_scope/FunctionNameLivenessAnalysis._father_name_scopes4 t$$ % *##D$9$9"$=>>r/c[UR5S:XagURSSS2H5n[U[R5(dM$UR U5s $ g)Nrrr)rrrKrrirr s r"_nearest_function_scope4FunctionNameLivenessAnalysis._nearest_function_scopesW t$$ % *))"&b&1D$ 0 011++D112r/cg)z[ i for i in range(10) ] In this case, `i` will not created in FunctionScope. We don't collect `i` by not calling generic_visit. NrCr s r"visit_ListComp+FunctionNameLivenessAnalysis.visit_ListComps r/cg)zthe same as ListComp.NrCr s r"visit_DictComp+FunctionNameLivenessAnalysis.visit_DictComps r/c"URU5 [R[R[R4n[ UR U5(a4UR5RRUR5 ggr() r6rrrAugStoreDelrKrprraddrr,r8 write_contexts r" visit_Name'FunctionNameLivenessAnalysis.visit_Names_ 4 T]]DHH= dhh . .  $ $ & - - 1 1$'' : /r/cH^^UU4SjnUU4SjnTRTX#5 g)Nc|>TR5=R[TRT55-slgr()rrar_get_argument_namesr8r,sr"pre_func@FunctionNameLivenessAnalysis.visit_FunctionDef..pre_funcs2  $ $ & + +s((.0  +r/c>^[[[[[[/mUU4SjnTR 5(aU"5(awTR 5=R TR5R -slTR 5=RTR5R-slggg)zNOTE: why we need merge w_vars and push_pop_vars here ? because we do ifelse_transformer after loop_transformer. Loops will changed into functions. but we know this function will be called in if. so we add w_vars to father function scope. c\>TH%nTRRU5(dM% g g)NTF)rG startswith)prefixcontrol_flow_function_defr8s r"is_control_flow_def_nodecFunctionNameLivenessAnalysis.visit_FunctionDef..post_func..is_control_flow_def_nodes*7Fyy++F33#8r/N) WHILE_BODY_PREFIXFOR_CONDITION_PREFIXFOR_BODY_PREFIXTRUE_FUNC_PREFIXFALSE_FUNC_PREFIXrrrr)r7r6r8r,s @r" post_funcAFunctionNameLivenessAnalysis.visit_FunctionDef..post_funcs "!$ ! ) %  &&((-E-G-G'')00,,.550'')77,,.<<7 .H(r/_visit_scope_noder,r8r0r>s`` r"visit_FunctionDef.FunctionNameLivenessAnalysis.visit_FunctionDefs   6 tX9r/cDURU5 URRU5 UR5R UR 55 U(aU"5 UR U5 U(aU"5 URR5 g)zAscope node main visit logic. pre_func and post_func is callbacks N)rrr2rrrr6r7rBs r"rA.FunctionNameLivenessAnalysis._visit_scope_nodesv t$ $$T*   "--d.J.J.LM  J 4  K !!#r/cH^^UU4SjnUU4SjnTRTX25 g)Nc>TR5RTR55 TR5RTR55 TR5R 5TR - TR5lTR5=R TR5R -slgr()rrrrrbefore_createdrr/sr"r>GFunctionNameLivenessAnalysis._visit_controlflow_node..post_func%s  # # % 0 01I1I1K L  ( ( * 5 5((* ,,.;;=%%&  $ $ & .  ( ( * 2 2((*22  2r/cL>TR5R5Tlgr()rrrIr/sr"r0FFunctionNameLivenessAnalysis._visit_controlflow_node..pre_func3s"&">">"@"M"M"OD r/r@)r,r8r>r0s`` r"_visit_controlflow_node4FunctionNameLivenessAnalysis._visit_controlflow_node$s   P tX9r/c&URU5 gr(rMr s r" visit_For&FunctionNameLivenessAnalysis.visit_For8 $$T*r/c&URU5 gr(rPr s r" visit_While(FunctionNameLivenessAnalysis.visit_While;rSr/c&URU5 gr(rPr s r"visit_If%FunctionNameLivenessAnalysis.visit_If>rSr/cpUR5=R[UR5-slgr()rrrrr s r" visit_Global)FunctionNameLivenessAnalysis.visit_GlobalAs$   "**c$**o=*r/cpUR5=R[UR5-slgr()rrrrr s r"visit_Nonlocal+FunctionNameLivenessAnalysis.visit_NonlocalDs$   ",,DJJ?,r/c@URU5 [R[R[R4n[ UR U5(aC[U5R5nUR5RRU5 ggr() r6rrrr%r&rKrprrrrr')r,r8r)rGs r"visit_Attribute,FunctionNameLivenessAnalysis.visit_AttributeGsn 4 T]]DHH= dhh . .%d+113D  $ $ & - - 1 1$ 7 /r/cJURU5 [R[R[R4n[ UR U5(a[ UR[R5(a7URn[ UR[R5(aM7[ UR[R5(a>UR5RRURR5 gggr()r6rrrr%r&rKrprM Subscriptrrrr'rr(s r"visit_Subscript,FunctionNameLivenessAnalysis.visit_SubscriptNs 4 T]]DHH= dhh . .TZZ88zzTZZ88$**dii00((*1155djjmmD1 /r/cdURU5 [UR[R5(dgSS/nURR U;ag[ URR5R5nUR5RRU5 g)Nr2r7) r6rKrrrattrrrMrrrr')r,r8variadic_length_methodrGs r" visit_Call'FunctionNameLivenessAnalysis.visit_CallWs 4 $))T^^44 "*E!2 99>>!7 7 !$))//288:   "0044T:r/cn[U[R5(dS5e[URR5nUR URR 5 UR URR5 UVs/sHo3cMURPM nnU$s snf)zmget all arguments name in the functiondef node. this node is local to the function and shouldn't be created. z&Input node is not function define node) rKrrirrar2varargkwargr)r,r8ris r"r.0FunctionNameLivenessAnalysis._get_argument_namesbs $ 0 011 4 1TYY^^$ TYY%%& TYY__%$6u!u6 7s B2B2)rN)r4r>r?r@rAr-rrrrrrr"r*rCrArMrQrUrXr[r^rarerjr.rBrCr/r"rrst(T$ ?? 2  ; !:F $:(+++>@8E ; r/r)4 __future__rrfr{rnumpyr paddle.baserpaddle.jit.dy2static.ast_utilsr paddle.utilsrutilsrr rzrrr<r=FOR_ITER_INDEX_PREFIXFOR_ITER_TUPLE_PREFIXFOR_ITER_TARGET_PREFIXFOR_ITER_ITERATOR_PREFIXFOR_ITER_TUPLE_INDEX_PREFIXFOR_ITER_VAR_LEN_PREFIXFOR_ITER_VAR_NAME_PREFIXFOR_ITER_ZIP_TO_LIST_PREFIXWHILE_CONDITION_PREFIXr9r:r;r# NodeVisitorr%rIrNrUr_rnrurrrLoadrhrrrrrrCr/r"rs# #=A!!  ./15;.03* +!d&&*) $0.  "9J#9L +&*YY[e@,<8a7a7HS4#3#3Sr/