a .=icI@s@ddlZddlmZddlmZddZGdddejjZdS)N) SimpleQueuecOs&ddl}tjj|i||dS)Nr)gcmultiprocessingpoolZworkerZcollect)argskwargsrr k/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/multiprocessing/pool.py clean_workersr 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 _outqueueZ_writersendZ _quick_put_readerrecvZ _quick_get)selfr r r _setup_queuess zPool._setup_queuescCst|jt|jD]t}|j|j|j|j|jf}t |drF||j f7}|j t |d}|j ||jdd|_d|_|tdqdS)zBring the number of pool processes up to the specified number, for use after reaping workers which have exited. _wrap_exception)targetrProcessZ PoolWorkerTz added workerN)rangeZ _processeslen_poolr rZ _initializerZ _initargsZ_maxtasksperchildhasattrrrr appendnamereplacedaemonstartutildebug)rirwr r r _repopulate_pools   zPool._repopulate_poolN)__name__ __module__ __qualname____doc__rr$r r r r r sr ) Zmultiprocessing.poolrZmultiprocessing.utilr queuerr rr r r r r s