U
    |ep                     @   sb  d dl mZ d dlZd dlZd dlmZ d dlmZmZm	Z	m
Z
mZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* d d
l+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z: d dl;m<Z<m=Z= d dl>m?Z?m@Z@mAZA G dd de?ZBeB ZCdeC_DdddddddddddddddgZEG dd  d ed!eEZFed"ddd#d$gZGd%d& ZHd'd( ZIdeCi fd)d*ZJdCd,d-ZKG d.d/ d/eLZMG d0d1 d1eNZOG d2d3 d3ePZQd4d5 ZRG d6d7 d7eLZSG d8d9 d9eSZTG d:d; d;eLZUdeTfd<d=ZVd>dd+deTfd?d@ZWdAdB ZXdS )D    )
namedtupleN)event)	utilserrorstypinginterpreterbytecodepostprocconfigcallconvcpu)ParforDiagnostics)CompilerError)lookup_environment)PassManager)ExtractByteCodeTranslateByteCode	FixupArgsIRProcessingDeadBranchPruneRewriteSemanticConstantsInlineClosureLikesGenericRewritesWithLiftingInlineInlinablesFindLiterallyCallsMakeFunctionToJitFunctionCanonicalizeLoopExitCanonicalizeLoopEntryLiteralUnrollReconstructSSARewriteDynamicRaises!LiteralPropagationSubPipelinePass)NopythonTypeInferenceAnnotateTypesNopythonRewritesPreParforPass
ParforPassDumpParforDiagnosticsIRLegalizationNoPythonBackendInlineOverloadsPreLowerStripPhisNativeLoweringNativeParforLowering"NoPythonSupportedFeatureValidationParforFusionPassParforPreLoweringPass)ObjectModeFrontEndObjectModeBackEnd)TargetConfigOptionConfigStackc                   @   sf  e Zd ZeedddZeedddZeedddZeedddZeedddZ	eedd	dZ
eedd
dZeedd
dZeedd
dZeedddZeedd
dZeedd
dZeejedddZeedd
dZeedd
dZeedd
dZeejedd
dZeedd
dZeejedd
dZeedddZeedddZeedddZeedddZ dS )FlagsFzEnable loop-lifting)typedefaultdocz!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 functionZTODOz7Force 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
detailpythonneverr   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__r5   boolenable_loopliftenable_pyobjectZenable_pyobject_looplift
enable_ssaforce_pyobjectrelease_gilZ
no_compile	debuginfoboundscheckZforceinlineno_cpython_wrapperZno_cfunc_wrapperr   ParallelOptionsauto_parallelnrtno_rewritesstrerror_modelFastMathOptionsfastmathnoaliasInlineOptionsinlinetarget_backenddbg_extend_lifetimesdbg_optnonedbg_directives_only rY   rY   P/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/core/compiler.pyr7   *   s   r7   Ttyping_contexttarget_contextentry_pointtyping_errortype_annotation	signature
objectmodeliftedfndesclibrarycall_helperenvironmentmetadatareload_initreferenced_envsc                   @   sF   e Zd ZdZdZdd Zdd Zedd Ze	d	d
 Z
dddZdS )CompileResultzI
    A structure holding results from the compilation of a function.
    rY   c              
   C   sT   | j  }t| j}| j}d |_|_|  }|| j| j| j	| j
