a 8Sic!@sddlmZddlZddlZddlZddlZddlZddlmZGddde Z Gddde Z Gd d d e Z d d Z Gd ddZGdddeZdddZdddZdS))OptionalN)_prctl_pr_set_pdeathsigcs6eZdZddgZeeedfdd ZddZZS)ProcessException error_index error_pid)msgrpidcs"t|||_||_||_dSN)super__init__rrr )selfrrr  __class__W/var/www/html/django/DPS/env/lib/python3.9/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 rrrrr6sz!ProcessExitedException.__reduce__)N) rrrrrrrrr rrrrrrr'srcCsfttjz||g|RWnBty0Yn2ty`ddl}||t dYn0dS)Nrr) rsignalSIGINTKeyboardInterrupt Exception tracebackput format_excsysexit)fniargs error_queuer%rrr_wrap=s   r.c@s&eZdZddZddZdddZdS) ProcessContextcCs$||_||_ddt|D|_dS)NcSsi|]\}}|j|qSr)sentinel).0indexprocessrrr Ssz+ProcessContext.__init__..) error_queues processes enumerate sentinelsr r6r5rrrr Ps zProcessContext.__init__cCsdd|jDS)NcSsg|]}t|jqSr)rr )r1r3rrr Yz'ProcessContext.pids..)r6rrrrpidsXszProcessContext.pidsNc CsDt|jdkrdStjj|j|d}d}|D]4}|j|}|j|}||j dkr0|}qfq0|dur|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_processrErJZoriginal_tracerrrrrD[s\        zProcessContext.join)N)rrrr r<rDrrrrr/Osr/cseZdZfddZZS) SpawnContextcs tdtt|||dS)Nzs   T