Χg\ddlZddlmZddlmZdZGddejjZdS)N) SimpleQueuecbddl}tjj|i||dS)Nr)gcmultiprocessingpoolworkercollect)argskwargsrs V/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/multiprocessing/pool.py clean_workerrs7 III0000JJLLLLLceZdZdZdZdZdS)PoolzPool implementation which uses our version of SimpleQueue. This lets us pass tensors in shared memory across processes instead of serializing the underlying data. ct|_t|_|jjj|_|jjj|_dS)N) r_inqueue _outqueue_writersend _quick_put_readerrecv _quick_get)selfs r _setup_queueszPool._setup_queuess:#  $-/4.05rct|jt|jz D]}|j|j|j|j|jf}t|dr ||j fz }| t|}|j ||jdd|_d|_|t%jddS)zIncrease the number of pool processes to the specified number. Bring the number of pool processes up to the specified number, for use after reaping workers which have exited. _wrap_exception)targetr Process PoolWorkerTz added workerN)range _processeslen_poolrr _initializer _initargs_maxtasksperchildhasattrrr rappendnamereplacedaemonstartutildebug)rir ws r _repopulate_poolzPool._repopulate_pools tTZ899 ' 'A !& Dt.// 0-// Lt <r<s~#######'#'#'#'#'?  $#'#'#'#'#'r