gdZddlmZddlZddlmZddlZddlmZddl m Z ddl m Z dZ e jZejd d(ejd dd kZn e Zejd  d ejd <ejd eejd Zn$ede dZdZdddZdiefdZdS)zi Routines for running Python functions in parallel using process pools from the multiprocessing library. ) annotationsN)ProcessPoolExecutor) QiskitError)local_hardware_info) user_configcVtjdkrd}ntjdkrd}nd}|S)z Returns the default parallelism flag value for the current platform. Returns: parallel_default: The default parallelism flag value for the current platform. win32FdarwinT)sysplatform)parallel_defaults K/granary/edev/quantum/lib/python3.11/site-packages/qiskit/utils/parallel.pyget_platform_parallel_defaultr>s> |w   ! !   QISKIT_PARALLELtrueQISKIT_IN_PARALLELFALSEQISKIT_NUM_PROCS num_processcpusc(|\}}}}||g|Ri|SN)paramtaskvalue task_args task_kwargss r _task_wrapperr es0,1)T5)[ 4 1 1 1 1[ 1 11r num_processes int | Nonereturnboolc|tn|}|dko8tjdddkotdt S)aReturn whether the current parallelisation configuration suggests that we should run things like :func:`parallel_map` in parallel (``True``) or degrade to serial (``False``). Args: num_processes: the number of processes requested for use (if given). Nrrparallel_enabled) CPU_COUNTosgetenvCONFIGgetPARALLEL_DEFAULT)r!s rshould_run_in_parallelr.jsQ"/!6IIMM = I*G 4 4 ? = JJ)+; < <rrc|t}t|dkrgSt|dkr|dgRigSt|rdtjd< g}t |5}fd|D}|t|}dddn #1swxYwYt|}n^#ttf$rJ} t| trdtjd<td | dtjd<| d} ~ wwxYwdtjd<|Sg}t|D]&\} } | gRi} || '|S) a Parallel execution of a mapping of `values` to the function `task`. This is functionally equivalent to:: result = [task(value, *task_args, **task_kwargs) for value in values] This will parallelise the results if the number of ``values`` is greater than one, and the current system configuration permits parallelization. Args: task (func): Function that is to be called for each value in ``values``. values (array_like): List or array of values for which the ``task`` function is to be evaluated. task_args (list): Optional additional arguments to the ``task`` function. task_kwargs (dict): Optional additional keyword argument to the ``task`` function. num_processes (int): Number of processes to spawn. Returns: result: The result list contains the value of ``task(value, *task_args, **task_kwargs)`` for each value in ``values``. Raises: QiskitError: If user interrupts via keyboard. Examples: .. code-block:: python import time from qiskit.utils import parallel_map def func(_): time.sleep(0.1) return 0 parallel_map(func, list(range(10))); Nrr&TRUEr) max_workersc3$K|] }|fV dSrr).0rrrrs r zparallel_map..s,SS5$y+>SSSSSSrrz#Keyboard interrupt in parallel_map.)r(lenr.r)environrmapr listKeyboardInterrupt Exception isinstancer enumerateappend) rvaluesrrr!resultsexecutorrfutureerror_rresults ` `` r parallel_maprEysL!  6{{a  6{{aVAY::::k::;;m,,+1 '( G$??? <8SSSSSSFSSS!mU;; < < < < < < < < < < < < < < <6llGG!9-   %!233 T3: /0!"GHHeS/6BJ+ ,K  ,3 '(Gf%%5e7i777;77v Ns='C 9,B1% C 1B55C 8B59C D'AD""D'r)r!r"r#r$)__doc__ __future__rr)concurrent.futuresrr qiskit.exceptionsrqiskit.utils.multiprocessingrqiskitrr get_configr+r*lowerr-r6intr(r,r r.rErrrrOs\ #""""" 222222 ))))))<<<<<<,   ! ! 29 %%1 ry!2D99??AAVK4466 29 !""*'.BJ#$ 29   ,IBI01122II =*=*=*?*?*GHHI222      BiFFFFFFr