o h2@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZddlmZdZeeZgdZGdd d eZGd d d eZGd d d eZddZGdddZGdddeZ     dddZdddZdS)N) as_completedThreadPoolExecutor)Optional)_prctl_pr_set_pdeathsigTORCH_MP_PARALLEL_START)ProcessContextProcessExceptionProcessExitedExceptionProcessRaisedExceptionspawn SpawnContextstart_processescs:eZdZddgZdededeffdd ZddZZS) r error_index error_pidmsgpidcs"t|||_||_||_dSN)super__init__rrr)selfrrr __class__o/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/multiprocessing/spawn.pyr$s  zProcessException.__init__cCst||j|j|jffSr)typerrrrrrr __reduce__*szProcessException.__reduce__) __name__ __module__ __qualname__ __slots__strintrr __classcell__rrrrr !sr cs.eZdZdZdededeffdd ZZS)r zNException raised when a process failed due to an exception raised by the code.rrrcst|||dSr)rr)rrrrrrrr1szProcessRaisedException.__init__)rrr __doc__r"r#rr$rrrrr .sr c sLeZdZdZdgZ d dededededeef fdd Zd d Z Z S) r zTException raised when a process failed due to signal or exited with a specific code. exit_codeNrrr signal_namecs t|||||_||_dSr)rrr&r')rrrrr&r'rrrr?s zProcessExitedException.__init__cCs t||j|j|j|j|jffSr)rrrrr&r'rrrrrKsz!ProcessExitedException.__reduce__r) rrr r%r!r"r#rrrr$rrrrr :s  r c Csttjz ||g|RWdStyYdStyIddl}t|d}t| |Wdn1sjs z+ProcessContext.__init__..) error_files processes enumerate sentinelsrr@r?rrrrgs  zProcessContext.__init__cCsdd|jDS)NcSsg|]}t|jqSr)r#r)r;r=rrr osz'ProcessContext.pids..)r@rrrrpidsnszProcessContext.pidstimeoutcCs8t|}|jD]}td|t}||q dS)z4Attempt to join all processes with a shared timeout.rN)time monotonicr@maxjoin)rrFendr= time_to_waitrrr_join_procs_with_timeoutqs   z'ProcessContext._join_procs_with_timeoutN grace_periodcCst|jdkr dStjj|j|d}d}|D]}|j|}|j|}||j dkr2|}nq|dur>t|jdkS|durG| ||jD]}| r[t d|j|qJ| |durddn||jD]}| r{t d|j||qj|j|}t|j|tjs|j|j } | dkrz t| j} Wntyd| d } Ynwtd |d d | ||j| | d td |d d| d ||j| dt|j|d } t| } Wdn1swYd|d d} | | 7} t| ||j)aLJoin 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 (optionally with a grace period) 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 (in seconds) before giving up on waiting. grace_period (float): When any processes fail, wait this long (in seconds) for others to shutdown gracefully before terminating them. If they still don't exit, wait another grace period before killing them. rT)rFNz)Terminating process %s via signal SIGTERMzJUnable to shutdown process %s via SIGTERM , forcefully exiting via SIGKILLzzprocess dz terminated with signal )rrr&r'z terminated with exit code )rrr&rbz -- Process z& terminated with the following error: )lenrBmultiprocessing connectionwaitkeyspopr@rJexitcoderMis_alivelogwarningr terminatekillosaccessr?R_OKr)Signalsname ValueErrorr r.r/loadr )rrFrNreadyrr:r<r=failed_processrYrcr8original_tracerrrrrJxsz           zProcessContext.join)NN) rrr rrEfloatrMrrJrrrrrfsrcseZdZfddZZS)r cstdt||dS)Nz.start_process) max_workerscsg|]}|qSr)submit)r;r5)executorr|rrrDsz#start_processes..)r_environgetENV_VAR_PARALLEL_STARTr[inforT get_contextrangerrresultrrJ)r4r6nprocsrJrs start_methodstart_parallelr?r@r5idxr=tf_namefuturesfutcontextr)r6rsrr4r{r|rrs:           rcCs8|dkrd|d}tj|tddt|||||ddS)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)rjrk FutureWarningr)r4r6rrJrsrrrrrr ,s ! )rrTFr )loggingrTmultiprocessing.connectionr_r/r)r2rtrGrjconcurrent.futuresrrtypingrrr getLoggerrr[__all__r,r r r r9rr rr rrrrs:      t D