U ,d[i@sRddlmZddlZddlZddlmZddlmZmZm 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?_@dddddddddddddddgZAGdd d ed!eAZBed"ddd#d$gZCd%d&ZDd'd(ZEde?ifd)d*ZFdCd,d-ZGGd.d/d/eHZIGd0d1d1eJZKGd2d3d3eLZMd4d5ZNGd6d7d7eHZOGd8d9d9eOZPGd:d;d;eHZQdePfddd+dePfd?d@ZSdAdBZTdS)D) 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 functionZTODOz7Force 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__r1boolZenable_loopliftenable_pyobjectZenable_pyobject_looplift enable_ssaforce_pyobjectZ release_gilZ no_compile debuginfo boundscheckZ forceinlineZno_cpython_wrapperZno_cfunc_wrapperr ParallelOptions auto_parallelnrt no_rewritesstr error_modelZFastMathOptionsfastmathZnoaliasZ InlineOptionsinlineZtarget_backendZdbg_extend_lifetimesZ dbg_optnonerKrK7/tmp/pip-unpacked-wheel-eu7e0c37/numba/core/compiler.pyr3)sr3Ttyping_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. rKc CsT|j}t|j}|j}d|_|_|}||j|j|j |j |j ||j t |f S)zA Reduce a CompileResult to picklable components. N)rVZserialize_using_object_coderGrQrUtypemap calltypes_find_referenced_environmentsrXrRrSrTrZtuple)selflibdatatypeannrUr[rKrKrL_reduces   zCompileResult._reducecCsN|jj}g}|jD]6}|j}|drt|}|dk r|r||q|S)z2Returns a list of referenced environments Z _ZN08NumbaEnvN)rVZ _final_moduleZglobal_variablesname startswithrZ can_cacheappend)ramodr[gvZgvnenvrKrKrLr_s   z+CompileResult._find_referenced_environmentsc Csv| r| D] } | q||} || ||} |||j| || |||||ddd| | d}| D]}| j|j|q\|S)N)rNrMrVrXrOrUrQrRrSrTrPrWrYrZr[)codegenZunserialize_libraryget_executablerMZset_envenv_name)clsrNrbrUrjrRrSrTrcrZr[fnrVcfunccrrKrKrL_rebuilds0zCompileResult._rebuildcCs |jSN)rNrkrarKrKrLrkszCompileResult.codegencCsDt|dt|jd|j|jj|ddt|ddS)NzDUMP  z )tabzEND DUMP)printr4r:rOrRdump)rarwrKrKrLrys zCompileResult.dumpN)ru) r:r;r<__doc__ __slots__rdr_ classmethodrrpropertyrkryrKrKrKrLr\s ! r\Z_CompileResult _LowerResultrprjcCsdt|}tt}||}|r(t|||}|D] }d||<q4|d}|dk r`|d|d<|S)NrP)setkeys CR_FIELDS NameErrorwith_traceback)entriesrfieldsetZbadnamesmissingkerrrKrKrLsanitize_compile_result_entriess  rcKst|}tf|Srs)rr\)rrKrKrLcompile_result$src CsZddlm}t}tj|dd}|||"t|||||||W5QRSQRXdS)zp Compile the function in an isolated environment (typing and target context). Good for testing. r) cpu_targetr )targetN)Znumba.core.registryrrContextr Z CPUContextZnested_context compile_extra)funcargs return_typeflagslocalsr typingctx targetctxrKrKrLcompile_isolated)s  rFc Csptj|}t|}tj|d}||}|rXddlm}||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_functionrZ InterpreterZByteCodeZ interpretZnumba.core.inline_closurecallrr rCrunr Z PostProcessor) rZinline_closuresZ emit_delsrZinterpbcfunc_irrZ inline_passZ post_procrKrKrL run_frontend9s        rc@s(eZdZdZddgZddZddZdS) _CompileStatuszC Describes the state of compilation. Used like a C record. fail_reason can_fallbackcCsd|_||_dSrs)rr)rarrKrKrL__init__Vsz_CompileStatus.__init__cCs4g}|jD]}|dj|t||dq d|S)Nz{k}={v})rvz, )r{rgformatgetattrjoin)ravalsrrKrKrL__repr__Zs z_CompileStatus.__repr__N)r:r;r<rzr{rrrKrKrKrLrPsrc@seZdZdZddZdS)_EarlyPipelineCompletionz@ Raised to indicate that a pipeline has completed early cCs ||_dSrs)result)rarrKrKrLrfsz!_EarlyPipelineCompletion.__init__N)r:r;r<rzrrKrKrKrLrasrc@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 ||WStk r&t|YnXdSrs)KeyErrorAttributeError)raattrrKrKrL __getattr__ps zStateDict.__getattr__cCs |||<dSrsrK)rarvaluerKrKrL __setattr__vszStateDict.__setattr__N)r:r;r<rzrrrKrKrKrLrjsrcCspi}|jrd|d<|jr d|d<|jr.d|d<|jr>|j|d<|jrN|j|d<t|j|}||d<|jf|S)zL Make a new target context from the given target context and flags. TZenable_debuginfoZenable_boundscheckZ enable_nrtrDrIrH) rArBrErDrIr Zcreate_error_modelrHZ subtarget)rrZsubtargetoptionsrHrKrKrL_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_diagnosticsZparfors)r)r Z reload_configZrefreshrstaterrrrVrrrrrrrrT lifted_fromr]r^rQrYrZpipeliner rrr>status)rarrrVrrrrrKrKrLrs:   zCompilerBase.__init__cCs6tj||j_t|jd|j_d|j_| S)NrK) rrrrrrrun_passrTr_compile_bytecode)rarrKrKrLrs zCompilerBase.compile_extrarKNcCsF|j|j_||j_||j_||j_|jjj|j_t|j| Srs) rrrTrr arg_countnargsrr _compile_ir)rarrTrrKrKrL compile_irs zCompilerBase.compile_ircCs tdS)zGChild classes override this to customize the pipelines in use. N)NotImplementedErrorrtrKrKrLdefine_pipelinesszCompilerBase.define_pipelinesc Cst|jj`|}|D]}|j}d|jjj|jjj f}t d||f||j i|jj d<||dk}d}z$| |j|jjdk rWq*Wq$tk r}z|j}WY q*W5d}~XYq$tk r}z0trt|tjs|||jj_|r|W5d}~XYq$Xq$tdd|j_|dk rL|W5QRS|jjdk s^t|jjW5QRSW5QRXdS)z4 Populate and run compiler pipeline z%s.%szPipeline: %s for %sZpipeline_timesNz!All available pipelines exhausted)r2Zenterrrcopyr pipeline_namermodnameZ func_qualnamerZ exec_timesrYrrqrr ExceptionrZuse_new_style_errors isinstancerZ NumbaErrorrrrrAssertionError)rapmspmr func_nameZis_final_pipelinereserKrKrL _compile_coresD         zCompilerBase._compile_corecCs|jjdkst|S)z> Populate and run pipeline for bytecode input NrrrrrtrKrKrLrszCompilerBase._compile_bytecodecCs|jjdk st|S)z8 Populate and run pipeline for IR input NrrtrKrKrLr szCompilerBase._compile_ir)rKN) r:r;r<rzrrrrrrrrKrKrKrLrs* .rc@seZdZdZddZdS)CompilerzThe default compiler cCsJg}|jjjs |t|j|jjjs4|jjjrF|t|j|Srs) rrr@rgDefaultPassBuilderdefine_nopython_pipelinerrdefine_objectmode_pipeline)rarrKrKrLrs  zCompiler.define_pipelinesN)r:r;r<rzrrKrKrKrLrsrc@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) rdeepcopyrFwarningscatch_warnings simplefilterrZ NumbaWarningrr)rrrrrrrrTrZis_lifted_looprVrZ norw_flagsrZ norw_cresZrw_cresZcresrrKrrLrs4   rc Cst|||||||}||S)z For internal use only. )rr) rrrVrrrrrrrKrKrLcompile_internalsr)FF)U collectionsrrrZnumba.core.tracingrZ numba.corerrrrrr r r r Znumba.parfors.parforr Znumba.core.errorsrZnumba.core.environmentrZnumba.core.compiler_machineryrZnumba.core.untyped_passesrrrrrrrrrrrrrrrr r!Znumba.core.typed_passesr"r#r$r%r&r'r(r)r*r+r,r-Znumba.core.object_mode_passesr.r/Znumba.core.targetconfigr0r1r2r3Z DEFAULT_FLAGSrErr\r~rrrrrrrrdictrrrrrrrrrKrKrKrLs~  ,    L 8 yR     @