
    Αi                     2    S SK r SSKJrJr   " S S\5      rg)    N   )
ControllerControllerModec                   0    \ rS rSr\S 5       rS rS rSrg)RpcController   c                     UR                   R                  [        R                  :X  a)  UR                  R                  U R                   S35        gg)Nz enabledTF)argsrun_moder   RPCloggerdebug__name__)clsctxs     i/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/distributed/launch/controllers/rpc.pyenableRpcController.enable   s=    88 2 22JJ~X67    c                 t    U R                   R                  R                  c   S5       eU R                  5         g )Nz*Master is None, Please set master address!)r   r
   master_build_pod_with_master)selfs    r   	build_podRpcController.build_pod   s4    xx}}##/ 	
8	
/ 	##%r   c           
         U R                  5       U R                  l        [        U R                  R
                  R                  5      U R                  l        U R                  R                  R                  5       nU R                  R                  R                  U R                  R                  5       Vs/ s H(  nU R                  R                  R                   SU 3PM*     nn[        R                  " U R                  R                  U R                  R                  U R                  R                  U R                  R                  R                  R                  U R                  R                  R                   SU 3SR!                  U5      S.5      nU R"                  R%                  SU R&                  R(                   S3U R                  R                  UU R&                  R                  U R                  R                  5      u  pVX`R                  l        [+        U5      S:  a  gU Vs/ s H  n[        R,                  " U5      PM     nnU R                  R.                  R1                  SU 35        U R3                  U5        [5        U Vs/ s H  owS	   PM	     sn5      n[5        US U  Vs/ s H  owS	   PM	     sn5      n	US
   S   n
U R                  R7                  5         [9        U R                  R                  5       H&  nU
X7   Xy-    U S.nSXy-    3nU R;                  XS9  M(     gs  snf s  snf s  snf s  snf )N:,)namerankreplicasdtype	candidate	endpoints/z/infor   Fzsync peers done r!   r   r#   )PADDLE_MASTER_ENDPOINTPADDLE_WORKER_ENDPOINTPADDLE_TRAINER_IDPADDLE_TRAINERS_NUMz
workerlog.)envslog_fileT)pod_replicaspodr!   intr   r
   r    nodeget_free_portget_free_portsipjsondumpsr   devicer"   joinr   
sync_peersjobidlenloadsr   r   save_pod_logsumresetrangeadd_container)r   portpr$   data	peer_listr    iglobal_sizerank_offset
rpc_masterer+   s                r   r   $RpcController._build_pod_with_master#   s    --/ DHHMM../xx}}**,
 XX]]11$((2C2CD
D xx}} !%D 	 

 zz HH----33 $ 0 014&9 XXi0	
 ++00}E"HHMMHHHHMM
	 y>A,56IqTZZ]I	6 0<=)$)<)QZ=)<=)ET2BC2BQZ=2BCDq\+.
txx(()A*4*3,()'8*5	A $AO#45HA9 * ]
6 7
 =Cs   7/L:4 L?M9M	 N)	r   
__module____qualname____firstlineno__classmethodr   r   r   __static_attributes__rK   r   r   r   r      s     &8r   r   )r3   
controllerr   r   r   rK   r   r   <module>rR      s     2GJ Gr   