U ,d}\@sJdZddlmZddlmZddlZddlmZddlm Z m Z ddl m Z m Z edd d gZed d Zed ZeZddZedZedZedZedZdZeeZGdddeZGdddeZGdddeZee ee gZe eZ!GdddeZ"GdddeZ#Gddde#Z$Gd d!d!e#Z%e$e%d"Z&d#d$Z'dS)%z3 Calling conventions for Numba-compiled functions. ) namedtuple)IterableN)ir)typescgutils)PYOBJECTGENERIC_POINTER TryStatusin_tryexcinfoStatus)codeis_okis_noneis_erroris_stop_iteration is_python_exc is_user_exc excinfoptr cCs tt|SN)rConstant errcode_t)r r7/tmp/pip-unpacked-wheel-eu7e0c37/numba/core/callconv.py _const_int'src@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS) BaseCallConvcCs ||_dSr)context)selfr!rrr__init__9szBaseCallConv.__init__c Cs|tjkr||n||kr|jj|||d}t||j}||&|j ||j |j }| ||W5QRX||n\t |tjs||j kr|jj||||j d}|j ||j |}| ||ntd||dS)N)value)ZfromtyZtotyzreturning {0} for {1})rnonereturn_native_noner!Z make_helperrZ as_bool_bitZvalidif_thenZget_return_valuetypedata return_value isinstanceOptionalcastNotImplementedErrorformat)r"builderrettyZvaltyr$optvalZvalidbitretvalrrrreturn_optional_value<s*        z"BaseCallConv.return_optional_valuecCs||tdSr)_return_errcode_raw RETCODE_NONEr"r0rrrr&YszBaseCallConv.return_native_nonecCs|j|tdddSNTmark_exc)r5 RETCODE_EXCr7rrr return_exc\szBaseCallConv.return_exccCs||tdSr)r5RETCODE_STOPITr7rrrreturn_stop_iteration_sz"BaseCallConv.return_stop_iterationcCs|jj|}|S)zQ Get the actual type of the return argument for Numba type *ty*. )r!data_model_managerget_return_typeZ as_pointer)r"tyrestyperrrr@bszBaseCallConv.get_return_typecCs||}||_|S)zS Initialize and return a call helper object for the given builder. )_make_call_helper_BaseCallConv__call_helper)r"r0chrrrinit_call_helperis zBaseCallConv.init_call_helpercCs|jSr)rDr7rrr_get_call_helperqszBaseCallConv._get_call_helperc Cs|j}||jN|||j}t|t ||| |W5QRX| |W5QRX||j | d| |W5QRX||j| |W5QRX|dd| |||dS)zT Given a non-ok *status*, raise the corresponding Python exception. ZPyExc_StopIterationZPyExc_SystemErrorz*unknown error when calling native functionN)functionZappend_basic_blockr'rZ err_clearZ unserializerrZ if_likelyZ is_not_nullZ raise_objectbranchrZ err_set_nonerZerr_set_stringZposition_at_end)r"r0apistatusZbbendexcrrr raise_errorts&     zBaseCallConv.raise_errorcCs ||}||}|||S)z Get the decoded (unpacked) Python arguments with *argtypes* from LLVM function *func*. A tuple of LLVM values is returned. ) get_arguments_get_arg_packerZfrom_arguments)r"r0argtypesfuncZraw_argsarginforrrdecode_argumentss  zBaseCallConv.decode_argumentscCs |j|S)zF Get an argument packer for the given argument types. )r!Zget_arg_packer)r"rPrrrrOszBaseCallConv._get_arg_packerN)__name__ __module__ __qualname__r#r4r&r<r>r@rFrGrMrSrOrrrrr 7s r c@sfeZdZdZddZddZdddZd d Zdd d ZddZ ddZ dddZ ddZ ddZ dS)MinimalCallConva A minimal calling convention, suitable for e.g. GPU targets. The implemented function signature is: retcode_t (*, ... ) The return code will be one of the RETCODE_* constants or a function-specific user exception id (>= RETCODE_USEREXC). Caller is responsible for allocating a slot for the return value (passed as a pointer in the first argument). cCstSr)_MinimalCallHelperr7rrrrCsz!MinimalCallConv._make_call_helpercCsP|jjd}|j|jjks4tt|jt|jjf|||||tdSNr) rHargsr(pointeeAssertionErrorstrstorer5 RETCODE_OKr"r0r3retptrrrrr*s   zMinimalCallConv.return_valueNc Cs|dk r t|ts td|f|dk r@t|ts@td|f|dk rx|}|dkr\|}||j|jf}d|kr|d}nd}||}| |||} |j |t | dddS)N-exc should be None or exception class, got %r(exc_args should be None or tuple, got %rTr9) issubclass BaseException TypeErrorr+tuple_raw_function_namefilenamelinerG_add_exceptionr5r) r"r0rLexc_argsloc func_namefnamelocinfoZ call_helperexc_idrrrreturn_user_excs& zMinimalCallConv.return_user_exccCs|||jdSr)r5r )r"r0rKrrrreturn_status_propagatesz'MinimalCallConv.return_status_propagateFcCs t|trt|}||dSr)r+intrret)r"r0r r:rrrr5s z#MinimalCallConv._return_errcode_rawc Csx|d|t}|d|t}|||}||}|d|t}|d|t}|d|t} t|||||| |dd} | S)z? Given a return *code*, get a Status instance. ==>=Nr rrrrrrr) icmp_signedr_r6or_not_r;r=RETCODE_USEREXCr ) r"r0r normr%okerrrLrrrKrrr_get_return_statuss"  z"MinimalCallConv._get_return_statuscCs4||}t|j}||}tt|g|}|SzQ Get the implemented Function type for *restype* and *argtypes*. )rOlistargument_typesr@r FunctionTyperr"rBrPrRZresptrZfntyrrrget_function_types    z!MinimalCallConv.get_function_typecCs>|rt||}|||dd|Dd|jd_|S)zA Set names and attributes of function arguments. cSsg|] }d|qSzarg.r.0arrr sz5MinimalCallConv.decorate_function..z.retr)r\rO assign_namesrNrZname)r"fnrZ fe_argtypesnoaliasrRrrrdecorate_functions    z!MinimalCallConv.decorate_functioncCs|jddS)@ Get the Python-level arguments of LLVM *func*. rNrZr"rQrrrrNszMinimalCallConv.get_argumentscCs|jdjj}t||}|t||||}|||}|gt |} | || } | || } | |} |j ||| } | | fS)z3 Call the Numba-compiled *callee*. r)rZr(r[r alloca_oncer^get_null_valuerO as_argumentsrcallrloadr!get_returned_value)r"r0calleerestyargtysrZr1 retvaltmprRrealargsr rKr3outrrr call_function s      zMinimalCallConv.call_function)NNN)F)F)rTrUrV__doc__rCr*rrrsr5rrrrNrrrrrrWs    rWc@s(eZdZdZddZddZddZdS) rXz A call helper object for the "minimal" calling convention. User exceptions are represented as integer codes and stored in a mapping for retrieval from the caller. cCs i|_dSr) exceptionsr"rrrr#'sz_MinimalCallHelper.__init__cCs"t|jt}|||f|j|<|S)aV Add a new user exception to this helper. Returns an integer that can be used to refer to the added exception in future. Parameters ---------- exc : exception type exc_args : None or tuple exception args locinfo : tuple location information )lenr FIRST_USEREXC)r"rLrlrprqrrrrk*sz!_MinimalCallHelper._add_exceptioncCsFz |j|WStk r@d|}t}|f}d}|||fYSXdS)z Get information about a user exception. Returns a tuple of (exception type, exception args, location information). Parameters ---------- id : integer The ID of the exception to look up z#unknown error %d in native functionN)rKeyError SystemError)r"rqmsgrLrlrprrr get_exception<s  z _MinimalCallHelper.get_exceptionN)rTrUrVrr#rkrrrrrrX srXc@seZdZdZedZddZddZd'dd Z d(d d Z d d Z ddZ ddZ ddZddZd)ddZddZddZd*ddZdd Zd!d"Zd#d$Zd+d%d&ZdS), CPUCallConva The calling convention for CPU targets. The implemented function signature is: retcode_t (*, excinfo **, ... ) The return code will be one of the RETCODE_* constants. If RETCODE_USEREXC, the exception info pointer will be filled with a pointer to a constant struct describing the raised exception. Caller is responsible for allocating slots for the return value and the exception info pointer (passed as first and second arguments, respectively). rcCsdSrrr7rrrrCfszCPUCallConv._make_call_helpercCsP||j}|j|jjks4tt|jt|jjf|||||tdSr) _get_return_argumentrHr(r[r\r]r^r5r_r`rrrr*is   zCPUCallConv.return_valueNc Cs|dk r t|ts td|f|dk r@t|ts@td|f|dkrNt}|j|}|dk r|}|dkrv|}||j|j f}d|krd}nd}|||f}| |} | |j } | | | dS)Nrbrc)rdrerfr+rgr!Zget_python_apirhrirjZserialize_object_get_excinfo_argumentrHr^) r"r0rLrlrmrnZpyapirorpZ struct_gvexcptrrrrset_static_user_excps.    zCPUCallConv.set_static_user_exccCsRt|dd}|j|||||d||}|r>||dn|j|tdddS)NZ _in_try_blockF)rlrmrntargetTr9)getattrrcheck_try_statusrIr5r|)r"r0rLrlrmrnZtry_info trystatusrrrrrs   zCPUCallConv.return_user_excc Cs@z|jWStk r:tj|tjddd}||_|YSXdS)NZ try_stateT)rzfill)Z_CPUCallConv__eh_try_stateAttributeErrorrrZintp_t)r"r0ptrrrr_get_try_stateszCPUCallConv._get_try_statecCsJ||}||}|d||d}||j}||}t||dS)N>r)r r )rrZ icmp_unsignedr(rrHr )r"r0 try_state_ptrZ try_depthr rr rrrrs     zCPUCallConv.check_try_statuscCs6||}||}|||d}|||dSNr)rraddr(r^)r"r0roldnewrrrset_try_statuss  zCPUCallConv.set_try_statuscCs\||}||}|||d}|||||j}t|jj }|||dSr) rrsubr(r^rrHrrr[)r"r0rrrrnullrrrunset_try_statuss    zCPUCallConv.unset_try_statusc CsX||}||j}||j||||j|j||j ddW5QRXdSr8) rrrHr^rr'r{r r5r )r"r0rKrrrrrrss   z#CPUCallConv.return_status_propagateFcCs6||}|r2|jtddg}|d|dS)NrZ ret_is_raise)rumodule add_metadatarIntTypeZ set_metadata)r"r0r r:ruZmdrrrr5s zCPUCallConv._return_errcode_rawc Cs|d|t}|d|t}|d|t}|d|t}|||}||} |d|t} || |t t t j }t ||| ||| ||d} | S)zP Given a return *code* and *excinfoptr*, get a Status instance. rvrwrx)ryr_r6r;r=rzr{r|selectrr excinfo_ptr_t Undefinedr ) r"r0r rr}r%rLrr~rrrKrrrrs(   zCPUCallConv._get_return_statuscCs<||}t|j}||}tt|ttg|}|Sr) rOrrr@rrr PointerTyperrrrrrs    zCPUCallConv.get_function_typec s|}||dd|D|}d|_|d|d|}d|_|d|d|r|}|D]&}t|jt j r|d|dqfdd} t t | |} | r|j d } | |j |g|S) zU Set names of function arguments, and add useful attributes to them. cSsg|] }d|qSrrrrrrrsz1CPUCallConv.decorate_function..raZ nocapturerr cs(t|tjs$jj}||r$dSdSNTF)r+rArrayr!r?Zcontains_nrt_meminfo)rAZdmmrrrtype_may_always_need_nrts   z?CPUCallConv.decorate_function..type_may_always_need_nrtZnumba_args_may_always_need_nrt)rOrrNrrZ add_attributerr+r(rranymaprZadd_named_metadatarr) r"rrZrrrRZretargZexcargrrZargs_may_always_need_nrtZnmdrrrrs8             zCPUCallConv.decorate_functioncCs|jddS)rNrrrrrrN&szCPUCallConv.get_argumentscCs |jdSrYrrrrrr,sz CPUCallConv._get_return_argumentcCs |jdSrrrrrrr/sz!CPUCallConv._get_excinfo_argumentcCs||jj}t||}|t||tj|tt dd} | |} t | ||}|| g|} |dkrvd} n&t |trt |tst|} ntd|j|| | d} ||| || }||}|j|||}||fS)aU Call the Numba-compiled *callee*. Parameters: ----------- attrs: LLVM style string or iterable of individual attributes, default is None which specifies no attributes. Examples: LLVM style string: "noinline fast" Equivalent iterable: ("noinline", "fast") r )rNrz,attrs must be an iterable of strings or None)attrs)rZ function_typer[rrr^rrr excinfo_trOrrr+rr]rgrfrrrr!r)r"r0rrrrZrr1rrrRr_attrsr rKr3rrrrr2s*      zCPUCallConv.call_function)NNN)NNN)F)F)N)rTrUrVr itertoolscountZ _status_idsrCr*rrrrrrrrsr5rrrrNrrrrrrrrUs0        -rc@seZdZddZdddZdS) ErrorModelcCs ||_dSr) call_conv)r"rrrrr#^szErrorModel.__init__NcCs$|jr|j|t||dSdSdSr)raise_on_fp_zero_divisionrrrZeroDivisionError)r"r0rlrmrrrfp_zero_divisionas  zErrorModel.fp_zero_division)NN)rTrUrVr#rrrrrr\src@seZdZdZdZdS)PythonErrorModelzL The Python error model. Any invalid FP input raises an exception. TNrTrUrVrrrrrrrjsrc@seZdZdZdZdS)NumpyErrorModela6 In the Numpy error model, floating-point errors don't raise an exception. The FPU exception state is inspected by Numpy at the end of a ufunc's execution and a warning is raised if appropriate. Note there's no easy way to set the FPU exception state from LLVM. Instructions known to set an FP exception can be optimized away: https://llvm.org/bugs/show_bug.cgi?id=6050 http://lists.llvm.org/pipermail/llvm-dev/2014-September/076918.html http://lists.llvm.org/pipermail/llvm-commits/Week-of-Mon-20140929/237997.html FNrrrrrrqs r)pythonZnumpycCst||jS)zF Create an error model instance for the given target context. ) error_modelsr)Z model_namer!rrrcreate_error_modelsr)(r collectionsrcollections.abcrrZllvmliterZ numba.corerrZnumba.core.baserrr r rZint32_trrr_r;r6r=rr|objectr rWrXZLiteralStructTyperrrrrrrrrrrrrsD    k~1