DŽg.8ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZddlmZdZej$eZgdZGdd eZGd d eZGd d eZdZGddZGddeZ ddZddZy)N) as_completedThreadPoolExecutor)Optional)_prctl_pr_set_pdeathsigTORCH_MP_PARALLEL_START)ProcessContextProcessExceptionProcessExitedExceptionProcessRaisedExceptionspawn SpawnContextstart_processesc:eZdZddgZdededeffd ZdZxZS)r error_index error_pidmsgpidcNt||||_||_||_yN)super__init__rrr)selfrrr __class__s c/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/multiprocessing/spawn.pyrzProcessException.__init__$s& &c^t||j|j|jffSr)typerrrrs r __reduce__zProcessException.__reduce__*s&DzDHHd&6&6AAAr) __name__ __module__ __qualname__ __slots__strintrr __classcell__rs@rr r !s- ,ICc Brr c0eZdZdZdededeffd ZxZS)r zNException raised when a process failed due to an exception raised by the code.rrrc(t||||yr)rr)rrrrrs rrzProcessRaisedException.__init__1s k95r)r!r"r#__doc__r%r&rr'r(s@rr r .s,X6 66 66rr c NeZdZdZdgZ d dededededeef fd ZdZ xZ S) r zTException raised when a process failed due to signal or exited with a specific code. exit_coderrr signal_namecDt||||||_||_yr)rrr-r.)rrrrr-r.rs rrzProcessExitedException.__init__?s% k95"&rct||j|j|j|j|j ffSr)rrrrr-r.rs rr z!ProcessExitedException.__reduce__Ks9 J XXt''4>>4CSCS T  rr) r!r"r#r+r$r%r&rrr r'r(s@rr r :sP^ I&* '  ' ' '  ' c] ' rr c:ttj ||g|y#t$rYyt$rbddl}t |d5}tj|j|dddn #1swYnxYwtjdYywxYw)Nrwbr) rsignalSIGINTKeyboardInterrupt Exception tracebackopenpickledump format_excsysexit)fniargs error_filer7fhs r_wraprCRs FMM*  1 t    *d # 4r KK ,,. 3 4 4 4  s, % BB%A6- B6A? ;BBc eZdZdZdZddZy)r c||_||_t|Dcic]\}}|j|c}}|_ycc}}wr) error_files processes enumeratesentinel sentinels)rrGrFindexprocesss rrzProcessContext.__init__gsA&":CI:N (6wG  e #  s>cf|jDcgc]}t|jc}Scc}wr)rGr&r)rrLs rpidszProcessContext.pidsns#04?WGKK ???s.Nct|jdk(rytjj |jj |}d}|D]O}|jj |}|j|}|j|jdk7sM|}n|t|jdk(Sd}|jD]C}|jstjd|j|jEtj |z}|jD]4}t#d|tj z }|j|6|jD]R}|jr0tjd|j|j%|jT|j|} t'j(|j*|t&j,s||j|j} | dkr@ t/j0| j2} t7d || fz|| j| | t7d || fz|| j| t9|j*|d5} t;j<| } dddd|z}| z }t?||| j#t4$r d| d } YwxYw#1swY@xYw)a#Join one or more processes within spawn context. Attempt to join one or more processes in this spawn context. If one of them exited with a non-zero exit status, this function kills the remaining processes and raises an exception with the cause of the first process exiting. Returns ``True`` if all processes have been joined successfully, ``False`` if there are more processes that need to be joined. Args: timeout (float): Wait this long before giving up on waiting. rT)timeoutNz)Terminating process %s via signal SIGTERMzJUnable to shutdown process %s via SIGTERM , forcefully exiting via SIGKILLzz$process %d terminated with signal %s)rrr-r.z'process %d terminated with exit code %d)rrr-rbz5 -- Process %d terminated with the following error: ) lenrJmultiprocessing connectionwaitkeyspoprGjoinexitcodeis_alivelogwarningr terminatetime monotonicmaxkillosaccessrFR_OKr3Signalsname ValueErrorr r8r9loadr )rrPreadyrrIrKrLend time_to_waitfailed_processr[rhrBoriginal_tracers rrZzProcessContext.joinqs t~~ ! # **// NN   !0    HNN&&x0EnnU+G LLN1$#     t~~&!+ +~~ $G! GU!!# $nn(~~ 'Gq#(8"89L LL & '~~ G! `KK  LLN  4yy))+6@~~k2;;H!|;!>>8)499D-:k4=PP +,00& $ -=h@WW +,00& $"";/ 6 -"#[[_N -H;V ~$S+~7I7IJJ+";-xi[:D;" - -s1 K&'K<&K98K9<Lr)r!r"r#rrNrZrrr r fs @ZKrr ceZdZfdZxZS)rcPtjdt| ||y)Nzmps r start_processz&start_processes..start_processs| ( (' %    "''**arww'   '277""r) max_workers)rdenvirongetENV_VAR_PARALLEL_STARTr]inforU get_contextrangersubmitrresultr rZ)r>r@nprocsrZr~ start_methodstart_parallelrFrGrr?idxrLtf_nameexecutorfuturesfutcontextrs`` ` @rrrsV  $ JJNN13 73 > 23  $ $\ 2B&6/KI#( v %A$1!$4 !C'&K $IcN %  F 3 )xBG-PQx}a8PGP#G, )(+ %Wg#* C !( #  ) )Y 4G lln llnQ ) )s)E7D=0E=EE cr|dk7r"d|d}tj|tdt|||||dS)aSpawns ``nprocs`` processes that run ``fn`` with ``args``. If one of the processes exits with a non-zero exit status, the remaining processes are killed and an exception is raised with the cause of termination. In the case an exception was caught in the child process, it is forwarded and its traceback is included in the exception raised in the parent process. Args: fn (function): Function is called as the entrypoint of the spawned process. This function must be defined at the top level of a module so it can be pickled and spawned. This is a requirement imposed by multiprocessing. The function is called as ``fn(i, *args)``, where ``i`` is the process index and ``args`` is the passed through tuple of arguments. args (tuple): Arguments passed to ``fn``. nprocs (int): Number of processes to spawn. join (bool): Perform a blocking join on all processes. daemon (bool): The spawned processes' daemon flag. If set to True, daemonic processes will be created. start_method (str): (deprecated) this method will always use ``spawn`` as the start method. To use a different start method use ``start_processes()``. Returns: None if ``join`` is ``True``, :class:`~ProcessContext` if ``join`` is ``False`` r z3This method only supports start_method=spawn (got: zU). To use a different start_method use: torch.multiprocessing.start_processes(...)) stacklevel)r)rsrt FutureWarningr)r>r@rrZr~rrs rr r sKBwA,P: :  c=Q7 2tVT6 PPr)rprTFr )loggingrUmultiprocessing.connectionrdr9r3r<rr`rsconcurrent.futuresrrtypingrrr getLoggerr!r]__all__r6r r r rCr rrr rprrrs!  ?%3g!  By B 6- 6 - 0(eKeKP1>1  A H(Qr