| j|| jt|f	S )zA
        Reduce a CompileResult to picklable components.
        N)rd   Zserialize_using_object_coderN   r_   rc   typemap	calltypes_find_referenced_environmentsrf   r`   ra   rb   rh   tuple)selflibdatatypeannrc   ri   rY   rY   rZ   _reduce   s    

   zCompileResult._reducec                 C   sN   | j j}g }|jD ]6}|j}|drt|}|dk	r| r|| q|S )z2Returns a list of referenced environments
        Z_ZN08NumbaEnvN)rd   Z_final_moduleglobal_variablesname
startswithr   	can_cacheappend)ro   modri   gvZgvnenvrY   rY   rZ   rm      s    

z+CompileResult._find_referenced_environmentsc                 C   sv   |	r|	D ]
}|  q|  |}||||}| ||j||||||||d d d |	|
d}|
D ]}|j |j| q\|S )N)r\   r[   rd   rf   r]   rc   r_   r`   ra   rb   r^   re   rg   rh   ri   )codegenZunserialize_libraryget_executabler[   Zset_envenv_name)clsr\   rp   rc   rz   r`   ra   rb   rq   rh   ri   fnrd   cfunccrrY   rY   rZ   _rebuild   s0    zCompileResult._rebuildc                 C   s
   | j  S N)r\   r{   ro   rY   rY   rZ   r{   
  s    zCompileResult.codegen c                 C   sD   t | dt| j d| j  | jj|d d t | d d S )NzDUMP  z  )tabzEND DUMP)printr8   r>   r]   r`   dump)ro   r   rY   rY   rZ   r     s     zCompileResult.dumpN)r   )r>   r?   r@   __doc__	__slots__rr   rm   classmethodr   propertyr{   r   rY   rY   rY   rZ   rj      s   
!
rj   Z_CompileResult_LowerResultr   rz   c                 C   sd   t |  }t t}|| }|r(t| || }|D ]}d | |< q4| d }|d k	r`|d | d< | S )Nr^   )setkeys	CR_FIELDS	NameErrorwith_traceback)entriesr   fieldsetZbadnamesmissingkerrrY   rY   rZ   sanitize_compile_result_entries  s    
r   c                  K   s   t | } tf | S r   )r   rj   )r   rY   rY   rZ   compile_result,  s    r   c              
   C   sZ   ddl m} t }tj|dd}|||" t||| ||||W  5 Q R  S Q R X dS )zp
    Compile the function in an isolated environment (typing and target
    context).
    Good for testing.
    r   )
cpu_targetr   )targetN)numba.core.registryr   r   Contextr   
CPUContextnested_contextcompile_extra)funcargsreturn_typeflagslocalsr   	typingctx	targetctxrY   rY   rZ   compile_isolated1  s     r   Fc           
      C   sp   t j| }t|}t j|d}||}|rXddlm} ||t	
di d}|  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)r   FunctionIdentityfrom_functionr   ZInterpreterZByteCodeZ	interpretnumba.core.inline_closurecallr   r   rJ   runr	   PostProcessor)
r   Zinline_closures	emit_delsr   interpbcfunc_irr   inline_pass	post_procrY   rY   rZ   run_frontendA  s    	

 

r   c                   @   s(   e Zd ZdZddgZdd Zdd ZdS )	_CompileStatuszC
    Describes the state of compilation. Used like a C record.
    fail_reasoncan_fallbackc                 C   s   d | _ || _d S r   )r   r   )ro   r   rY   rY   rZ   __init__^  s    z_CompileStatus.__init__c                 C   s4   g }| j D ]}|dj|t| |d q
d|S )Nz{k}={v})r   vz, )r   rw   formatgetattrjoin)ro   valsr   rY   rY   rZ   __repr__b  s    
z_CompileStatus.__repr__N)r>   r?   r@   r   r   r   r   rY   rY   rY   rZ   r   X  s   r   c                   @   s   e Zd ZdZdd ZdS )_EarlyPipelineCompletionz@
    Raised to indicate that a pipeline has completed early
    c                 C   s
   || _ d S r   )result)ro   r   rY   rY   rZ   r   n  s    z!_EarlyPipelineCompletion.__init__N)r>   r?   r@   r   r   rY   rY   rY   rZ   r   i  s   r   c                   @   s    e Zd ZdZdd Zdd ZdS )	StateDictz
    A dictionary that has an overloaded getattr and setattr to permit getting
    and setting key/values through the use of attributes.
    c                 C   s,   z
| | W S  t k
r&   t|Y nX d S r   )KeyErrorAttributeError)ro   attrrY   rY   rZ   __getattr__x  s    
zStateDict.__getattr__c                 C   s   || |< d S r   rY   )ro   r   valuerY   rY   rZ   __setattr__~  s    zStateDict.__setattr__N)r>   r?   r@   r   r   r   rY   rY   rY   rZ   r   r  s   r   c                 C   sp   i }|j rd|d< |jr d|d< |jr.d|d< |jr>|j|d< |jrN|j|d< t|j| }||d< | jf |S )zL
    Make a new target context from the given target context and flags.
    Tenable_debuginfoZenable_boundscheck
enable_nrtrK   rQ   rO   )	rG   rH   rL   rK   rQ   r   create_error_modelrO   	subtarget)r   r   ZsubtargetoptionsrO   rY   rY   rZ   _make_subtarget  s    

r   c                   @   sJ   e Zd ZdZdd Zdd Zddd	Zd
d Zdd Zdd Z	dd Z
dS )CompilerBasez4
    Stores and manages states for the compiler
    c                 C   s   t   |  |  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_| jj| jjd< i | jjd< t| jjjd| j_d S )Nparfor_diagnosticsZparfors)r   )r
   reload_configrefreshr   stater   r   r   rd   r   r   r   r   r   r   r   rb   lifted_fromrk   rl   r_   rg   rh   pipeliner   r   r   rC   status)ro   r   r   rd   r   r   r   r   rY   rY   rZ   r     s:    

zCompilerBase.__init__c                 C   s6   t j|| j_t | j d| j_d | j_| 	 S )NrY   )
r   r   r   r   r   r   run_passrb   r   _compile_bytecode)ro   r   rY   rY   rZ   r     s
    zCompilerBase.compile_extrarY   Nc                 C   sF   |j | j_ || j_|| j_|| j_| jjj| j_t | j | 	 S r   )
r   r   rb   r   r   	arg_countnargsr   r   _compile_ir)ro   r   rb   r   rY   rY   rZ   
compile_ir  s    
zCompilerBase.compile_irc                 C   s
   t  dS )zGChild classes override this to customize the pipelines in use.
        N)NotImplementedErrorr   rY   rY   rZ   define_pipelines  s    zCompilerBase.define_pipelinesc                 C   s  t  | jj ` |  }|D ]}|j}d| jjj| jjj	f }t
