a dp@sDddlmZddlZddlZddlmZddlmZmZm 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_DgdZEGdddedeEZFedgdZGddZHddZIdeCifddZJd4ddZKGdd d eLZMGd!d"d"eNZOGd#d$d$ePZQd%d&ZRGd'd(d(eLZSGd)d*d*eSZTGd+d,d,eLZUdeTfd-d.ZVd/dddeTfd0d1ZWd2d3ZXdS)5) 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 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.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_onlyrOrO\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numba/core/compiler.pyr7*sr7T)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. rOc CsT|j}t|j}|j}d|_|_|}||j|j|j |j |j ||j t |f S)zA Reduce a CompileResult to picklable components. N)rZZserialize_using_object_coderKrUrYtypemap calltypes_find_referenced_environmentsr\rVrWrXr^tuple)selflibdatatypeannrYr_rOrOrP_reduces   zCompileResult._reducecCsN|jj}g}|jD]6}|j}|drt|}|dur|r||q|S)z2Returns a list of referenced environments Z _ZN08NumbaEnvN)rZZ _final_moduleZglobal_variablesname startswithrZ can_cacheappend)remodr_gvZgvnenvrOrOrPrcs   z+CompileResult._find_referenced_environmentsc Csv| r| D] } | q||} || ||} |||j| || |||||ddd| | d}| D]}| j|j|q\|S)N)rRrQrZr\rSrYrUrVrWrXrTr[r]r^r_)codegenZunserialize_libraryget_executablerQZset_envenv_name)clsrRrfrYrnrVrWrXrgr^r_fnrZcfunccrrOrOrP_rebuilds0zCompileResult._rebuildcCs |jSN)rRrorerOrOrPro szCompileResult.codegencCsDt|dt|jd|j|jj|ddt|ddS)NzDUMP  z )tabzEND DUMP)printr8r>rSrVdump)rer{rOrOrPr}s zCompileResult.dumpN)ry) r>r?r@__doc__ __slots__rhrc classmethodrvpropertyror}rOrOrOrPr`s ! r`Z_CompileResult _LowerResult)rYr[rtrncCsdt|}tt}||}|r(t|||}|D] }d||<q4|d}|dur`|d|d<|S)NrT)setkeys CR_FIELDS NameErrorwith_traceback)entriesrfieldsetZbadnamesmissingkerrrOrOrPsanitize_compile_result_entriess  rcKst|}tfi|Srw)rr`)rrOrOrPcompile_result,src Csfddlm}t}tj|dd}|||$t|||||||WdS1sX0YdS)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 targetctxrOrOrPcompile_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 rGrunr Z PostProcessor) rZinline_closuresZ emit_delsrZinterpbcfunc_irrZ inline_passZ post_procrOrOrP run_frontendAs        rc@s(eZdZdZddgZddZddZdS) _CompileStatuszC Describes the state of compilation. Used like a C record. fail_reason can_fallbackcCsd|_||_dSrw)rr)rerrOrOrP__init__^sz_CompileStatus.__init__cCs4g}|jD]}|dj|t||dq d|S)Nz{k}={v})rvz, )rrkformatgetattrjoin)revalsrrOrOrP__repr__bs z_CompileStatus.__repr__N)r>r?r@r~rrrrOrOrOrPrXsrc@seZdZdZddZdS)_EarlyPipelineCompletionz@ Raised to indicate that a pipeline has completed early cCs ||_dSrw)result)rerrOrOrPrnsz!_EarlyPipelineCompletion.__init__N)r>r?r@r~rrOrOrOrPrisrc@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 ||WSty$t|Yn0dSrw)KeyErrorAttributeError)reattrrOrOrP __getattr__xs  zStateDict.__getattr__cCs |||<dSrwrO)rervaluerOrOrP __setattr__~szStateDict.__setattr__N)r>r?r@r~rrrOrOrOrPrrsrcCsti}|jrd|d<|jr d|d<|jr.d|d<|jr>|j|d<|jrN|j|d<t|j|}||d<|jfi|S)zL Make a new target context from the given target context and flags. TZenable_debuginfoZenable_boundscheckZ enable_nrtrHrMrL) rErFrIrHrMr Zcreate_error_modelrLZ subtarget)rrZsubtargetoptionsrLrOrOrP_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_configZrefreshrstaterrrrZrrrrrrrrX lifted_fromrarbrUr]r^pipeliner rrrBstatus)rerrrZrrrrrOrOrPrs:   zCompilerBase.__init__cCs6tj||j_t|jd|j_d|j_| S)NrO) rrrrrrrun_passrXr_compile_bytecode)rerrOrOrPrs zCompilerBase.compile_extrarONcCsF|j|j_||j_||j_||j_|jjj|j_t|j| Srw) rrrXrr arg_countnargsrr _compile_ir)rerrXrrOrOrP compile_irs zCompilerBase.compile_ircCs tdS)zGChild classes override this to customize the pipelines in use. N)NotImplementedErrorrxrOrOrPdefine_pipelinesszCompilerBase.define_pipelinesc Cst|jj||}|D]}|j}d|jjj|jjj f}t d||f||j i|jj d<||dk}d}z$| |j|jjdurWq8Wq$ty}z"|j}WYd}~q8WYd}~q$d}~0ty,}z Populate and run pipeline for bytecode input NrrrrxrOrOrPr szCompilerBase._compile_bytecodecCs|jjdusJ|S)z8 Populate and run pipeline for IR input NrrxrOrOrPrszCompilerBase._compile_ir)rON) r>r?r@r~rrrrrrrrOrOrOrPrs* .rc@seZdZdZddZdS)CompilerzThe default compiler cCsJg}|jjjs |t|j|jjjs4|jjjrF|t|j|Srw) rrrDrkDefaultPassBuilderdefine_nopython_pipelinerrdefine_objectmode_pipeline)rerrOrOrPrs  zCompiler.define_pipelinesN)r>r?r@r~rrOrOrOrPrsrc@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)zr?r@r~ staticmethodrrrrrrrrOrOrOrPr*s        ,rc Cs||||||||} | |S)aDCompiler entry point Parameter --------- typingctx : typing context targetctx : target context func : function the python function to be compiled args : tuple, list argument types return_type : Use ``None`` to indicate void return flags : numba.compiler.Flags compiler flags library : numba.codegen.CodeLibrary Used to store the compiled code. If it is ``None``, a new CodeLibrary is used. pipeline_class : type like numba.compiler.CompilerBase compiler pipeline )r) rrrrrrrrZpipeline_classrrOrOrPrsrrOc  s| rt|} d| _f dd} | || }d}|jstDtdtjz| ||}WntyYn0Wdn1s0Y|dur|}n|}|S|}|j |dSdS)zK Compile a function with the given IR. For internal use only. Tcs$|}|j|dS)NrrXr)r)Zthe_irZ the_flagsr rrZrXrrrrrrrOrP compile_locals z!compile_ir..compile_localNignorer) rdeepcopyrJwarningscatch_warnings simplefilterrZ NumbaWarningrr)rrrrrrrrXrZis_lifted_looprZrZ norw_flagsrZ norw_cresZrw_cresZcresrrOrrPrs.    $rc Cst|||||||}||S)z For internal use only. )rr) rrrZrrrrrrrOrOrPcompile_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_FLAGSrIrr`rrrrrrrrrdictrrrrrrrrrOrOrOrPsP  ,    P D R   8  @