o h@s@ddlZddlmZddlmZddZGdddejjZdS)N) SimpleQueuecOs&ddl}tjj|i||dS)Nr)gcmultiprocessingpoolworkercollect)argskwargsrr n/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/multiprocessing/pool.py clean_workers r c@s eZdZdZddZddZdS)PoolzPool implementation which uses our version of SimpleQueue. This lets us pass tensors in shared memory across processes instead of serializing the underlying data. cCs,t|_t|_|jjj|_|jjj|_dS)N) r_inqueue _outqueue_writersend _quick_put_readerrecv _quick_get)selfr r r _setup_queuess zPool._setup_queuescCst|jt|jD]:}|j|j|j|j|jf}t |dr#||j f7}|j t |d}|j ||jdd|_d|_|tdq dS)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_maxtasksperchildhasattrrrr appendnamereplacedaemonstartutildebug)r_r wr r r _repopulate_pools     zPool._repopulate_poolN)__name__ __module__ __qualname____doc__rr.r r r r rs r) multiprocessing.poolrmultiprocessing.utilr*queuerr rrr r r r s