U hdp@sbddlmZddlZddlZddlmZddlmZmZm 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)m*Z*dd l+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:dd l;mm?Z?m@Z@mAZAGd dde?ZBeBZCdeC_DdddddddddddddddgZEGdd d ed!eEZFed"ddd#d$gZGd%d&ZHd'd(ZIdeCifd)d*ZJdCd,d-ZKGd.d/d/eLZMGd0d1d1eNZOGd2d3d3ePZQd4d5ZRGd6d7d7eLZSGd8d9d9eSZTGd:d;d;eLZUdeTfddd+deTfd?d@ZWdAdBZXdS)D) namedtupleN)event) utilserrorstyping interpreterbytecodepostprocconfigcallconvcpu)ParforDiagnostics) CompilerError)lookup_environment) PassManager)ExtractByteCodeTranslateByteCode FixupArgs IRProcessingDeadBranchPruneRewriteSemanticConstantsInlineClosureLikesGenericRewrites WithLiftingInlineInlinablesFindLiterallyCallsMakeFunctionToJitFunctionCanonicalizeLoopExitCanonicalizeLoopEntry LiteralUnrollReconstructSSARewriteDynamicRaises!LiteralPropagationSubPipelinePass)NopythonTypeInference AnnotateTypesNopythonRewrites PreParforPass ParforPassDumpParforDiagnosticsIRLegalizationNoPythonBackendInlineOverloadsPreLowerStripPhisNativeLoweringNativeParforLowering"NoPythonSupportedFeatureValidationParforFusionPassParforPreLoweringPass)ObjectModeFrontEndObjectModeBackEnd) TargetConfigOption ConfigStackc@sfeZdZeedddZeedddZeedddZeedddZeedddZ eedd dZ eedd dZ eedd dZ eedd dZ eedd dZeedd dZeedd dZeejedd dZeedd dZeedd dZeed d dZeejedd dZeedd dZeejedd dZeedddZeedddZeedddZeedddZ dS)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.zDMake debug emissions directives-only. Used when generating lineinfo.N)!__name__ __module__ __qualname__r5boolZenable_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_optnoneZdbg_directives_onlyrPrPL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/core/compiler.pyr7*sr7Ttyping_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. rPc CsT|j}t|j}|j}d|_|_|}||j|j|j |j |j ||j t |f S)zA Reduce a CompileResult to picklable components. N)r[Zserialize_using_object_coderLrVrZtypemap calltypes_find_referenced_environmentsr]rWrXrYr_tuple)selflibdatatypeannrZr`rPrPrQ_reduces   zCompileResult._reducecCsN|jj}g}|jD]6}|j}|drt|}|dk r|r||q|S)z2Returns a list of referenced environments Z _ZN08NumbaEnvN)r[Z _final_moduleZglobal_variablesname startswithrZ can_cacheappend)rfmodr`gvZgvnenvrPrPrQrds   z+CompileResult._find_referenced_environmentsc Csv| r| D] } | q||} || ||} |||j| || |||||ddd| | d}| D]}| j|j|q\|S)N)rSrRr[r]rTrZrVrWrXrYrUr\r^r_r`)codegenZunserialize_libraryget_executablerRZset_envenv_name)clsrSrgrZrorWrXrYrhr_r`fnr[cfunccrrPrPrQ_rebuilds0zCompileResult._rebuildcCs |jSN)rSrprfrPrPrQrp szCompileResult.codegencCsDt|dt|jd|j|jj|ddt|ddS)NzDUMP  z )tabzEND DUMP)printr8r?rTrWdump)rfr|rPrPrQr~s zCompileResult.dumpN)rz) r?r@rA__doc__ __slots__rird classmethodrwpropertyrpr~rPrPrPrQras ! raZ_CompileResult _LowerResultrurocCsdt|}tt}||}|r(t|||}|D] }d||<q4|d}|dk r`|d|d<|S)NrU)setkeys CR_FIELDS NameErrorwith_traceback)entriesrZfieldsetZbadnamesmissingkerrrPrPrQsanitize_compile_result_entriess  rcKst|}tf|Srx)rra)rrPrPrQcompile_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 targetctxrPrPrQcompile_isolated1s  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 rHrunr Z PostProcessor) rZinline_closuresZ emit_delsrZinterpbcfunc_irrZ inline_passZ post_procrPrPrQ run_frontendAs        rc@s(eZdZdZddgZddZddZdS) _CompileStatuszC Describes the state of compilation. Used like a C record. fail_reason can_fallbackcCsd|_||_dSrx)rr)rfrrPrPrQ__init__^sz_CompileStatus.__init__cCs4g}|jD]}|dj|t||dq d|S)Nz{k}={v})rvz, )rrlformatgetattrjoin)rfvalsrrPrPrQ__repr__bs z_CompileStatus.__repr__N)r?r@rArrrrrPrPrPrQrXsrc@seZdZdZddZdS)_EarlyPipelineCompletionz@ Raised to indicate that a pipeline has completed early cCs ||_dSrx)result)rfrrPrPrQrnsz!_EarlyPipelineCompletion.__init__N)r?r@rArrrPrPrPrQrisrc@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|YnXdSrx)KeyErrorAttributeError)rfattrrPrPrQ __getattr__xs zStateDict.__getattr__cCs |||<dSrxrP)rfrvaluerPrPrQ __setattr__~szStateDict.__setattr__N)r?r@rArrrrPrPrPrQrrsrcCspi}|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_nrtrIrNrM) rFrGrJrIrNr Zcreate_error_modelrMZ subtarget)rrZsubtargetoptionsrMrPrPrQ_make_subtargets  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_configrefreshrstaterrrr[rrrrrrrrY lifted_fromrbrcrVr^r_pipeliner rrrCstatus)rfrrr[rrrrrPrPrQrs:   zCompilerBase.__init__cCs6tj||j_t|jd|j_d|j_| S)NrP) rrrrrrrun_passrYr_compile_bytecode)rfrrPrPrQrs zCompilerBase.compile_extrarPNcCsF|j|j_||j_||j_||j_|jjj|j_t|j| Srx) rrrYrr arg_countnargsrr _compile_ir)rfrrYrrPrPrQ compile_irs zCompilerBase.compile_ircCs tdS)zGChild classes override this to customize the pipelines in use. N)NotImplementedErrorryrPrPrQdefine_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)r6Zenterrrcopyr pipeline_namermodnameZ func_qualnamerZ exec_timesr^rrvrr ExceptionrZuse_new_style_errors isinstancerZ NumbaErrorrrrrAssertionError)rfpmspmr func_nameZis_final_pipelinereserPrPrQ _compile_coresD         zCompilerBase._compile_corecCs|jjdkst|S)z> Populate and run pipeline for bytecode input NrrrrryrPrPrQr szCompilerBase._compile_bytecodecCs|jjdk st|S)z8 Populate and run pipeline for IR input NrryrPrPrQrszCompilerBase._compile_ir)rPN) r?r@rArrrrrrrrrPrPrPrQrs* .rc@seZdZdZddZdS)CompilerzThe default compiler cCsJg}|jjjs |t|j|jjjs4|jjjrF|t|j|Srx) rrrErlDefaultPassBuilderdefine_nopython_pipelinerrdefine_objectmode_pipeline)rfrrPrPrQrs  zCompiler.define_pipelinesN)r?r@rArrrPrPrPrQrsrc@sreZdZdZedddZedddZedd d Zedd d ZedddZ edddZ edddZ 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) rdeepcopyrKwarningscatch_warnings simplefilterrZ NumbaWarningrr)rrrrrrrrYrZis_lifted_loopr[rZ norw_flagsrZ norw_cresZrw_cresZcresrrPrrQrs4   rc Cst|||||||}||S)z For internal use only. )rr) rrr[rrrrrrrPrPrQcompile_internal=sr)FF)Y collectionsrrrZnumba.core.tracingrZ numba.corerrrrrr r r r Znumba.parfors.parforr Znumba.core.errorsrZnumba.core.environmentrZnumba.core.compiler_machineryrZnumba.core.untyped_passesrrrrrrrrrrrrrrrr r!r"Znumba.core.typed_passesr#r$r%r&r'r(r)r*r+r,r-r.r/r0r1Znumba.core.object_mode_passesr2r3Znumba.core.targetconfigr4r5r6r7Z DEFAULT_FLAGSrJrrarrrrrrrrrdictrrrrrrrrrPrPrPrQs  ,    P D R   8  @