9iM SSKrSSKJrJr SSKrSSKrSSKJr SSK J r \R"5r S\\ \44SjrSSjrSr\R$"5SS \S \ S \4S jj5r\R$"5SS \4S jj5rg)N)AnyUnion) device_guardsizecp[U[5(aU$UR5RS5(a[USS5S-$UR5RS5(a[USS5S-$UR5RS5(a[USS5S-$UR5RS 5(a.[USS 5S -nURS 5(aUS -$U$UR5RS5(a.[USS 5S-nURS 5(aUS -$U$UR5RS5(a.[USS 5S-nURS 5(aUS -$U$[ S5e)z Converts a size expressed as a string with digits an unit (like `"5MB"`) to an integer (in bytes). Args: size (`int` or `str`): The size to convert. Will be directly returned if an `int`. GIBNi@MIBiKIBiGBiʚ;bMBi@BKBizR`size` is not in a valid format. Use an integer followed by the unit, e.g., '5GB'.) isinstanceintupperendswith ValueError)rint_sizes o/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/inference/models/common/vlm/distributed.pyconvert_file_size_to_intrs $  zz|U##49~'' zz|U##49~'' zz|U##49~'' zz|T""tCRy>U+ $ c 2 2x1}@@ zz|T""tCRy>U+ $ c 2 2x1}@@ zz|T""tCRy>U+ $ c 2 2x1}@@ \ c## UR[R:Xa![R"UR 5nO-[ [R"U5R55n[U5nURS/5 U[UR5-n[SX#5H&nX[X$U-5nXT[X$U-544v M( g7f)Nr)dtypepaddleint8npprodshapernumelitemrreshape_dtype_byte_sizerangemin)tensor buffer_sizer$ send_sizex part_tensors r reduce_tensorr/7s ||v{{" %FLL(--/0*;7K OORDv||<3u:6J9%BCC:$$&q)*H q=rr*dstreturnc ^^^[U[[45(a[U5"UUU4SjU55$[U[5(a2UR 5VVs0sHupEU[ UTTT5_M snn$SnT[R"TS9:HnU(aT(a.[[R"TS95Vs/sHn/PM nnOq[[R"TS95Vs/sHn[R"U5PM nnUV s/sH#n [U R5S:aU OU SPM% nn T(GaURn URS/5 [!U5GHupSn [R"TS9T:XaA[[R"TS95Vs/sHn[R"U 5PM n n[RR"R$R'U U T(aTR(TOTTSSS9 U(dM[[U55H4nXnR+XR-5R/55 M6 GM URU 5 U(a[[1S5 /nUH;n[2R4"U5n U R7U 5n UR+U 5 M= UnSSS5 U$U$[RR"R$R'UUT(aTR(TOTTSSS9 U$s snnfs snfs snfs sn fs snf!,(df  U$=f![8a [9S 5ef=f) Nc3@># UHn[UTTT5v M g7fN)distributed_gather).0tr9groupoffloads r %distributed_gather..Ws# DJq"1c5'::FsrArrTF)r9rAsync_opuse_calc_streamcpu(Not currently using distributed training)rtuplelisttypedictitemsr> distributedget_rankr(get_world_sizer empty_likelenr#r&r/ communicationstreamgatherranksappendrHnumpyrr! concatenatereshapeAssertionError)r*r9rArBkvoutput_tensorsis_dst_r@ origin_shape slice_tensorindexslice_output_tensorsinew_output_tensorsr-s ``` rr>r>SsXJI fudm , ,< DJ   fd # #JP,,.JX$!%aeW==.  ,,599  %k&@&@u&M N" N1B N" #;#=#=E#JK"K%%f-K" AO"@N1QWW)Aqw6" !<> (,1 C(s $) ?6"3~#67&)001E1H1L1L1N1T1T1VW8#(=( OOL )!%()+&+NN1-IIl3*11!4,&8N )$'&    , , 3 3 : :(-EKK$3 % ; C""",&)($ IGHHIs7M2(M2%MM2AM2 M "M28 MM2*MA2M2: MAM21A4M2%AM + M25M28A M2M2 M/*M2/M22Nc ^^[U[[45(a[U5"UU4SjU55$[U[5(a1UR 5VVs0sHup4U[ UTT5_M snn$/nT(aX[S5 [[R"T55Vs/sHn[R"U5PM nnSSS5 OC[[R"T55Vs/sHn[R"U5PM nnUVs/sH#n[UR5S:aUOUSPM% nnT(GaURnURS/5 UHn U RS/5 M [!U5Hup[[R"T55Vs/sHn[R"U 5PM n n[R""XTS9 [%X5H1up[S5 U R'5XSU S&SSS5 M3 M URU5 UHn U RU5 M U$[R""XP5 U$s snnfs snf!,(df  GN|=fs snfs snfs snf!,(df  M=f![(a [)S5ef=f) anested all gather function with offload Args: tensor (Any): the desired tensor, list of tensor, dict of tensor to allgather. group (_type_, optional): the communication group. Defaults to None. offload (bool, optional): If True, we offload the received tensor to cpu/(numpy). Defaults to False. Raises: AssertionError: Unexpected errors. Returns: tensor list: list of all gathered tensors c3>># UHn[UTT5v M g7fr=)distributed_allgather)r?r@rArBs rrC(distributed_allgather..s! BHQ%a88&srHNrrrErrI)rrJrKrLrMrNrjrr(rOrQrrRrSr#r&r/ all_gatherziprHr\)r*rArBr]r^r_rar@rbr-rcrdreys `` rrjrjs4I fudm , ,< BH   fd # #HN HV(E7;;  e$#;#=#=e#DE"E%%f-E"%${99%@AAA!!&)A  GUUns177|a/!QtW :W"%$  V( -, IGHHIs6K(K#J=KK"J7 J J*K J$#K)*J)A4K J.'2KJ32rjrrrvs  (  ' ' ) 5c?8 : KIsKIKIPSKIKI\BI#BIBIr