Χg_ 0dZddlZddlZddlZddlmZgdZddlTeejz Zej ddl m Z m Z m Z m Z m Z mZmZejdks ejd krd ad hZnd ad d hZd Zd ZdZdeddfdZdefdZedS)atorch.multiprocessing is a wrapper around the native :mod:`multiprocessing` module. It registers custom reducers, that use shared memory to provide shared views on the same data in different processes. Once the tensor/storage is moved to shared_memory (see :func:`~torch.Tensor.share_memory_`), it will be possible to send it to other processes without making any copies. The API is 100% compatible with the original module - it's enough to change ``import multiprocessing`` to ``import torch.multiprocessing`` to have all the tensors sent through the queues or shared via other mechanisms, moved to shared memory. Because of the similarity of APIs we do not document most of this package contents, and we recommend referring to very good docs of the original module. N)init_reductions)set_sharing_strategyget_sharing_strategyget_all_sharing_strategies)*)ENV_VAR_PARALLEL_STARTProcessContextProcessExitedExceptionProcessRaisedExceptionspawn SpawnContextstart_processesdarwinwin32 file_systemfile_descriptorc |tvsJ|adS)zSet the strategy for sharing CPU tensors. Args: new_strategy (str): Name of the selected strategy. Should be one of the values returned by :func:`get_all_sharing_strategies()`. N)_all_sharing_strategies_sharing_strategy) new_strategys Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/multiprocessing/__init__.pyrr<s" 2 2 2 2 2$ctS)z4Return the current strategy for sharing CPU tensors.)rrrrrHs rctS)zAReturn a set of sharing strategies supported on a current system.)rrrrrrMs ""rnamereturncDtj|dS)zcSet the name of the current thread. Args: name (str): Name of the current thread. N)torch_C_set_thread_name)rs rr"r"Rs  Hd#####rc>tjS)z_Get the name of the current thread. Returns: str: Name of the current thread. )r r!_get_thread_namerrrr$r$[s 8 $ $ & &&r)__doc__multiprocessingsysr reductionsr__all__r!_multiprocessing_initr r r r r rrplatformrrrrrstrr"r$rrrr-s '''''' Y X X? ""  F<8s|w66%,o)0-@ % % % ### $3$4$$$$'#''''r