d||f  ||ji| jjd< ||d k}d}z$|| j | jjdk	rW  q*W q$ tk
r } z|j}W Y 
 q*W 5 d}~X Y q$ tk
r } z0t rt|tjs||| jj_|r|W 5 d}~X Y q$X q$tdd| j_|dk	rL|W  5 Q R  S | jjdk	s^t| jjW  5 Q R  S W 5 Q R X dS )z4
        Populate and run compiler pipeline
        z%s.%szPipeline: %s for %sZpipeline_timesNz!All available pipelines exhausted)r6   enterr   r   copyr   pipeline_namer   modnamefunc_qualnamer   Z
exec_timesrg   r   r   r   r   	Exceptionr   use_new_style_errors
isinstancer   
NumbaErrorr   r   r   r   AssertionError)ro   pmspmr   	func_nameZis_final_pipelinereserY   rY   rZ   _compile_core  sD    



zCompilerBase._compile_corec                 C   s   | j jdkst|  S )z>
        Populate and run pipeline for bytecode input
        Nr   r   r   r   r   rY   rY   rZ   r     s    zCompilerBase._compile_bytecodec                 C   s   | j jdk	st|  S )z8
        Populate and run pipeline for IR input
        Nr   r   rY   rY   rZ   r     s    zCompilerBase._compile_ir)rY   N)r>   r?   r@   r   r   r   r   r   r   r   r   rY   rY   rY   rZ   r     s   *

.r   c                   @   s   e Zd ZdZdd ZdS )CompilerzThe default compiler
    c                 C   sJ   g }| j jjs |t| j  | j jjs4| j jjrF|t| j  |S r   )	r   r   rE   rw   DefaultPassBuilderdefine_nopython_pipeliner   r   define_objectmode_pipeline)ro   r   rY   rY   rZ   r     s    

