ϑiSSKrSSKJr SSKJrJr "SS\R RRR5r "SS5r S S jr S S jr g) N)Variable) LayerHelpercorec8^\rSrSrSrU4SjrU4SjrSrU=r$)BlockGuardServz` BlockGuardServ class. BlockGuardServ class is used to create an op with a block in a program. c>[U[5(d [S5e[TU]UR R 5 Xlg)Nz$BlockGuardServ takes a ListenAndServ) isinstance ListenAndServ TypeErrorsuper__init__helper main_programserver)selfr __class__s [/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/incubate/nn/layer/io.pyrBlockGuardServ.__init__s86=11BC C 334 c`>UbgURR5 [TU] XU5$)NF)r complete_opr__exit__)rexc_typeexc_valexc_tbrs rrBlockGuardServ.__exit__!s.   !w6::r)r) __name__ __module__ __qualname____firstlineno____doc__rr__static_attributes__ __classcell__)rs@rrrs  ;;rrc:\rSrSrSrS SjrSrSrSrSr Sr g ) r )aV **ListenAndServ Layer** ListenAndServ is used to create a rpc server bind and listen on specific TCP port, this server will run the sub-block when received variables from clients. Args: endpoint(string): IP:port string which the server will listen on. inputs(list): a list of variables that the server will get from clients. fan_in(int): how many client are expected to report to this server, default: 1. optimizer_mode(bool): whether to run the server as a parameter server, default: True. Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:DISTRIBUTED) >>> from paddle.incubate.nn.layer.io import ListenAndServ >>> import paddle >>> paddle.enable_static() >>> place = paddle.CPUPlace() >>> main = paddle.static.Program() >>> with paddle.static.program_guard(main): ... serv = ListenAndServ( ... "127.0.0.1:6170", ["X"], optimizer_mode=False) ... with serv.do(): ... x = paddle.static.data( ... shape=[32, 32], ... dtype='float32', ... name="X") ... paddle.nn.initializer.Constant(value=1.0)(x, main.global_block()) ... paddle.scale(x=x, scale=10.0) >>> exe = paddle.static.Executor(place) >>> exe.run(main) cb[S5UlX l/UlXlX0lX@lg)Nlisten_and_serv)rrinputsoutputsendpointfan_inoptimizer_mode)rr,r*r-r.s rrListenAndServ.__init__Os-!"34     -rc[U5$)N)r)rs rdoListenAndServ.doYs d##rcURRnUR5nUR5n/n/nURHnUR (avSUR ;adSUR ;aRURUR SR5 URUR SR5 MMMURH[nURU5HCnURURU55 URURU55 ME M] M XE4$)NGradParam) rr current_block parent_blockopsr.r*appendname input_namesinputvar) rrr6r7paramsgradsopiname in_var_names rget_params_and_grads"ListenAndServ.get_params_and_grads\s{{// $224 ((* ##B""RYY&7bii+?MM"))G"4"9"9:LL6!2!7!78,@&  ^^E')xx  l&6&6{&CD \%5%5k%BC(7,$}rcURRnUR5RnUS:deUR U5nU$)Nr)rrr6 parent_idxblock)rprogrFr7s rr7ListenAndServ.parent_blockrsE{{'''')44 Qzz*- rc SSKJn URRnUR 5nUR 5nUR SSUR00URURU/URS/S.S9 g)Nr)DistributedModer)X)r,Faninoptimize_blocksdistributed_modegrad_to_block_idtyper*r+attrs) 7paddle.incubate.distributed.fleet.parameter_server.moderKrrr6r7 append_opr*r,r-SYNC)rrKrr6r7s rrListenAndServ.complete_opys {{// $224 ((* "% MM!$%4$8$8%'D  r)r,r-rr*r.r+N)T) rr r!r"r#rr1rCr7rr$rrr r )s!#J-$, rr c [U5[:XdeUc/nO[U[5(aU/n[U5[:XdeUR S5n[[ U55n[ S 0[5D6n[RR5nURSSU0SU0SUSUU[RRR0S9 U(aURS SU0S/0SU0S9 gg) aK Send variables to the server side, and get vars from server side when server have finished running server side program. Args: endpoints (str): comma separated IP:PORT pairs in the order of send_vars to send send_vars (list): variables to send to server sync (bool): whether to wait the request finish N,sendrLOut endpointsepmaprR send_barrier)Send)rSlistr rsplitsetrlocalsrop_proto_and_checker_makerkOpRoleAttrNamerVOpRoleRPC)r_ send_vars dummy_outputsyncr`rrpc_op_role_names rrbrbs   ?d "" " L( + +$~   %% % OOC ESZ I  ,68 ,F66FFH  Y %  U d==DDHH   &BK *   rcn[U5[:XdeUc/nO[U[5(aU/n[U5[:XdeUR S5n[[ U55n[ S 0[5D6nURSSU0SU0XS.S9 U(aURSSU0SU0S 9 U$) a5 Receive variables from server side Args: endpoints (str): comma separated IP:PORT pairs in the order of send_vars to send get_vars (list): vars to get from server after send completes. sync (bool): whether to wait the request finish Returns: list: list of received variables r\recvrLr^)r_r`rR fetch_barrierr_)rSr+rT)Recv) rSrcr rrdrerrfrV)r_get_vars dummy_inputrmr`rs rrrrrs >T !! ! K * *"m   $$ $ OOC ESZ I  ,68 ,F  [!!%6    H% *  Or)NT)paddlebase.frameworkr frameworkrrstaticnn control_flow BlockGuardrr rbrrrZrrr|sJ'+;V]]%%22==;*f f R+ \&r