o 8ήc}\@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 r:/tmp/pip-target-vg8gfxp4/lib/python/numba/core/callconv.py _const_int's rc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS) BaseCallConvcC ||_dSr)context)selfr"rrr__init__9 zBaseCallConv.__init__cCs|tjkr ||dS||krL|jj|||d}t||j}|||j ||j |j }| ||Wdn1s@wY||dSt |tjss||j krb|jj||||j d}|j ||j |}| ||dStd||)N)value)fromtytotyzreturning {0} for {1})rnonereturn_native_noner" make_helperr as_bool_bitvalidif_thenget_return_valuetypedata return_value isinstanceOptionalcastNotImplementedErrorformat)r#builderrettyvaltyr&optvalvalidbitretvalrrrreturn_optional_value<s,      z"BaseCallConv.return_optional_valuecC||tdSr)_return_errcode_raw RETCODE_NONEr#r8rrrr*YzBaseCallConv.return_native_nonecCs|j|tdddSNTmark_exc)r@ RETCODE_EXCrBrrr return_exc\szBaseCallConv.return_exccCr?r)r@RETCODE_STOPITrBrrrreturn_stop_iteration_rCz"BaseCallConv.return_stop_iterationcCs|jj|}|S)zQ Get the actual type of the return argument for Numba type *ty*. )r"data_model_managerget_return_type as_pointer)r#tyrestyperrrrLbszBaseCallConv.get_return_typecCs||}||_|S)zS Initialize and return a call helper object for the given builder. )_make_call_helper_BaseCallConv__call_helper)r#r8chrrrinit_call_helperis zBaseCallConv.init_call_helpercCs|jSr)rQrBrrr_get_call_helperqzBaseCallConv._get_call_helperc Cs,|j}||j6|||j}t|t || | |Wdn1s0wY| |Wdn1sDwY||j | d| |Wdn1sdwY||j | |Wdn1swY|dd| |||dS)zT Given a non-ok *status*, raise the corresponding Python exception. NPyExc_StopIterationPyExc_SystemErrorz*unknown error when calling native function)functionappend_basic_blockr.r err_clear unserializerr if_likely is_not_null raise_objectbranchr err_set_nonererr_set_stringposition_at_end)r#r8apistatusbbendexcrrr 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_packerfrom_arguments)r#r8argtypesfuncraw_argsarginforrrdecode_argumentss   zBaseCallConv.decode_argumentscCs |j|S)zF Get an argument packer for the given argument types. )r"get_arg_packer)r#rkrrrris zBaseCallConv._get_arg_packerN)__name__ __module__ __qualname__r$r>r*rHrJrLrSrTrgrorirrrrr 7s r c@sjeZdZdZddZddZ  dddZd 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)_MinimalCallHelperrBrrrrPrUz!MinimalCallConv._make_call_helpercCsP|jjd}|j|jjksJt|jt|jjf|||||tdSNr)rXargsr0pointeestrstorer@ RETCODE_OKr#r8r=retptrrrrr2  zMinimalCallConv.return_valueNc Cs|durt|tstd|f|dur t|ts td|f|dur<|}|dur.|}||j|jf}d|vr;d}nd}||}| |||} |j |t | dddS)N-exc should be None or exception class, got %r(exc_args should be None or tuple, got %rTrE) issubclass BaseException TypeErrorr3tuple_raw_function_namefilenamelinerT_add_exceptionr@r) r#r8rfexc_argsloc func_namefnamelocinfo call_helperexc_idrrrreturn_user_excs( zMinimalCallConv.return_user_exccCs|||jdSr)r@r )r#r8rdrrrreturn_status_propagatesz'MinimalCallConv.return_status_propagateFcCs t|tr t|}||dSr)r3intrret)r#r8r rFrrrr@s 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{rAor_not_rGrIRETCODE_USEREXCr ) r#r8r normr)okerrrfrrrdrrr_get_return_statuss"  z"MinimalCallConv._get_return_statuscCs4||}t|j}||}tt|g|}|SzQ Get the implemented Function type for *restype* and *argtypes*. )rilistargument_typesrLr FunctionTyperr#rOrkrnresptrfntyrrrget_function_types   z!MinimalCallConv.get_function_typecCs>|rJ||}|||dd|Dd|jd_|S)zA Set names and attributes of function arguments. cSg|]}d|qSzarg.r.0arrr z5MinimalCallConv.decorate_function..z.retr)ri assign_namesrhrwname)r#fnrw fe_argtypesnoaliasrnrrrdecorate_functions    z!MinimalCallConv.decorate_functioncC|jddS)@ Get the Python-level arguments of LLVM *func*. rNrwr#rlrrrrhzMinimalCallConv.get_argumentscCs|jdjj}t||}|t||||}|||}|gt |} | || } | || } | |} |j ||| } | | fS)z3 Call the Numba-compiled *callee*. r)rwr0rxr alloca_oncerzget_null_valueri as_argumentsrcallrloadr"get_returned_value)r#r8calleerestyargtysrwr9 retvaltmprnrealargsr rdr=outrrr call_function s      zMinimalCallConv.call_functionNNNF)rqrrrs__doc__rPr2rrr@rrrrhrrrrrrts     rtc@s(eZdZdZddZddZddZdS) ruz 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$'r%z_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#rfrrrrrrr*sz!_MinimalCallHelper._add_exceptioncCs@z|j|WStyd|}t}|f}d}|||fYSw)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#rmsgrfrrrrr get_exception<s  z _MinimalCallHelper.get_exceptionN)rqrrrsrr$rrrrrrru s  ruc@seZdZdZedZddZddZ  d'dd Z  d'd d Z d d Z ddZ ddZ ddZddZd(ddZddZddZd(ddZdd Zd!d"Zd#d$Z d)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). rcCsdSrrrBrrrrPfszCPUCallConv._make_call_helpercCsP||j}|j|jjksJt|jt|jjf|||||tdSr)_get_return_argumentrXr0rxryrzr@r{r|rrrr2ir~zCPUCallConv.return_valueNc Cs|durt|tstd|f|dur t|ts td|f|dur't}|j|}|durI|}|dur;|}||j|j f}d|vrHd}nd}|||f}| |} | |j } | | | dS)Nrr)rrrr3rr"get_python_apirrrserialize_object_get_excinfo_argumentrXrz) r#r8rfrrrpyapirr struct_gvexcptrrrrset_static_user_excps0    zCPUCallConv.set_static_user_exccCsTt|dd}|j|||||d||}|r ||ddS|j|tdddS)N _in_try_blockF)rrrtargetTrE)getattrrcheck_try_statusr_r@r)r#r8rfrrrtry_info trystatusrrrrs   zCPUCallConv.return_user_excc Cs:z|jWStytj|tjddd}||_|YSw)N try_stateT)rzfill)_CPUCallConv__eh_try_stateAttributeErrorrrintp_t)r#r8ptrrrr_get_try_states  zCPUCallConv._get_try_statecCsJ||}||}|d||d}||j}||}t||dS)N>r)r r )rr icmp_unsignedr0rrXr )r#r8 try_state_ptr try_depthr rr rrrrs     zCPUCallConv.check_try_statuscCs6||}||}|||d}|||dSNr)rraddr0rz)r#r8roldnewrrrset_try_statuss  zCPUCallConv.set_try_statuscCs\||}||}|||d}|||||j}t|jj }|||dSr) rrsubr0rzrrXrrrx)r#r8rrrrnullrrrunset_try_statuss    zCPUCallConv.unset_try_statuscCsn||}||j}||j||||j|j||j ddWddS1s0wYdSrD) rrrXrzrr.rr r@r )r#r8rdrrrrrrs  "z#CPUCallConv.return_status_propagateFcCs:||}|r|jtddg}|d|dSdS)Nr ret_is_raise)rmodule add_metadatarIntType set_metadata)r#r8r rFrmdrrrr@s 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. rrr)rr{rArGrIrrrselectrr excinfo_ptr_t Undefinedr ) r#r8r rrr)rfrrrrrdrrrrs(   zCPUCallConv._get_return_statuscCs<||}t|j}||}tt|ttg|}|Sr) rirrrLrrr PointerTyperrrrrrs    zCPUCallConv.get_function_typec s|}||dd|D|}d|_|d|d|}d|_|d|d|rT|}|D]}t|jt j rS|d|dq@fdd} t t | |} | rs|j d } | |j |g|S) zU Set names of function arguments, and add useful attributes to them. cSrrrrrrrrrz1CPUCallConv.decorate_function..r} nocapturerr cs(t|tjsjj}||rdSdSNTF)r3rArrayr"rKcontains_nrt_meminfo)rNdmmrrrtype_may_always_need_nrts  z?CPUCallConv.decorate_function..type_may_always_need_nrtnumba_args_may_always_need_nrt)rirrhrr add_attributerr3r0rranymapradd_named_metadatarr) r#rrwrrrnretargexcargrrargs_may_always_need_nrtnmdrrrrs:             zCPUCallConv.decorate_functioncCr)rNrrrrrrh&rzCPUCallConv.get_argumentscC |jdSrvrrrrrr,r%z CPUCallConv._get_return_argumentcCr rrrrrrr/r%z!CPUCallConv._get_excinfo_argumentcCs||jj}t||}|t||tj|tt dd} | |} t | ||}|| g|} |dur;d} nt |trJt |tsJt|} 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)r function_typerxrrrzrrr excinfo_trirrr3rryrrrrrr"r)r#r8rrrrwrr9rrrnr_attrsr rdr=rrrrr2s*     zCPUCallConv.call_functionrrr)rqrrrsr itertoolscount _status_idsrPr2rrrrrrrr@rrrrhrrrrrrrrUs2        -rc@seZdZddZdddZdS) ErrorModelcCr!r) call_conv)r#rrrrr$^r%zErrorModel.__init__NcCs |jr|j|t||dSdSr)raise_on_fp_zero_divisionrrZeroDivisionError)r#r8rrrrrfp_zero_divisionas  zErrorModel.fp_zero_division)NN)rqrrrsr$rrrrrr\src@eZdZdZdZdS)PythonErrorModelzL The Python error model. Any invalid FP input raises an exception. TNrqrrrsrrrrrrrjsrc@r)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)pythonnumpycCst||jS)zF Create an error model instance for the given target context. ) error_modelsr) model_namer"rrrcreate_error_modelrr")(r collectionsrcollections.abcrrllvmliter numba.corerrnumba.core.baserrr r rint32_trrr{rGrArIrrobjectr rtruLiteralStructTyperrrrrrrr r"rrrrsF    k~1