t:is'b%SrSSKrSSKrSSKrSSKrSSKJr SSKJrJ r SSK J r J r J r JrJr SSKrSSKJr SSKJrJrJrJrJrJr \R6"S 5r\ \\\\44r\\ S '\ \\\4r!\\ S '\ \\!4r"\\ S '\"S S55r#S\!SS4Sjr$S&Sjr%S\4Sjr&S\4Sjr'S\(4Sjr)S\ SS4Sjr*S\4Sjr+S\S\S\4Sjr,S\RZS\4Sjr.\ "SS\55r/S\R`S \/S\R`4S!jr1S&S"jr2S\ \4S#jr3"S$S%5r4g)'z-XGBoost collective communication related API.N) dataclass)IntEnumunique)AnyDictOptional TypeAliasUnion)_T)_LIB _check_call build_infoc_str make_jcargspy_strz[xgboost.collective]_Conf_ArgVals_Argsc\rSrSr%SrSr\\\S'Sr \\\S'Sr \\ \S'Sr \\\S'Sr \\\S'S \S \4S jrS rg) ConfigagUser configuration for the communicator context. This is used for easier integration with distributed frameworks. Users of the collective module can pass the parameters directly into tracker and the communicator. .. versionadded:: 3.0 Attributes ---------- retry : See `dmlc_retry` in :py:meth:`init`. timeout : See `dmlc_timeout` in :py:meth:`init`. This is only used for communicators, not the tracker. They are different parameters since the timeout for tracker limits only the time for starting and finalizing the communication group, whereas the timeout for communicators limits the time used for collective operations, like :py:meth:`allreduce`. tracker_host_ip : See :py:class:`~xgboost.tracker.RabitTracker`. tracker_port : See :py:class:`~xgboost.tracker.RabitTracker`. tracker_timeout : See :py:class:`~xgboost.tracker.RabitTracker`. Nretrytimeouttracker_host_ip tracker_porttracker_timeoutargsreturncvURbURUS'URbURUS'U$)z*Update the arguments for the communicator. dmlc_retry dmlc_timeout)rrselfrs U/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/xgboost/collective.pyget_comm_configConfig.get_comm_config:s7 :: !!%D  << ##'<ns  $ $ &C Jr(c0[R"5nU$)zLGet total number workers. Returns ------- n : Total number of process. )r XGCommunicatorGetWorldSizer<s r%get_world_sizerAzs  ) ) +C Jr(cB[R"5n[U5$)z.If the collective communicator is distributed.)r XGCommunicatorIsDistributedbool)is_dists r%is_distributedrFs..0G =r(msgc[U[5(d [U5n[R"5nUS:wa7[ [R "[ UR5555 g[UR5SS9 g)zPrint message to the communicator. This function can be used to communicate the information of the progress to the communicator. Parameters ---------- msg : str The message to be printed to the communicator. rT)flushN) isinstancer1r rCrXGCommunicatorPrintrstripprint)rGrEs r%communicator_printrNs] c3  #h..0G!|D,,U399;-?@A ciik&r(c[R"5n[[R"[R "U555 UR n[U5$)zPGet the processor name. Returns ------- name : The name of processor(host) )ctypesc_char_prr XGCommunicatorGetProcessorNamebyrefvaluer)name_strrTs r%get_processor_namerVs? H33FLL4JKL NNE %=r(datarootc l[5n[R"5nX:Xa=UcS5e[R"U[R S9n[ U5Ul[[R"[R"U5[R"[R5U55 X:wa[RUR-"5n[[R"[R"U[R5URU55 [R "UR"5nAU$[[R"[R"[R$"W5[R5URU55 AU$)a'Broadcast object from one node to all other nodes. Parameters ---------- data : any type that can be pickled Input data, if current rank does not equal root, this can be None root : int Rank of the node to broadcast data from. Returns ------- object : int the result of broadcast. z&need to pass in data when broadcasting)protocol)r>rPc_ulongpickledumpsHIGHEST_PROTOCOLlenrTrr XGCommunicatorBroadcastrSsizeofc_charcastc_void_ploadsrawrQ)rWrXranklengthsdptrs r% broadcastrks2 :D ^^ F |I!II LL(?(? @1v  $$ LL &--"?   |  ,/  ( ( D&//2FLL$  ||DHH%  K   ( ( FOOA.@&,,PT   Kr(dtypec[R"S5S[R"S5S[R"S5S[R"S5S[R"S 5S [R"S 5S [R"S 5S[R"S5S[R"S5S[R"S5S[R"S5S0 nUR[R"S5S05 X;a[SUS35eX$![a N$f=f)Nfloat16rfloat32r float64int8int16int32int64uint8uint16 uint32 uint64 float128z data type z* is not supported on the current platform.)nprlupdate TypeError)rl dtype_maps r% _map_dtypers Q Q Q ! 1 1 1 1 A B B I "((:.23 *UG+UVWW      s6'D55 EEc0\rSrSrSrSrSrSrSrSr Sr S r g ) Opz#Supported operations for allreduce.rr rqrrsrur)N) r*r+r,r-r.MAXMINSUM BITWISE_AND BITWISE_OR BITWISE_XORr2r)r(r%rrs#- C C CKJKr(ropc t[U[R5(d [S5eUR 5R 5n[ [R"URR[R5UR[UR5[U555 U$)zPerform allreduce, return the result. Parameters ---------- data : Input data. op : Reduction operator. Returns ------- result : The result of allreduce, have same shape as data Notes ----- This function is not thread-safe. z%allreduce only takes in numpy.ndarray)rJrndarrayrravelcopyrr XGCommunicatorAllreducerPdata_asrdsizerrlr/)rWrbufs r% allreducers& dBJJ ' '?@@ **,   C $$ JJ  v / HH syy ! G   Jr(c@[[R"55 g)zKill the process.N)rr XGCommunicatorSignalErrorr)r(r% signal_errorr$s..01r(cSSKJn URb*[RR UR5nO<[ US5(a)[UR5S:aURSnOSnU(dg[R"U5nU(dgSnUHnURS5(dMUn O Ub![RRX5nU$g)Nr)lib__path__z libnccl.so) nvidia.ncclr__file__ospathdirnamehasattrr_rlistdir startswithjoin)rrfileslibnamenamers r% _find_ncclr)s ||!#!> j ! !c#,,&7!&;,,q/  JJw E !G ??< ( (G  ww||G- r(cJ\rSrSrSrS\SS4SjrS\4SjrS\ SS4Sjr S r g) CommunicatorContextiKzNA context controlling collective communicator initialization and finalization.rrNc XlSnURUS5bg[5nUS(dg[5nU(aX@RU'gg![a gf=f)Ndmlc_nccl_pathUSE_DLOPEN_NCCL)rgetrr ImportError)r$rkeybinfors r%__init__CommunicatorContext.__init__Nsd  88C  *  &'  rArDrFrNrVrkrlrrrrrrrr)r(r%rs3  ! 88KK   1 2U38_,-y-uS#X/)/X &y& '' 'T#>#>d#>L/ #    'C'D'( C ,B,c,b,`bhh32BJJB2::B2 HSMDPPr(