zCompiler.define_pipelinesN)r>   r?   r@   r   r   rY   rY   rY   rZ   r     s   r   c                   @   sr   e Zd ZdZedddZedddZedd	d
ZedddZedddZ	edddZ
edddZdS )r   z
    This is the default pass builder, it contains the "classic" default
    pipelines as pre-canned PassManager instances:
      - nopython
      - objectmode
      - interpreted
      - typed
      - untyped
      - nopython lowering
    nopythonc                 C   s`   t }t|}|| }|j|j || }|j|j || }|j|j |  |S )z<Returns an nopython mode pipeline based PassManager
        )r   r   define_untyped_pipelineZpassesextenddefine_typed_pipeline!define_nopython_lowering_pipelinefinalize)r   rt   Zdpbr   Zuntyped_passestyped_passesZlowering_passesrY   rY   rZ   r   5  s    


z+DefaultPassBuilder.define_nopython_pipelinenopython_loweringc                 C   st   t |}|td |td |td | jjjrD|td n|t	d |t
d |td |  |S )N3ensure features that are in use are in a valid form$ensure IR is legal prior to loweringannotate typesnative parfor loweringnative loweringnopython mode backendzdump parfor diagnostics)r   add_passr/   r)   r$   r   rK   enabledr.   r-   r*   r(   r   r   rt   r   rY   rY   rZ   r   H  s    
z4DefaultPassBuilder.define_nopython_lowering_pipelineparfor_gufunc_nopython_loweringc                 C   sh   t |}|td |td |td | jjjrD|td n|t	d |t
d |  |S )Nr   r   r   r  r  r  )r   r  r/   r)   r$   r   rK   r  r.   r-   r*   r   r  rY   rY   rZ   /define_parfor_gufunc_nopython_lowering_pipeline\  s    
zBDefaultPassBuilder.define_parfor_gufunc_nopython_lowering_pipelinetypedc                 C   s   t |}|td |td |td | jjjrB|td | jj	sV|t
d | jjjr|td |td |td |  |S )	/Returns the typed part of the nopython pipelinenopython frontendremove phis nodeszinline overloaded functionszPreprocessing for parforsnopython rewriteszconvert to parforszfuse parforsparfor prelowering)r   r  r#   r,   r+   r   rK   r  r&   rM   r%   r'   r0   r1   r   r  rY   rY   rZ   r   p  s    

z(DefaultPassBuilder.define_typed_pipelineparfor_gufunc_typedc                 C   sB   t |}| jst|td |td |td |  |S )r
  processing IRr  r  )r   r   r   r  r   r#   r1   r   r  rY   rY   rZ   define_parfor_gufunc_pipeline  s    
z0DefaultPassBuilder.define_parfor_gufunc_pipelineuntypedc                 C   s   t |}| jdkr*|td |td |td |td |td | jj	sz|t
d |td |td	 |td
 |td |td | jj	s|td |td |td | jjr|td |td |  |S )z0Returns an untyped part of the nopython pipelineNanalyzing bytecodefix up argsr  zHandle with contexts(inline calls to locally defined closureszrewrite semantic constantszdead branch pruningr  zrewrite dynamic raises(convert make_function into JIT functionszinline inlinable functionszfind literally callszhandles literal_unrollZssazLiteral propagation)r   r   r  r   r   r   r   r   r   rM   r   r   r   r!   r   r   r   r   rD   r    r"   r   r  rY   rY   rZ   r     s6    
z*DefaultPassBuilder.define_untyped_pipelineobjectc                 C   s   t |}| jdkr,|td |td n|td |td |td |td |t	d |t
d	 |td
 |td |td |td |  |S )z:Returns an object-mode pipeline based PassManager
        Nr  r  r  r  zcanonicalize loop entryzcanonicalize loop exitzobject mode frontendr  r  r   r   zobject mode backend)r   r   r  r   r   r,   r   r   r   r2   r   r   r)   r$   r3   r   r  rY   rY   rZ   r     s(    
z-DefaultPassBuilder.define_objectmode_pipelineN)r   )r   )r  )r	  )r  )r  )r  )r>   r?   r@   r   staticmethodr   r   r  r   r  r   r   rY   rY   rY   rZ   r   *  s    
 ,r   c	           
      C   s   || ||||||}	|	 |S )aD  Compiler 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   )
r   r   r   r   r   r   r   rd   pipeline_classr   rY   rY   rZ   r     s       r   rY   c              	      s   |	rt |}d|_ f	dd}||  |}d}|jst < tdtj z||  |}W n tk
r   Y nX W 5 Q R X |dk	r|}n|}|S  |}|j	|dS dS )zK
    Compile a function with the given IR.

    For internal use only.
    Tc                    s$    |}|j | dS )Nr   rb   r   )r   )Zthe_irZ	the_flagsr   	r   rd   rb   r   r   r  r   r   r   rY   rZ   compile_local  s       z!compile_ir.<locals>.compile_localNignorer  )
r   deepcopyrM   warningscatch_warningssimplefilterr   NumbaWarningr   r   )r   r   r   r   r   r   r   rb   r   is_lifted_looprd   r  Z
norw_flagsr  Z	norw_cresZrw_crescresr   rY   r  rZ   r     s4    


   r   c           	      C   s   t | ||||||}||S )z 
    For internal use only.
    )r   r   )	r   r   rd   r   r   r   r   r   r   rY   rY   rZ   compile_internal=  s       r%  )FF)Ycollectionsr   r   r  Znumba.core.tracingr   
numba.corer   r   r   r   r   r	   r
   r   r   Znumba.parfors.parforr   numba.core.errorsr   numba.core.environmentr   Znumba.core.compiler_machineryr   numba.core.untyped_passesr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   numba.core.typed_passesr#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   Znumba.core.object_mode_passesr2   r3   Znumba.core.targetconfigr4   r5   r6   r7   DEFAULT_FLAGSrL   r   rj   r   r   r   r   r   r  r   r   r   dictr   r   r   r   r   r   r   r%  rY   rY   rY   rZ   <module>   s   ,PD
 R

	  8 
   
@