o 8ήc,@sdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZeeZdZdidddfdd Zd d Z  dd d ZddZiddfddZddZdS)z% Define @jit and related decorators. N)DeprecationErrorNumbaDeprecationWarning)stencil)config extendingsigutilsregistryz`Deprecated keyword argument `{0}`. Signatures should be passed as the first positional argument.Fc Ksd|vr ttdd|vrttd|ddr&|ddr&tdd|vr0|d|d<|dd }||d <|d urCd }d }nt|trMd }|}nt |rXd }|g}n|}d }i} |d urf|| d <t |f||||d | } |d ur{| |S| S)a This decorator is used to compile a Python function into native code. Args ----- signature_or_function: The (optional) signature or list of signatures to be compiled. If not passed, required signatures will be compiled when the decorated function is called, depending on the argument values. As a convenience, you can directly pass the function to be compiled instead. locals: dict Mapping of local variable names to Numba types. Used to override the types deduced by Numba's type inference engine. pipeline_class: type numba.compiler.CompilerBase The compiler pipeline type for customizing the compilation stages. options: For a cpu target, valid options are: nopython: bool Set to True to disable the use of PyObjects and Python API calls. The default behavior is to allow the use of PyObjects and Python API. Default value is False. forceobj: bool Set to True to force the use of PyObjects for every value. Default value is False. looplift: bool Set to True to enable jitting loops in nopython mode while leaving surrounding code in object mode. This allows functions to allocate NumPy arrays and use Python objects, while the tight loops in the function can still be compiled in nopython mode. Any arrays that the tight loop uses should be created before the loop is entered. Default value is True. error_model: str The error-model affects divide-by-zero behavior. Valid values are 'python' and 'numpy'. The 'python' model raises exception. The 'numpy' model sets the result to *+/-inf* or *nan*. Default value is 'python'. inline: str or callable The inline option will determine whether a function is inlined at into its caller if called. String options are 'never' (default) which will never inline, and 'always', which will always inline. If a callable is provided it will be called with the call expression node that is requesting inlining, the caller's IR and callee's IR as arguments, it is expected to return Truthy as to whether to inline. NOTE: This inlining is performed at the Numba IR level and is in no way related to LLVM inlining. boundscheck: bool or None Set to True to enable bounds checking for array indices. Out of bounds accesses will raise IndexError. The default is to not do bounds checking. If False, bounds checking is disabled, out of bounds accesses can produce garbage results or segfaults. However, enabling bounds checking will slow down typical functions, so it is recommended to only use this flag for debugging. You can also set the NUMBA_BOUNDSCHECK environment variable to 0 or 1 to globally override this flag. The default value is None, which under normal execution equates to False, but if debug is set to True then bounds checking will be enabled. Returns -------- A callable usable as a compiled function. Actual compiling will be done lazily if no explicit signatures are passed. Examples -------- The function can be used in the following ways: 1) jit(signatures, **targetoptions) -> jit(function) Equivalent to: d = dispatcher(function, targetoptions) for signature in signatures: d.compile(signature) Create a dispatcher object for a python function. Then, compile the function with the given signature(s). Example: @jit("int32(int32, int32)") def foo(x, y): return x + y @jit(["int32(int32, int32)", "float32(float32, float32)"]) def bar(x, y): return x + y 2) jit(function, **targetoptions) -> dispatcher Create a dispatcher function object that specializes at call site. Examples: @jit def foo(x, y): return x + y @jit(nopython=True) def bar(x, y): return x + y argtypesrestypenopythonFforceobjz1Only one of 'nopython' or 'forceobj' can be True._targettarget_backendcpu boundscheckNpipeline_class)localstargetcache targetoptions) r_msg_deprecated_signature_argformatget ValueErrorpop isinstancelistr is_signature_jit) signature_or_functionrrrroptionsrpyfuncsigsdispatcher_argswrapperr%.wrapper)numba.core.target_extensionr()r"rrrrr#r(r$r%r=r&rs "rc KsDi}|dur ||d<tddid||dd|}|dur ||S|S)a This decorator allows flexible type-based compilation of a jitted function. It works as `@jit`, except that the decorated function is called at compile-time with the *types* of the arguments and should return an implementation function for those types. Nrr generated)r"rrrr impl_kindr%)r)functionrrr r#r$r%r%r& generated_jits rCcOsJd|vr tdtd|vrtdt|d=|ddit|i|S)zr Equivalent to jit(nopython=True) See documentation for jit function/decorator for full description. r z'nopython is set for njit and is ignoredr z'forceobj is set for njit and is ignoredT)warningswarnRuntimeWarningupdater')argskwsr%r%r&njits  rJc s"tfdd}|S)z This decorator is used to compile a Python function into a C callback usable with foreign C libraries. Usage:: @cfunc("float64(float64, float64)", nopython=True, cache=True) def add(a, b): return a + b csPddlm}i}dur|d<||fd|}r"|||S)Nr)CFuncr)rr )numba.core.ccallbackrKr5r8)r:rKadditional_argsresrrr rr<r%r&r$s zcfunc..wrapper)rnormalize_signature)r<rrrr r$r%rOr&cfuncs  rQcKsvtd}t|d}|jD]&\}}t|r8t||kr8td||j |t |fi||j|<qdS)a Automatically ``jit``-wraps functions defined in a Python module Note that ``jit_module`` should only be called at the end of the module to be jitted. In addition, only functions which are defined in the module ``jit_module`` is called from are considered for automatic jit-wrapping. See the Numba documentation for more information about what can/cannot be jitted. :param kwargs: Keyword arguments to pass to ``jit`` such as ``nopython`` or ``error_model``. rzCAuto decorating function {} from module {} with jit and options: {}N) r/stack getmodule__dict__itemsr0_loggerdebugr__name__r')kwargsframemodulenameobjr%r%r& jit_module s  r_)NFN)__doc__sysrDr/loggingnumba.core.errorsrrnumba.stencils.stencilrr6rrrr getLoggerrYrWrr'rrCrJrQr_r%r%r%r&s*   *