o 8ήc[i@s4ddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6dd l7m8Z8m9Z9dd l:m;Z;me>Z?de?_@gdZAGdddedeAZBedgdZCddZDddZEde?ifddZFd4ddZGGdd d eHZIGd!d"d"eJZKGd#d$d$eLZMd%d&ZNGd'd(d(eHZOGd)d*d*eOZPGd+d,d,eHZQdePfd-d.ZRd/dddePfd0d1ZSd2d3ZTdS)5) namedtupleN)event) utilserrorstyping interpreterbytecodepostprocconfigcallconvcpu)ParforDiagnostics) CompilerError)lookup_environment) PassManager)ExtractByteCodeTranslateByteCode FixupArgs IRProcessingDeadBranchPruneRewriteSemanticConstantsInlineClosureLikesGenericRewrites WithLiftingInlineInlinablesFindLiterallyCallsMakeFunctionToJitFunctionCanonicalizeLoopExitCanonicalizeLoopEntry LiteralUnrollReconstructSSA!LiteralPropagationSubPipelinePass) NopythonTypeInference AnnotateTypesNopythonRewrites PreParforPass ParforPassDumpParforDiagnosticsIRLegalizationNoPythonBackendInlineOverloadsPreLowerStripPhisNativeLowering"NoPythonSupportedFeatureValidation)ObjectModeFrontEndObjectModeBackEnd) TargetConfigOption ConfigStackc@sXeZdZeedddZeedddZeedddZeedddZeedddZ eedd dZ eedd dZ eedd dZ eedd dZ eedd dZeedd dZeedd dZeejedd dZeedd dZeedd dZeed d dZeejedd dZeedd dZeejedd dZeedddZeedddZeedddZdS)FlagsFzEnable loop-lifting)typedefaultdocz!Enable pyobject mode (in general)z(Enable pyobject mode inside lifted loopsTz Enable SSAz-Force pyobject mode inside the whole functionz&Release GIL inside the native functionTODOz7Force inlining of the function. Overrides _dbg_optnone.zEnable automatic parallel optimization, can be fine-tuned by taking a dictionary of sub-options instead of a boolean, see parfor.py for detailpythonneverr backendzTExtend variable lifetime for debugging. This automatically turns on with debug=True.z\Disable optimization for debug. Equivalent to adding optnone attribute in the LLVM Function.N) __name__ __module__ __qualname__r1boolenable_loopliftenable_pyobjectenable_pyobject_looplift enable_ssaforce_pyobject release_gil no_compile debuginfo boundscheck forceinlineno_cpython_wrapperno_cfunc_wrapperr ParallelOptions auto_parallelnrt no_rewritesstr error_modelFastMathOptionsfastmathnoalias InlineOptionsinlinetarget_backenddbg_extend_lifetimes dbg_optnonerYrY:/tmp/pip-target-vg8gfxp4/lib/python/numba/core/compiler.pyr3)s r3T)typing_contexttarget_context entry_point typing_errortype_annotation signature objectmodeliftedfndesclibrary call_helper environmentmetadata reload_initreferenced_envsc@sFeZdZdZdZddZddZeddZe d d Z dd d Z dS) CompileResultzI A structure holding results from the compilation of a function. rYc CsT|j}t|j}|j}d|_|_|}||j|j|j |j |j ||j t |f S)zA Reduce a CompileResult to picklable components. N)rdserialize_using_object_coderOr_rctypemap calltypes_find_referenced_environmentsrfr`rarbrhtuple)selflibdatatypeannrcrirYrYrZ_reduces   zCompileResult._reducecCsN|jj}g}|jD]}|j}|dr$t|}|dur$|r$||q |S)z2Returns a list of referenced environments _ZN08NumbaEnvN)rd _final_moduleglobal_variablesname startswithr can_cacheappend)rpmodrigvgvnenvrYrYrZrns   z+CompileResult._find_referenced_environmentsc Csv| r | D]} | q||} || ||} |||j| || |||||ddd| | d}| D] }| j|j|q.|S)N)r\r[rdrfr]rcr_r`rarbr^rergrhri)codegenunserialize_libraryget_executabler[set_envenv_name)clsr\rqrcr~r`rarbrrrhrifnrdcfunccrrYrYrZ_rebuilds0zCompileResult._rebuildcCs |jSN)r\rrprYrYrZrs zCompileResult.codegencCsDt|dt|jd|j|jj|ddt|ddS)NzDUMP  z )tabzEND DUMP)printr4r;r]r`dump)rprrYrYrZrs zCompileResult.dumpN)r) r;r<r=__doc__ __slots__rsrn classmethodrpropertyrrrYrYrYrZrjs ! rj_CompileResult _LowerResult)rcrerr~cCsdt|}tt}||}|rt|||}|D]}d||<q|d}|dur0|d|d<|S)Nr^)setkeys CR_FIELDS NameErrorwith_traceback)entriesrfieldsetbadnamesmissingkerrrYrYrZsanitize_compile_result_entriess  rcKst|}tdi|SNrY)rrj)rrYrYrZcompile_result$src Csfddlm}t}tj|dd}|||t|||||||WdS1s,wYdS)zp Compile the function in an isolated environment (typing and target context). Good for testing. r) cpu_targetr )targetN)numba.core.registryrrContextr CPUContextnested_context compile_extra)funcargs return_typeflagslocalsr typingctx targetctxrYrYrZcompile_isolated)s  $rFc Csptj|}t|}tj|d}||}|r,ddlm}||t did}| t |} | ||S)a Run the compiler frontend over the given Python function, and return the function's canonical Numba IR. If inline_closures is Truthy then closure inlining will be run If emit_dels is Truthy the ir.Del nodes will be emitted appropriately )func_idr)InlineClosureCallPassF)rFunctionIdentity from_functionr InterpreterByteCode interpretnumba.core.inline_closurecallrr rKrunr PostProcessor) rinline_closures emit_delsrinterpbcfunc_irr inline_pass post_procrYrYrZ run_frontend9s       rc@s(eZdZdZddgZddZddZdS) _CompileStatuszC Describes the state of compilation. Used like a C record. fail_reason can_fallbackcCsd|_||_dSr)rr)rprrYrYrZ__init__Vs z_CompileStatus.__init__cCs4g}|jD]}|dj|t||dqd|S)Nz{k}={v})rvz, )rrzformatgetattrjoin)rpvalsrrYrYrZ__repr__Zs  z_CompileStatus.__repr__N)r;r<r=rrrrrYrYrYrZrPs  rc@eZdZdZddZdS)_EarlyPipelineCompletionz@ Raised to indicate that a pipeline has completed early cCs ||_dSr)result)rprrYrYrZrfs z!_EarlyPipelineCompletion.__init__N)r;r<r=rrrYrYrYrZras rc@s eZdZdZddZddZdS) StateDictz A dictionary that has an overloaded getattr and setattr to permit getting and setting key/values through the use of attributes. cCs"z||WStyt|wr)KeyErrorAttributeError)rpattrrYrYrZ __getattr__ps   zStateDict.__getattr__cCs |||<dSrrY)rprvaluerYrYrZ __setattr__vs zStateDict.__setattr__N)r;r<r=rrrrYrYrYrZrjs rcCsti}|jr d|d<|jrd|d<|jrd|d<|jr|j|d<|jr'|j|d<t|j|}||d<|jd i|S) zL Make a new target context from the given target context and flags. Tenable_debuginfoenable_boundscheck enable_nrtrLrRrPNrY) rFrGrMrLrRr create_error_modelrP subtarget)rrsubtargetoptionsrPrYrYrZ_make_subtargetzs  rc@sJeZdZdZddZddZddd Zd d Zd d ZddZ ddZ dS) CompilerBasez4 Stores and manages states for the compiler cCst||t|_||j_t|||j_||j_||j_ ||j_ ||j_ ||j_ d|j_ d|j_d|j_d|j_d|j_d|j_d|j_d|j_i|j_g|j_||j_t|j_|jj|jjd<i|jjd<t|jj jd|j_dS)Nparfor_diagnosticsparfors)r)r reload_configrefreshrstaterrrrdrrrrrrrrb lifted_fromrlrmr_rgrhpipeliner rrr@status)rprrrdrrrrrYrYrZrs:  zCompilerBase.__init__cCs6tj||j_t|jd|j_d|j_| Sr) rrrrrrrun_passrbr_compile_bytecode)rprrYrYrZrs zCompilerBase.compile_extrarYNcCsF|j|j_||j_||j_||j_|jjj|j_t|j| Sr) rrrbrr arg_countnargsrr _compile_ir)rprrbrrYrYrZ compile_irs zCompilerBase.compile_ircCst)zGChild classes override this to customize the pipelines in use. )NotImplementedErrorrrYrYrZdefine_pipelinesszCompilerBase.define_pipelinesc Csvt|jj|}|D]w}|j}d|jjj|jjj f}t d||f||j i|jj d<||dk}d}z| |j|jjdurKWnBWqtyb}z |j}WYd}~n/d}~wty}ztrut|tjsu|||jj_|r~|WYd}~qd}~wwtdd|j_|dur|WdS|jjdusJ|jjWdS1swYdS)z4 Populate and run compiler pipeline z%s.%szPipeline: %s for %spipeline_timesNz!All available pipelines exhausted)r2enterrrcopyr pipeline_namermodname func_qualnamer exec_timesrgrrrr Exceptionruse_new_style_errors isinstancer NumbaErrorrrrr)rppmspmr func_nameis_final_pipelinereserYrYrZ _compile_coresR        '$zCompilerBase._compile_corecCs|jjdusJ|S)z> Populate and run pipeline for bytecode input NrrrrrYrYrZrzCompilerBase._compile_bytecodecCs|jjdusJ|S)z8 Populate and run pipeline for IR input NrrrYrYrZr rzCompilerBase._compile_ir)rYN) r;r<r=rrrrrrrrrYrYrYrZrs*  . rc@r)CompilerzThe default compiler cCsJg}|jjjs|t|j|jjjs|jjjr#|t|j|Sr) rrrCrzDefaultPassBuilderdefine_nopython_pipelinerrdefine_objectmode_pipeline)rprrYrYrZrs  zCompiler.define_pipelinesN)r;r<r=rrrYrYrYrZrs rc@sVeZdZdZedddZedddZedd d Zedd d ZedddZ dS)rz This is the default pass builder, it contains the "classic" default pipelines as pre-canned PassManager instances: - nopython - objectmode - interpreted - typed - untyped - nopython lowering nopythoncCs`t}t|}||}|j|j||}|j|j||}|j|j||S)z.compile_localNignorer+) rdeepcopyrNwarningscatch_warnings simplefilterr NumbaWarningrr)rrrrrrrrbris_lifted_looprdr* norw_flagsr/ norw_cresrw_crescresrrYr.rZrs6   rc Cst|||||||}||S)z For internal use only. )rr) rrrdrrrrrrrYrYrZcompile_internals r;)FF)U collectionsrrr2numba.core.tracingr numba.corerrrrrr r r r numba.parfors.parforr numba.core.errorsrnumba.core.environmentrnumba.core.compiler_machineryrnumba.core.untyped_passesrrrrrrrrrrrrrrrr r!numba.core.typed_passesr"r#r$r%r&r'r(r)r*r+r,r-numba.core.object_mode_passesr.r/numba.core.targetconfigr0r1r2r3 DEFAULT_FLAGSrMrrjrrrrrr%rrrdictrrrrrrrr;rYrYrYrZsP  ,    L8  yR    @