ϑinSSKJr SSKrSSKrSSKJr SSKrSSKJr SSK J r SSK J r J r SSKJrJrJr SSKJr SS KJr SS KJr SS KJr SS KJrJr S SKJr S SKJ r J!r!J"r"J#r#J$r$ /r%SS)r?S*r@S+rAS,rB"S-S.5rCS/rDS0rES1rFS2rGS3rHS>S4jrIS5S6SS7S8.S9jrJ\S?S:j5rKS@S;jrLg)A) annotationsN)contextmanager) PyLayerMeta) convert_dtype)_convert_into_variablein_to_static_mode)Variablecoredefault_main_program) use_pir_api) OrderedSet)Value) AmpOptions) is_sequence map_structure) StaticPyLayer)RETURN_NO_VALUE_VAR_NAMEDygraph2StaticExceptionGetterSetterHelper UndefinedVarcreate_undefined_variablec[U[5(aUcSOUn[R"/XS9$[U[5(aUcSOUn[R"/XS9$[U[ 5(aUcSOUn[R"/XS9$[ 5(d#[U[5(dUc [5$[U5(a[[U5$U$)z? Translate a Python Tensor to PaddlePaddle static graph Tensor bool)shapedtype fill_valuefloat64int64) isinstancerpaddlefullfloatintr rrrrto_static_variable)xrs f/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/jit/dy2static/convert_operators.pyr%r%,s!T-U{{5??!U"] {{5??!S =e{{5?? ==jL99QY )**1~~/33 Hc[U[5(a5[U[R5(dUS:XaUR 5$[ X5$)Nsize)r r r!Tensorr*getattr)r&attrs r' convert_attrr.Cs? 1h1fmm,, FNvvxqr(c[U[5(a [U5$[5(Ga/[U[R 5(a [ U5$[U[5(a9SSKJ n UR[RR5U5nUbU$[U[5(ao[5R5nSSKJn UR#5R$RUR&UR(R+55nUbU$U[R,R.La[15(d[2$U$)Nr_global_inplace_mapProgramTranslator)r rrrr!r+rr'paddle.jit.dy2static.parameter_recorderr1getstaticr r current_block'paddle.jit.dy2static.program_translatorr3 get_instance _inplace_mapprogramdescidamp auto_castr convert_auto_cast)r&r1new_var cur_blockr3s r' convert_loadrCRs![!!Q a ' ' *!, , a   *-- 224aG" a " ",.<<>I (446CCGG!!16699;G"  $$ $[]]$ $ Hr(c[U[[45(aU$[US5(a [ U5$[US5(a[US5(aU$[ S5e)N__iter____len__ __getitem__z"X can't be convert into indexable.)r r rhasattrlist RuntimeError)r&codes r' indexablerL}s`!h&'' J  Aw I  7 =$$?@@r(cd[U[[45(a [X5$[ X5$)z/unified unpack interface for paddle and python.)r r r_unpack_by_structure_paddle_unpack_by_structure_pythontarget structures r'unpack_by_structurerSs*&8U+,,*6==*6==r(c[X5$)zJTODO(xiongkun): analysis the differences between python and paddle unpack.)rNrPs r'rOrOs &v 99r(cUS:XaU$/n[U5H]up4US:XaURX5 M [U[5(aUR[ XU55 MT[ S5e U$)Nrz#structure element must be 1 or list) enumerateappendr rIrSAssertionError)rQrRretidxeles r'rNrNsuA~ Ci( !8 JJv{ #  c4 JJ*6;< = BCC) Jr(c|U"5n[U[[45(a[XX#XE5 g[ XX#5 g)a A function representation of a Python ``while`` statement. Args: cond(Callable): A callable object that returns a boolean variable to control whether to execute the loop body. It takes ``loop_vars`` as arguments. body(Callable): A callable object that returns a tuple or list of variables with the same arguments ``loops_vars`` as ``cond`` . get_args(callable): Get all arguments that needed in true_fn and false_fn. set_args(callable): Update arguments that modified in trure_fn and false_fn. return_name_ids(list[string], optional): the returned names. push_pop_names(list[string], optional): the names on which called .append() or .pop(). Returns: A list or tuple of variables which returned by ``body``. N)r r r_run_paddle_while _run_py_while)condbodygettersetterreturn_name_idspush_pop_namespreds r'convert_while_looprfs8( 6D$5)**   d&1r(c URU5nUcgSnURXVs/sH oC"U5PM sn5 gs snf)Nc[U[5(a\[5(a.[RR R ROSn[RRXS9$U$)Nfloat32)initialized_list) r rIr r!base libpaddleDataType UNDEFINEDtensor create_array)vrs r'maybe_to_tensor_arrayA_convert_tensor_array_if_necessary..maybe_to_tensor_arraysZ a  == %%..88  ==--e-H HHr()r5set) setterhelperrd push_pop_varsrrrqs r'"_convert_tensor_array_if_necessaryrwsM $$^4M =I=a.q1=IIsA c^^^ ^ [X#XE5m [T U5 U(a [U5O [5U(a [U5O [5-m [T 5m UU U 4SjnU4SjnT R T 5Vs/sH%n[ U[ 5(d [U5OUPM' n nT RT U 5 SSK J n U "XvU 5n T RT U 5 U $s snf)Nc\>UnTRTU5 T"5 TRT5$)z\wrap the body() and add return value for `while_loop` the args may be differ from getter(). )rtr5)argsmutable_loop_varsr`helper union_names r' new_body_fn&_run_paddle_while..new_body_fns.! :01 zz*%%r(c>T"5$)zIcond is a zero-args function, which is not compatible with `while_loop`. )rzr_s r' new_cond_fn&_run_paddle_while..new_cond_fns v r(r) while_loop) rrwr rIr5r rr%rtpaddle.static.nnr) r_r`rarbrcrdr~rvar loop_varsrr|r}s `` @@r'r]r]s  PF&v~>(7 ?#JL'5N #:<IJj!J&::j))C(2#|'D'D3#M) JJz9%+;Y?I JJz9% s,C%cU"5n[U[[45(a [S5eU(dgU"5 M>)Nz/python while pred change from bool to variable.)r r rr)r_r`rarbres r'r^r^s@ v dXu- . .)A    r(c^^U"5m[T[[45(d[U4SjU5$U"5m[T[[45(d[U4SjU4Sj5$[ TT5$)aa A function representation of a Python ``and`` statement. Args: x_func(callable): x_func() is the left hand operand of ``and`` operator. x_func() is bool or Tensor. y_func(callable): y_func() is the right hand operand of ``and`` operator. y_func() is bool or Tensor. Returns: A python bool variable or a bool Tensor. NOTE(liym27): 1) The operands are executed sequentially according to the running logic of Python. So here the arguments should be callable. 2) If the left hand operand is False, the right hand operand should be executed. For example: a = x > 1 and y < 1 Transformed code: a = paddle.jit.dy2static.convert_logical_and(lambda:x>1, lambda:y<1) In `convert_logical_and(lambda:x>1, lambda:y<1)`, `lambda:y<1` must be run after `lambda:x>1`. And if `x>1` is False, `y<1` should NOT be run. c>T$Nrx_valuesr'%convert_logical_and..'7r(c>T$rry_valuesr'rr+rr(c>T$rrrsr'rr+sGr()r r r_run_py_logical_and_run_paddle_logical_andx_funcy_funcrrs @@r'convert_logical_andr s^0hG g%0 1 1"?F;;hG g%0 1 1"?ODD "7G 44r(cZ[U5n[U5n[R"X5$r)cast_bool_if_necessaryr! logical_andr&ys r'rr0s'q!Aq!A   a ##r(cjU"5n[U[[45(aeU=(a U"5$rr r rrrrs r'rr6s0hG'He#4555 5  vxr(c^^U"5m[T[[45(d[U4SjU5$U"5m[T[[45(d[U4SjU4Sj5$[ TT5$)aW A function representation of a Python ``or`` statement. Args: x_func(callable): x_func() is the left hand operand of ``or`` operator. x_func() is bool or Tensor. y_func(callable): y_func() is the right hand operand of ``or`` operator. y_func() is bool or Tensor. Returns: A python bool variable or a bool Tensor. NOTE(liym27): 1) The operands are executed sequentially according to the running logic of Python. So here the arguments should be callable. 2) If the left hand operand is True, the right hand operand should be executed. For example: a = x > 1 or y < 1 Transformed code: a = paddle.jit.dy2static.convert_logical_or(lambda:x>1, lambda:y<1) In `convert_logical_or(lambda:x>1, lambda:y<1)`, `lambda:y<1` must be run after `lambda:x>1`. And if `x>1` is True, `y<1` should NOT be run. c>T$rrrsr'r$convert_logical_or..Z'r(c>T$rrrsr'rr^rr(c>T$rrrsr'rr^s7r()r r r_run_py_logical_or_run_paddle_logical_orrs @@r'convert_logical_orr@s^0hG g%0 1 1!/6::hG g%0 1 1!/?CC !'7 33r(cZ[U5n[U5n[R"X5$r)rr! logical_orrs r'rrcs'q!Aq!A   Q ""r(cjU"5n[U[[45(aeU=(d U"5$rrrs r'rris0hG'He#4555 5  fhr(cd[U[[45(a [U5$[ U5$)z A function representation of a Python ``not`` statement. Args: x(bool|Tensor): Operand of ``not`` operator. Returns: A python bool variable or a bool Tensor. )r r r_run_paddle_logical_not_run_py_logical_notr&s r'convert_logical_notrss,!h&''&q))"1%%r(cD[U5n[R"U5$r)rr! logical_notrs r'rrsq!A   a  r(cU(+$rrrs r'rrs 5Lr(c |[U[[45(a[UUUUUUU5nU$[ XX#XE5nU$)a A function representation of a Python ``if/else`` statement. Args: pred(bool|Tensor): A boolean Tensor which determines whether to return the result of ``true_fn`` or ``false_fn`` . true_fn(callable): A callable to be performed if ``pred`` is true. false_fn(callable): A callable to be performed if ``pred`` is false. get_args(callable): Get all arguments that needed in true_fn and false_fn. set_args(callable): Update arguments that modified in trure_fn and false_fn. return_name_ids(list[string], optional): the returned names. push_pop_names(list[string], optional): the names on which called .append() or .pop(). Returns: ``true_fn()`` if the predicate ``pred`` is true else ``false_fn()`` . )r r r_run_paddle_cond_run_py_ifelse)retrue_fnfalse_fnget_argsset_argsrcrdouts r'convert_ifelsersV2$5)**         J  8x  Jr(c8^^^^ ^^^[X4TU5m [T U5 [U5nT RT5mSSKJn SSKJn [5(aUmOUR5RmSmTcUcSmODT(a [T5O [5U(a [U5O [5-m[T5mU UUUUU4Sjn UU UUUU4Sjn [RRR!X U ST5n U U4S jnU U4S jn[-XUT5$!["amn [$R&"S[)U 55(a[+SU 35e[$R&"S [)U 55(a[+S U 35eU eSn A ff=f) z@ Paddle cond API will evaluate both true_fn and false_fn codes. rr0r2Nc>TR5nTRT[RR T55 T"5nUcTR T5nTR U5 U$rsave_checkpointrtr!utilscopy_mutable_varsr5restore_checkpoint)inplace_map_checkpointrYr| init_args inplace_maprcrr}s r' new_true_fn%_run_paddle_cond..new_true_fnsg"-!  LL * *9 5 i ;**Z(C&&'=> r(c>TR5nTRT[RR T55 T"5nUcTR T5nTR U5 U$rr)rrYrr|rrrcr}s r' new_false_fn&_run_paddle_cond..new_false_fnse"-!  LL * *9 5 j ;**Z(C&&'=> r(z7Unsupported return type of true_fn and false_fn in condzCYour if/else have different return type. TODO: add link to modify. zIncompatible return values ofzNYour if/else have different number of return value. TODO: add link to modify. c&>TRT5$r)r5)r|r}sr'r"_run_paddle_cond..svzz*-r(c(>TRTU5$r)rt)vsr|r}s r'rrs&**Z4r()rrwrr5r4r1r8r3r r9r:r rIr!r6nnr_ Exceptionresearchstrr_recover_args_state)rerrrrrcrdr1r3rr cond_outser|rrr}s `` ` @@@@r'rrs}  O^F'v~> !$ 'D ?+IKI}}) '446CC J>#9 ,;J ' +9Z 'z|M *%   MM$$)) |T: .H4H yHj II   99 Es1v  *UVWUXY  994c!f = =)`ab`cd  s,D"" F,A(FFc4U(a U"5nU$U"5nU$)z3 Evaluate python original branch function if-else. r)rerrrrrcpy_outss r'rrs  giG N&.ZG Nr(cF[U[5(Ga [U5S:aUnUSn[U[5(au[U5Hfup4[U[[ 45(dM"[ UR;dM8US:Xa S/USSQ7nOUS:Xa USSUSS-nO USU/USSQ7n O [U5H=up4[U[[ 45(dM"[ UR;dM8USUnM? U(dg[U5S:XaUS$U$U$)Nrr)r tuplelenrVr rrname)r processed_out align_retindexitems r'_remove_no_value_return_varrs##u#c(Q, F i ' '(3 dXu$566, 9z)-(8AB(8 !(1"1 AB(? )26E):(ESW(E  4%]3KE$5 122(DII5 -fu 5 4   1 $ # #  r(cUcg[U[[45(dU/n[[U5U5H,up4[U[5(dM[ SUSUS35e g)Nz Required 'zO' must be initialized both in if-else branch, but found it not initialized in 'z'.)r rIrzipr ValueError)outsnames branch_namerrs r'_check_no_undefined_varr5sp } dT5M * *vdU+  c< ( (TF"qr}q~AB ,r(cUcU$U"5n[U5n[U5nXV::deUS:Xa*[U[[45(dU4O [U5nOUS:XaU4O [U5nXUS-nU"U5 U$)z Currently we support variant length of early return statement by padding _no_return_value. # TODO(dev): We shall consider to evaluate whether should support this for Python if-else? Nr)rr rIr)rrrrcrnum_outsnum_args final_outss r'rrAs  I?#H9~H   1}%dT5M::TGd  #a-wU4[hi00  Z r(c[U[5(GaCURS:dS5eUR[R R R[R R R4;amURSS:aAUR[R R R:XaURS$[R"U5S$UR[R R R:Xa[RRU5$[S[U5S35e[U[5(aUR!5(dUR#5(agURS:dS5eURSS:a$UR!5(aURS$[R"U5S$UR%5(a[RRU5$[SS[U5S3-5e[U[&5(aUR)5$[+U5$)z Returns variable(length) from shape ops based on var.type Note: In addition to some ast transformations, some block-related operations are added in `len` transformation, such as appending `shape_op` in var.block. rzlen() of a 0-D tensor is wrongzOlen(var) only supports DenseTensor/DenseTensorArray/SelectedRows, but received .zBlen(var) only supports DenseTensor/DenseTensorArray/SelectedRows, z but received )r r ndimtyper VarDescVarType DENSE_TENSOR SELECTED_ROWSrr!DENSE_TENSOR_ARRAYro array_length TypeErrorris_dense_tensor_typeis_selected_row_typeis_dense_tensor_array_type VariableTuplerFrrs r' convert_lenr^s#x  xx!|===| 88 LL - - LL . .   ! q HH 4 4 A AAyy|#<<$Q' ' XX--@@ @==--c2 2abfgjbkallmn  C    # # % %)A)A)C)C88a< A!A A<yy|aC$<$<$>$>yy|#<<$Q' '  + + - -==--c2 2T!$s)A./  c= ) );;= 3xr(c[U5HDup[U[[45(dM"URSS:XdM7[ SUS35e [ U6$)NrzHNot support zip(tensor, ...) when tensor.shape[0] == -1, but found args[z].shape[0] == -1 in 'zip')rVr r rrrJr)rziargs r' convert_ziprs^D/ cHe, - -#))A,"2D""#$=? " :r(c&^T[LaT$U4Sj$)Nc>T"5$rr)clsinstancesuper_fns r'rconvert_super..sr()super)r s`r' convert_superrs5 ++r(c.\rSrSrSrSSjrSrSrSrg) riz this class will cause enumerate can't be wrapped by other iterator change function. this will be fixed when list is produced. VariableTuple can only deal with variables which is fixed. c Xl[U5Ul[UR[[ 45(a+[ R"X"UR-SS5Ulg[X"UR-5Ulg)Nrr) rrrr r rr!arangeragrange)selfrstarts r'__init__VariableTuple.__init__s[s# dhh5 1 2 2}}UDHH,URUURU4$r)rr)rrZs r'rGVariableTuple.__getitem__sxx}dhhsm++r(cUR$r)r)rs r'rFVariableTuple.__len__s xxr()rrrN)r) __name__ __module__ __qualname____firstlineno____doc__rrGrF__static_attributes__rr(r'rrs 6,r(rcT[SU55nU(a[U6$[U6$)Nc3N# UHn[U[[45v M g7frr.0r&s r' $convert_enumerate..FAz!h%677#%)anyrrVrz has_variables r'convert_enumerater-s+FFFLd## d r(cX[SU55nU(a[U5S:Xa[R"SUSSS5$[U5S:Xa[R"USUSSS5$[U5S:Xa"[R"USUSUSS5$[ U6$)Nc3N# UHn[U[[45v M g7frrr$s r'r& convert_range..r(r)rrr)r*rr!rrr+s r' convert_ranger3sFFFL t9>==DGQ8 8 t9>==a$q'1g> > t9>==a$q'47GD D $<r(cSn[U[[45(a_[UR5nU"U5(a;[ R"U5n[ U5HupEUbUS:dMX4X$'M U$UR$)z5 A function representation of the shape of variable. c&[SU55$)Nc3*# UH oS:v M g7f)rNrr$s r'r&6convert_shape..has_negative..s-*Qq5*s)r*) list_shapes r' has_negative#convert_shape..has_negatives-*---r(r)r r rrIrr!rV)r&r9values shape_tensorrrqs r' convert_shaper=su .!h&''agg   !<<?L!&)9A ,FI* wwr(c[U[[45(de[UR5S;a[ R "USS9nU$)N)rrr)r r rrrr!castrs r'rrs@ cHe, - -- -SYYx/kk#V, Jr(c@[U[[45(ab[UR5nUS;dSUR SUS35eUS;d SUS35eSS S S S.n[ R"XUS 9$US;d SUS35e[U5"U5$) N)rfloat16rirint32ruint8zThe dtype of var z is z(, which is not supported in the cast op.)rr$r#complexzThe casted target dtype is z), which is not supported in type casting.rrCri complex64r?) r r rrrrr!r@eval)rr src_dtypecast_maps r'convert_var_dtyperJs#%())!#)),    zi[8` a     *%0Y Z   "   {{3uo66   *%0Y Z   E{3r(c[U[[45(a%[R"US5nSSKJn U"U5$U(dU5eg)z= A function representation of a Python ``assert`` statement. rr)AssertN)r r rr!r@paddle.static.nn.control_flowrL)r_messagerLs r'convert_assertrOs@$5)**{{4(8d|Wtr(  FsependfileflushcUH?n[U[[45(dM [RR U5 MA [ X@XUS.6 g)z A function representing Python ``print`` function. It will print all arguments at compile time and only print the Tensor values at runtime. rRN)r r rr!r6Printprint)rSrTrUrVobjectsobjs r' convert_printr\#s@  cHe, - - MM   $ 7u=r(c## SSKJn [R"S5 UR 5R n[ RR5nURn UR5n [U R5n [XX#XE5n Sv [U R5n X;a/Xy'XyRXU 45 g7f)Nrr2zopaddle.amp.auto_cast is an experimental features in auto parallel.This will take no effect in normal dy2static.)program_translatorr3warningswarnr9 _amp_recordsr!r6r current_block_idxr7ropsrrW)enablecustom_white_listcustom_black_listlevelr use_promoter3 amp_records main_programrbr7 start_op_idx amp_options end_op_idxs r'r@r@.s6 MM : $002??K==557L$66 ..0M}(()L#4UK ]&&'J+)+ &")) J/sC C ch[U[[45(a[R"/USS9$U$)z6 Create a bool variable, which type is the same as x. r)rrr)r r rr!r")r&values r'create_bool_as_typerpOs.!h&''{{VDD r(r)NN))TNNO1rBT)T)M __future__rrr_ contextlibrr!paddle.autograd.py_layerrpaddle.base.data_feederrpaddle.base.dygraph.baserrpaddle.base.frameworkr r r paddle.frameworkr paddle.jit.utilsr paddle.pirrpaddle.static.amp.fp16_utilsr paddle.utilsrrpy_layerrrrrrrr__all__r%r.rCrLrSrOrNrfrwr]r^rrrrrrrrrrrrrrrrrrrr-r3r=rrJrOr\r@rprr(r'rsU# % 01NFF('33#  .  ( V A>:  FJ2:*&R  5F$   4F# &"! (VMJ` F :0f,, 4& R !$D>    @r(