U ʹj!@sddlZddlZddlZddlZddlZddlmZddlmZGddde Z Gddde Z Gd d d e Z d d Z Gd ddZGdddeZdddZdddZdS)N)Optional)_prctl_pr_set_pdeathsigcs6eZdZddgZeeedfdd ZddZZS)ProcessException error_index error_pid)msgrpidcs"t|||_||_||_dSN)super__init__rrr )selfrrr  __class__Z/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/multiprocessing/spawn.pyr s zProcessException.__init__cCst||j|j|jffSr )typerrr r rrr __reduce__szProcessException.__reduce__) __name__ __module__ __qualname__ __slots__strintr r __classcell__rrrrr srcs*eZdZdZeeedfdd ZZS)ProcessRaisedExceptionz^ Exception is thrown when the process failed due to exception raised by the code. )rrrcst|||dSr )r r )r rrrrrrr szProcessRaisedException.__init__)rrr__doc__rrr rrrrrrs rcsBeZdZdZdgZd eeeeeedfdd ZddZ Z S) ProcessExitedExceptionzg Exception is thrown when the process failed due to signal or exited with a specific code. exit_codeN)rrrr signal_namecs t|||||_||_dSr )r r rr )r rrrrr rrrr /szProcessExitedException.__init__cCs t||j|j|j|j|jffSr )rrrr rr rrrrr;sz!ProcessExitedException.__reduce__)N) rrrrrrrrr rrrrrrr's rcCshttjz||f|WnFtk r0Yn4tk rbddl}||t dYnXdS)Nrr) rsignalSIGINTKeyboardInterrupt Exception tracebackput format_excsysexit)fniargs error_queuer%rrr_wrapBs r.c@s&eZdZddZddZdddZdS) ProcessContextcCs$||_||_ddt|D|_dS)NcSsi|]\}}|j|qSr)sentinel).0indexprocessrrr Ysz+ProcessContext.__init__..) error_queues processes enumerate sentinelsr r6r5rrrr Vs zProcessContext.__init__cCsdd|jDS)NcSsg|]}t|jqSr)rr )r1r3rrr ^sz'ProcessContext.pids..)r6rrrrpids]szProcessContext.pidsNc CsDt|jdkrdStjj|j|d}d}|D]4}|j|}|j|}||j dkr0|}qfq0|dkr|t|jdkS|jD]}| r| |q|j|}|j | r|j|j }|dkrt| j} td|| f||j|| dntd||f||j|d|j |} d |} | | 7} t| ||jdS) a Tries 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$process %d terminated with signal %s)rrrr z'process %d terminated with exit code %d)rrrz5 -- Process %d terminated with the following error: )lenr8multiprocessing connectionwaitkeyspopr6joinexitcodeis_alive terminater5emptyr!Signalsnamerr getr) r r<readyrr0r2r3Zfailed_processrDrIZoriginal_tracerrrrrC`sT         zProcessContext.join)N)rrrr r;rCrrrrr/Usr/cseZdZfddZZS) SpawnContextcstdt||dS)NzZ get_contextrange SimpleQueueProcessr.startappendr/rC) r*r,nprocsrCrQ start_methodmpr5r6r+r-r3contextrrrstart_processess$      r[cCs.|dkrd|}t|t|||||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`` rOzThis method only supports start_method=spawn (got: %s). To use a different start_method use: torch.multiprocessing.start_processes(...))rX)rMrNr[)r*r,rWrCrQrXrrrrrOs! )rrTFrO)rrTFrO)r>Zmultiprocessing.connectionr!r(rMtypingrrr$rrrr.r/rLr[rOrrrrs&   Q