Αi,$LSSKJr "SS5r"SS5r"SS5rg) )Variablec\rSrSrSrSrSrg) VarStructz1 record part properties of a Variable in python. cLXlX lX0lX@lXPlX`lgN)nameshapedtypetype lod_level persistable)selfr r r r r rs v/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/distributed/transpiler/details/vars_distributed.py__init__VarStruct.__init__s!    "&)r r r rr r N)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__rrrrs 'rrcR\rSrSrSrS Sjr\S5r\S5rSr Sr g) VarDistributedz a class to record the var distributed on parameter servers. the class will record the relationship between origin var and slice var. the slice var's properties, such as type/shape/offset/endpoint. Nc[U[5(aURU5UlOXl[U[5(aURU5UlOX lUR URUR5(aSUlSUlSUlOSUlSUlSUlUbX0lUbX@lUbXPlX`l Xpl g)a Args: origin_var(Variable|VarStruct): origin var properties slice_var(Variable|VarStruct): slice var properties is_slice(bool|None): slice or not, slice_var=True/False and its block size > 8192 are the judgement standard. block_id(int|None): the number about the slice var. offset(int|None): if the slice var is sliced, offset is the numel before the var. vtype(str|None): a tag, such as Optimizer/Param/RemotePrefetch. endpoint(str|None): which parameter the slice var on, such as "127.0.0.1:1001" FrTN) isinstancer"_VarDistributed__create_var_structoriginsliceequalis_sliceblock_idoffsetvtypeendpointr origin_var slice_varr$r%r&r'r(s rrVarDistributed.__init__&s* j( + +22:>DK$K i * *11)PPXY]YcYcYiYiXjkmm_HT]]O9T[[MQR T  TZZL(88HHYZgYhhvw{xExEwFFGH Hr)r%r(r$r&r!r"r'NNNNN) rrrrrr staticmethodr r#r<rrrrrrsL0!d    "Hrrch\rSrSrSrSrS SjrSr\S5r Sr SS jr SS jr S r S rg)VarsDistributedaG a gather about VarDistributed with many methods to find distributed vars. through the class, we can get overview about the distributed parameters on parameter servers. this class may centralized and convenient for developer to manage and get variable's distribute. other module can also use this to find variables such io.py. c/Ulgrdistributed_vars)rs rrVarsDistributed.__init__s "rNc XURR[UUUUUUU55 g)a@ add distributed var in this. Args: origin_var(Variable|VarStruct): origin var properties slice_var(Variable|VarStruct): slice var properties is_slice(bool|None): slice or not, slice_var=True/False and its block size > 8192 are the judgement standard. block_id(int|None): the number about the slice var. offset(int|None): if the slice var is sliced, offset is the numel before the var. vtype(str|None): a tag, such as Optimizer/Param/RemotePrefetch. endpoint(str|None): which parameter the slice var on, such as "127.0.0.1:1001" Returns: None N)rEappendrr)s radd_distributed_var#VarsDistributed.add_distributed_vars50 $$   rcfURH!nURRU:XdMUs $ g)z get distributed var by conditions. Args: var_name(str): slice var name, such as "w.trainer0.block1" Returns: VarDistributed: distributed var. N)rEr"r )rvar_namedist_vars rget_distributed_var_by_slice,VarsDistributed.get_distributed_var_by_slices0--H~~""h..rctURUR:H=(a URUR:H=(ay URUR:H=(aY URUR:H=(a9 URUR:H=(a UR UR :H$r2)r r r r r rr3s rr#VarsDistributed.equals II " 5 TYY& 5 djj( 5 djj( 5$..0  5   D$4$44  rcURH3nURRU:XdMURU:XdM1Us $ g)z get distributed var by conditions. Args: origin_var_name(str): endpoint(str): the parameter endpoint, such as "127.0.0.1:1001" Returns: VarDistributed: distributed var. N)rEr!r r()rorigin_var_namer(rMs r$get_distributed_var_by_origin_and_ep4VarsDistributed.get_distributed_var_by_origin_and_eps>--H$$7%%1 . rcB/nURH&nURU;dMURU5 M( U(dU$0nUHUnURRnXeR 5;aUR U5nO/nURU5 XuU'MW U$)a get distributed vars by conditions. Args: vtype(str|None): distributed var's vtype, such as "Optimizer", "RemotePrefetch" groupby(bool|False): group by origin var or not. Returns: list: distributed var list. dict: distributed var map when groupby=True )rEr'rHr!r keysget)rvtypesgroupby vtype_varsr. params_maprS optimizerss rget_distributed_vars_by_vtypes.VarsDistributed.get_distributed_vars_by_vtypess ((CyyF"!!#&)  C!jjooO//"33'^^O<    c "*4 'rc/nURH&nURU:XdMURU5 M( U(dU$/nUH&nURU:XdMURU5 M( U$)z get distributed vars by conditions. Args: endpoint(str): the parameter server endpoint, such as "127.0.0.1:2001" vtype(str|None): distributed var's vtype, such as "Optimizer", "RemotePrefetch" Returns: list: distributed var list. )rEr(rHr')rr(r' endpoint_varsr.r[s rget_distributed_vars_by_ep*VarsDistributed.get_distributed_vars_by_epso ((C||x'$$S))   CyyE!!!#&!rc/nURHnUR[U55 M SRU5$)zi get the overview string about all params on all parameter servers. Returns: Str: overview string.  )rErHstrjoin)rvars_strr.s roverviewVarsDistributed.overviews8((C OOCH %)yy""rrDr>)Fr)rrrrrrrIrNr?r#rTr^rbrirrrrrArAsP#" H     $>0 #rrAN) paddle.staticrrrrArrrrls/# ' '^H^HBa#a#r