B \+@sddlmZmZmZmZmZddlmZddlm Z Gddde Z e e ddZ d d Zd d ZGd dde ZeZGddde ZeZGddde ZeZddZddZddZdS))irir_utilstypeserrorssigutils) typeof_impl)find_region_inout_varsc@s,eZdZdZdZddZddZddZd S) WithContextzXA dummy object for use as contextmanager. This can be used as a contextmanager. FcCsdS)N)selfr r 1lib/python3.7/site-packages/numba/withcontexts.py __enter__ szWithContext.__enter__cCsdS)Nr )r typvaltbr r r __exit__szWithContext.__exit__cCstdS)aMutate the *blocks* to implement this contextmanager. Parameters ---------- func_ir : FunctionIR blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-maanger. body_block: sequence[int] A sequence of int's representing labels of the with-body dispatcher_factory : callable A callable that takes a `FunctionIR` and returns a `Dispatcher`. N)NotImplementedError)r func_irblocks blk_startblk_end body_blocksdispatcher_factoryextrar r r mutate_with_bodyszWithContext.mutate_with_bodyN)__name__ __module__ __qualname____doc__ is_callablerrrr r r r r s r cCs t|S)N)rZContextManager)rcr r r typeof_contextmanager$sr"cCs|ds|ddSdS)z.Get parent of the variable given its name $.rN) startswithsplit)namer r r _get_var_parent)s r(cCsx|D] }||=qWdS)z-Remove keys in *to_clear* from *blocks*. Nr )rZto_clearbr r r _clear_blocks2s r*c@seZdZdZddZdS)_ByPassContextTypez_A simple context-manager that tells the compiler to bypass the body of the with-block. c sp|dks t|j}dd|j|Ddd|j|DfddttD} t|||| t||dS)NcSsi|]}|t|qSr )r().0kr r r Bsz7_ByPassContextType.mutate_with_body..cSsi|]}|t|qSr )r()r,r-r r r r.Cscsi|]}||qSr r )r,r-)inmapoutmapr r r.Ds)AssertionErrorvariable_lifetimelivemapfilterbool_bypass_with_contextr*) r rrrrrrrvlt forwardvarsr )r/r0r r=s z#_ByPassContextType.mutate_with_bodyN)rrrrrr r r r r+9sr+c@seZdZdZddZdS)_CallContextTypezqA simple context-manager that tells the compiler to lift the body of the with-block as another function. cs|dks t|j}t|j||t|d\} } fdd|D} t| ||| | |j| t| t| dd} || } t | ||| | }||<t || S)N)rr3callfromreturntobody_block_idscsi|]}||qSr r )r,r-)rr r r.^sz5_CallContextType.mutate_with_body..T)r arg_names arg_countforce_non_generator) r1r2r r3set_mutate_with_block_calleederivetuplelen_mutate_with_block_callerr*)r rrrrrrrr7inputsoutputs lifted_blks lifted_ir dispatchernewblkr )rr rQs,  z!_CallContextType.mutate_with_bodyN)rrrrrr r r r r9Msr9c@s,eZdZdZdZddZddZddZd S) _ObjModeContextTypeaCreates a contextmanager to be used inside jitted functions to enter *object-mode* for using interpreter features. The body of the with-context is lifted into a function that is compiled in *object-mode*. This transformation process is limited and cannot process all possible Python code. However, users can wrap complicated logic in another Python function, which will then be executed by the interpreter. Use this as a function that takes keyword arguments only. The argument names must correspond to the output variables from the with-block. Their respective values are strings representing the expected types. When exiting the with-context, the output variables are cast to the expected nopython types according to the annotation. This process is the same as passing Python objects into arguments of a nopython function. Example:: import numpy as np from numba import njit, objmode def bar(x): # This code is executed by the interpreter. return np.asarray(list(reversed(x.tolist()))) @njit def foo(): x = np.arange(5) y = np.zeros_like(x) with objmode(y='intp[:]'): # annotate return type # this region is executed by object-mode. y += bar(x) return y .. note:: Known limitations: - with-block cannot use incoming list objects. - with-block cannot use incoming function objects. - with-block cannot ``yield``, ``break``, ``return`` or ``raise`` such that the execution will leave the with-block immediately. - with-block cannot contain `with` statements. - random number generator states do not synchronize; i.e. nopython-mode and object-mode uses different RNG states. .. note:: When used outside of no-python mode, the context-manager has no effect. .. warning:: This feature is experimental. The supported features may change with or without notice. TcCs~|dkr iSt|ddkr&td|d}i}xF|D]:\}}t|tjr\t|jtsftdt |j||<q.strip_var_verzkInvalid type annotation on non-outgoing variables: {}.Suggestion: remove annotation of the listed variablesz1missing type annotation on outgoing variables: {}csg|] }|qSr r )r,rU)rTr r sz8_ObjModeContextType.mutate_with_body..csi|]}||qSr r )r,r-)rr r r.sz8_ObjModeContextType.mutate_with_body..T)rr=r>r?)Z objectmodeZ output_types)rVrSr2r r3r@listmaprZ TypingErrorformatrZTuplerArBrCrDrEr*)r rrrrrrrr7rFrGrXZ stripped_outsZextra_annotatedmsgZ not_annotatedZouttuprHrIrJrKr )rrTr rsD z$_ObjModeContextType.mutate_with_bodycOs|S)Nr )r rMrNr r r __call__sz_ObjModeContextType.__call__N)rrrrr rVrr^r r r r rLxs 2;rLc Csz||}|j}|j}tj||d}x6|D]*\}} |tj|||| |dq,W|tj||d|||<dS)zGiven the starting and ending block of the with-context, replaces the head block with a new block that jumps to the end. *blocks* is modified inplace. )scoperS)rQtargetrS)r`rSN) r_rSrBlockrOappendZAssignZ get_exactZJump) rrrr8sblkr_rSrKr-rUr r r r6sr6c Cs:||}|j}|j}tj||d} tj| ||||d| S)akMake a new block that calls into the lifeted with-context. Parameters ---------- dispatcher : Dispatcher blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-manager. inputs: sequence[str] Input variable names outputs: sequence[str] Output variable names )r_rS)newblockZcallee label_nextrFrG)r_rSrrarZfill_block_with_call) rJrrrrFrGrcr_rSrdr r r rEsrEc CsZt|}||}|j}|j}tjtj||d||d||<tjtj||d|d||<dS)aFMutate *blocks* for the callee of a with-context. Parameters ---------- blocks : dict[ir.Block] blk_start, blk_end : int labels of the starting and ending block of the context-manager. inputs: sequence[str] Input variable names outputs: sequence[str] Output variable names )r_rS)blockrFre)rfrGN)minr_rSrZfill_callee_prologuerraZfill_callee_epilogue) rrrrFrGZhead_blkZtemp_blkr_rSr r r rA8s    rAN)ZnumbarrrrrZnumba.typing.typeofrZ transformsr objectr registerr"r(r*r+Zbypass_contextr9Z call_contextrLZobjmode_contextr6rErAr r r r s    (