o 1&i@sddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl mZmZddlmZmZmZddlmZddlmZddlmZddlmZddlmZdd lmZdd l m!Z!e"gd Z#d d Z$ddZ%ddZ&Gddde'Z(GdddedZ)Gddde)Z*Gddde*Z+Gddde*Z,Gddde'Z-dd Z.Gd!d"d"e'Z/Gd#d$d$edZ0Gd%d&d&e0Z1Gd'd(d(e1Z2Gd)d*d*e1Z3d+d,Z4d-d.Z5dS)/N)abstractmethodABCMeta)utilsconfigcgutils)create_pass_builder)remove_redundant_nrt_refct)rtsys)require_global_compiler_lock)NumbaInvalidConfigWarning)disassemble_elf_to_cfg)PassTimingsCollection)x86i386Zi486Zi586i686Zi786Zi886Zi986cCs|dd}|tvS)N-r)split_x86arch)triplearchr=C:\wamp64\www\opt\env\Lib\site-packages\numba/core/codegen.py_is_x86src Csjtjd}|s dSd}|D]$}|}z |ttj|O}Wqty2t d|t Yqw|S)zParse refprune flags from the `config`. Invalid values are ignored an warn via a `NumbaInvalidConfigWarning` category. Returns ------- flags : llvmlite.binding.RefPruneSubpasses ,rzinvalid refprune flags ) rZLLVM_REFPRUNE_FLAGSrstripgetattrllZRefPruneSubpassesupperAttributeErrorwarningswarnr )flagsvalitemrrr_parse_refprune_flags s   r$c stjrHzddl}Wn tyd}t|wddlmddlm}ddlm}ddl m ddl m ||d}||fd d }nt }t d t |d d ||t d dS)Nrz0Please install pygments to see highlighted dumps) highlight)GasLexer) LlvmLexer)Terminal256Formatter)by_colorscheme)llvmasmcst|ddS)N)style)print)argr(r)r%lexerrrprinterHs   zdump..printerzP================================================================================Pr)rZHIGHLIGHT_DUMPSpygments ImportError ValueErrorr%Zpygments.lexersr&r'Zpygments.formattersr(Znumba.misc.dump_styler)r-center) headerbodylangr3msgZ gas_lexerZ llvm_lexerZ lexer_mapr1rr/rdump8s(         r;c@sDeZdZdZddZ    ddd Zdd d Zd dZddZdS)_CFGa Wraps the CFG graph for different display method. Instance of the class can be stringified (``__repr__`` is defined) to get the graph in DOT format. The ``.display()`` method plots the graph in PDF. If in IPython notebook, the returned image can be inlined. cKs2||_||_||_||}t||_||_dSN)cresnamepy_func get_functionrget_function_cfgdotkwargs)selfr>r?r@rDfnrrr__init__[s    z _CFG.__init__NTF c K sLddlddl ddl} ddl} ddlm} ddlm} ddlm } ddl m }d}| ||||||||d}| ||d d d }||||||d d dd<dd<dd<dd<dd<dd<dd<dd<dd<d d!<d"d#<|j |j}|j } fd$d%}|| fd&d'}||j|j|} d(} d)} d* d+} | d,} d-} d.} d/} d0} d1} d2} d3}!d4d5}"d6d7i}#i}$jr| |j\ | |d8d9}%|%jd:d;}&| |&d<}'d}(|'d=D]})d>\|)d?}*|)d@}+|)dA},|+|#|,<|*dBdC}*|*dD}-g}.dBd9}/||-dC}0|-}1|0dur|0d}2|2dE}3t|3dF}4g}5|jrldddG}6ni}6|3D]}7||7\}8}9|6|9dH}:|5 |4!|:|8|9qpd9"|5}/|-ddC}1dI};|. |;!dJ|1d#dK f dLdM}<|1dBdD]}=||=rϐqňj$s׈jr|<|=|.}>|>dur|>\|j%r|&|=rd}?n^|j'r|&|=rd}?nO|j(r|&|=rd}?n@|j)r!| &|=r!d}?n1|j*r0|!&|=r0d}?n"|j+r?|&|=r?d!}?n|j,rN|&|=rNd#}?ndJ}?|?dJus\|sp|"|=D] }@|. !|?|@qbq|/r}|. dN!|/d9"|.}A|ArdO|(|Af}BdP!|B}*nd9}*|j-|+|*dQqdR|'vr|'dRD]}C|CdA},|CdSd}D|CdT|CdU|Df|$|,<q|$.D]&\},}C|#|CdB}E|#|Cd}F|CdV}G|Gdur|EdW|G7}E|/|E|Fq|rg}H.D]\}I}J|H dX!|J|Iq|j-dYdZ!d9"|HdQ|dus|dur |j0|||d[|jd\d;S)]z "Pretty" prints the DOT graph of the CFG. For explanation of the parameters see the docstring for numba.core.dispatcher::inspect_cfg. rN)binding)List)SimpleNamespace) defaultdictF)increfdecrefreturnsraisesmeminfobranchesllvm_intrin_callsfunction_calls)pythonlineinfocSst|tr|jD]}t|||qdSt|trC|D]$\}}||jvr+td|t|tr7t|||qd}t|||fdSt|tr`|D]}||jvrWtd|t||dqJdSd}t|t|)z Parses the kwarg into a consistent format for use in configuring the Digraph rendering. _config is the configuration instance to update, kwarg is the kwarg on which to base the updates. zUnexpected key in kwarg: %sz$Unexpected value for key: %s, got:%sTz)Unhandled configuration type for kwarg %sN) isinstancebool__dict__setattrdictitemsr5settype)_configkwargattrkvr:r#rrr parse_configs*         z)_CFG.pretty_printer..parse_configcSdS)Nwhiterrrrrz%_CFG.pretty_printer..ZorangemarkeryellowrUgreenZtruebrredZfalsebrcyanrMZ turquoiserNZ lightpinkraiseZ lightseagreenrQpurplereturnZ rosybrownrStomatorTcsbi}d}|D]#}||}|dur.|}|dur.t|dks&J|d||d<q |S)z Gets the metadata entries from the LLVM IR, these look something like '!123 = INFORMATION'. Returns a map of metadata key to metadata value, i.e. from the example {'!123': INFORMATION}z(^[!][0-9]+)(\s+=\s+.*)Nr)compile splitlinesmatchgroupslen)llvm_strmdZmetadata_entryxrvg)rerr get_metadatas   z)_CFG.pretty_printer..get_metadatacsjd}t||krd|d}t|t|d|}j||d}|jdd|jddd t|d |S) NzCFG output filename "z9" exceeds maximum supported length, it will be truncated.)filenameTB)Zrankdirnodenonez%s)shapefontsize)rxrr r ZDigraphrastr)r?fnamerZcmaxZwstrf)gvrr init_digraphs     z)_CFG.pretty_printer..init_digraphz .*{(.*)}.*z .*<(.*)>(.*)z.*!dbg\s+(![0-9]+).*z7.*!DILocation\(line:\s+([0-9]+),\s+column:\s+([0-9]),.*z.*call void @llvm.dbg.value.*z @NRT_incref\bz @NRT_decref\bz @NRT_MemInfoz.*call.*@llvm\..*z .*call.*@.*z"store .*\!numba_exception_output.*zret i32 [^1],?.*cSstj|dddS)Nxz... )widthsubsequent_indent)textwrapwrap)srrrrz!_CFG.pretty_printer..wrapcSszd}t||krtt|}d|d||}t|}|dd}|dd}|dd}|d d }|d d }|S) Ni,z{}...z\{z{z\}z}\z\%z%!z!)rxrhashformathtmlescapereplace)rnhsrrrcleans        z"_CFG.pretty_printer..cleanz\l...Zdot_json)rutf-8objects)rlabelr?Z_gvidrsrz\l|z<{})TFrfzF{}defaultzF{}cs. |}|dur|}|durt|dksJ||d} |d}|dur|}|durt|dks>J||\}}|ksL|krjrhd} | |||f} d| } || jrt|d} | d} d| } || ||fSdSdSdSdSdS)z Search line `l` for metadata associated with python or line info and inject it into `new_lines` if requested. NrsrrrzMarker %s, Line %s, column %srirU) rvrwrxgetrVrappendrUint)l new_linesmatchedr|riZ debug_dataldlinecolZmfmtZ mark_linelnZlidxZ source_line) _interleavercol_spancscur_colcur_line firstlinenofmtlocation_entryrzmetadata_markersrc_coderrmetadata_interleaveisB         z0_CFG.pretty_printer..metadata_interleavez {}zS%s
z<{}>)redgesZtailportheadtailrrz:%sz;{}Keyzn<{}
Key:
>)rviewrsvg)1graphvizr}jsoninspectZllvmliterIZ numba.typedrJtypesrK collectionsrLr>rAr? get_llvm_strrtrUgetsourcelinesr@rBrZSourcepipeloadsdecoderrvrwrxrRrrrjoinrrVrMsearchrNrQrPrOrSrTrr\edgerender)KrErr render_formatr% interleaveZstrip_irZshow_keyrrrrrJrKrL_default _highlightrdrFryr~rrZ port_matchZport_jmp_matchZ location_exprZ dbg_valueZ nrt_increfZ nrt_decrefZ nrt_meminfoZll_intrin_callsZll_function_callZll_raiseZ ll_returnrZnode_idsZedge_idsZraw_dotZ json_bytesZjzonZidcobjrr?ZgvidlinesrZ port_linerZ sliced_linesZportsZ ports_tokensZtdfmtZtbl_datacolorstoktargetvaluecolorZ fmtheaderrrZupdated_lineinfocolourr{ZdattabrtprrportZkey_tabrbrcr)rrrrrrrrrrrzrr}rrpretty_printercs>                                "+                     z_CFG.pretty_printerpdfcCs$|jd|||d|j}|dS)a Plot the CFG. In IPython notebook, the return image object can be inlined. The *filename* option can be set to a specific path for the rendered output to write to. If *view* option is True, the plot is opened by the system default application for the image format (PDF). *format* can be any valid format string accepted by graphviz, default is 'pdf'. )rrrrNrrrDr)rErrrZrawbytrrrdisplays  z _CFG.displaycCs|jdi|jdS)NrrrrErrr _repr_svg_sz_CFG._repr_svg_cC|jSr=)rCrrrr__repr__sz _CFG.__repr__)NNNTFFTrH)NrF) __name__ __module__ __qualname____doc__rGrrrrrrrrr<Ss   r<c@seZdZdZdZdZdZdddefddZe dd Z e d d Z e d d Z e ddZ ddZddZddZddZeddZeddZeddZeddZed d!Zed"d#Zd$d%Zd&d'Zd(d)Zd*S)+ CodeLibraryz An interface for bundling LLVM code together and compiling it. It is tied to a *codegen* instance (e.g. JITCPUCodegen) that will determine how the LLVM code is transformed and linked together. Fcodegen CPUCodegenr?cCs6||_||_|jjd|jd}t||_g|_dS)N())_codegen_name __class__rr _recorded_timings_dynamic_globals)rErr?Zptc_namerrrrGs   zCodeLibrary.__init__cCs|t|jdkS)Nr)_ensure_finalizedrxrrrrrhas_dynamic_globalsszCodeLibrary.has_dynamic_globalscCrr=)rrrrrrecorded_timingszCodeLibrary.recorded_timingscCr)z9 The codegen object owning this library. )rrrrrr"zCodeLibrary.codegencCrr=)rrrrrr?)rzCodeLibrary.namecCsd|jt|fS)Nz)r?idrrrrr-szCodeLibrary.__repr__cCs|jr td|fdS)Nz+operation impossible on finalized object %r) _finalized RuntimeErrorrrrr_raise_if_finalized0s zCodeLibrary._raise_if_finalizedcCs|js |dSdSr=)rfinalizerrrrr5s zCodeLibrary._ensure_finalizedcCs||j|}|S)zC Create an LLVM IR module for use by this library. )rr_create_empty_modulerEr? ir_modulerrrcreate_ir_module9s zCodeLibrary.create_ir_modulecCre)zk Add a library for linking into this library, without losing the original library. NrrElibraryrrradd_linking_libraryArhzCodeLibrary.add_linking_librarycCre)zC Add an LLVM IR module's contents to this library. Nr)rErrrr add_ir_moduleHrhzCodeLibrary.add_ir_modulecCre)z Finalize the library. After this call, nothing can be added anymore. Finalization involves various stages of code optimization and linking. NrrrrrrNrhzCodeLibrary.finalizecCre)z5 Return the function named ``name``. NrrEr?rrrrAVrhzCodeLibrary.get_functioncCre)zA Get the human-readable form of the LLVM module. Nrrrrrr\rhzCodeLibrary.get_llvm_strcCre)z2 Get the human-readable assembly. Nrrrrr get_asm_strbrhzCodeLibrary.get_asm_strcCsd|_d|_d|_dS)NTF)_object_caching_enabled_compiled_object _compiledrrrrenable_object_cachingls z!CodeLibrary.enable_object_cachingcCs2|js td|f|jdurtd|f|jS)N object caching not enabled in %szno compiled object yet for %s)rr5r rrrrr_get_compiled_objectqs  z CodeLibrary._get_compiled_objectcCs8|js td|f|jrtd|f||_d|_dS)Nr zlibrary already compiled: %sT)rr5r r _disable_inspection)rErrrr_set_compiled_objectxs  z CodeLibrary._set_compiled_objectN)rrrrrrrrrGpropertyrrrr?rrrrrrrrrArrr r rrrrrrsB            r) metaclasscseZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd2d"d#Zd$d%Zed&d'Zed(d)Zed*d+Zd,d-Zd.d/Zed0d1ZZS)3CPUCodeLibrarycsHt||g|_tt|j|j|_ t |j|j _d|_ dSr=) superrG_linking_librariesrparse_assemblyrrrr? _final_modulernormalize_ir_text_shared_module)rErr?rrrrGs zCPUCodeLibrary.__init__c Csn|jj|_|jD],}|j\}}d|j}|j|||||Wdn1s/wYqdS)zP Internal: run function-level optimizations inside *ll_module*. zFunction passes on N) r _data_layout data_layout functions_function_pass_managerr?rrecordrun)rE ll_modulefuncZfpmpbrbrrr_optimize_functionss   z"CPUCodeLibrary._optimize_functionscCs|jj|jjd|jjdd\}}|j\}}d}|j||||j|Wdn1s2wYtj s@t |j|_d}|j||||j|WddS1s\wYdS)zA Internal: optimize this library's final module. Fcheap)loop_vectorize slp_vectorizeoptcostz/Module passes (cheap optimization for refprune)Nz!Module passes (full optimization)) r_module_pass_manager _loopvect _opt_levelrrrrrLLVM_REFPRUNE_PASSr)rEZ mpm_cheapZ mpb_cheapZmpm_fullZmpb_fullZ cheap_nameZ full_namerrr_optimize_final_modules"  "z%CPUCodeLibrary._optimize_final_modulecCs||jdur |jS|j}g}d}|jD]}|d7}|js,|jtjjkr,| |j q|dkr8t d|f|rI| }|D]}d| |_q@||_|S)a: Internal: get a LLVM module suitable for linking multiple times into another library. Exported functions are made "linkonce_odr" to allow for multiple definitions, inlining, and removal of unused exports. See discussion in https://github.com/numba/numba/pull/890 Nrrsz7library unfit for linking: no available functions in %sZ linkonce_odr)rrrris_declarationlinkagerZLinkageexternalrr?rclonerA)rEmodZto_fixZnfuncsrFr?rrr_get_module_for_linkings*   z&CPUCodeLibrary._get_module_for_linkingcCs||j|dSr=)rrrrrrrrsz"CPUCodeLibrary.add_linking_librarycCsN|t|tjs Jtt|}t|}|j |_ | | |dSr=) rrWllvmirModulerrrrrr?verifyadd_llvm_module)rErirr rrrrs zCPUCodeLibrary.add_ir_modulecCs(||tjs t|}|j|dSr=)r#rr,rrlink_in)rEr rrrr7s zCPUCodeLibrary.add_llvm_modulecCst|j|tjrtd|j|dt }|j D]}||vr5| ||j j |ddq ||j |dS)NzFUNCTION OPTIMIZED DUMP %sr*T)Zpreserve)r r_check_llvm_bugsrrZ DUMP_FUNC_OPTr;r?rr]raddrr9r3r-r6_finalize_final_module)rEseenrrrrrs$      zCPUCodeLibrary.finalizecCs,|jjD]}|jdr|j|jqdS)Nznumba.dynamic.globals)rglobal_variablesr? startswithrr)rErrrr_finalize_dynamic_globalss  z(CPUCodeLibrary._finalize_dynamic_globalscCs8|jjD]}|jr|jdrd}t||jqdS)NZ _ZN5numbazSymbol {} not linked properly)rrr.r?r?AssertionErrorr)rErFr:rrr_verify_declare_only_symbols s z+CPUCodeLibrary._verify_declare_only_symbolscCs||t||j_|j|j}|rt||| d|_ t j r3t d|j|dt jrCt d|j|ddSdS)z? Make the underlying LLVM module ready to use. TzOPTIMIZED DUMP %sr*z ASSEMBLY %sr+N)r@rBweakrefproxyr_CPUCodeLibrary__libraryr _add_moduler_finalize_specificrrZDUMP_OPTIMIZEDr;r?rZ DUMP_ASSEMBLYr)rEcleanuprrrr<s z%CPUCodeLibrary._finalize_final_moduleccs$|j}|jD]}|js|VqdS)zj Get all functions defined in the library. The library must have been finalized. N)rrr.)rEr2rFrrrget_defined_functions.s z$CPUCodeLibrary.get_defined_functionscCs |j|Sr=)rrArrrrrA8s zCPUCodeLibrary.get_functioncCs|jr tddSdS)Nz@Inspection disabled for cached code. Invalid result is returned.)rrr rrrr _sentry_cache_disable_inspection;sz/CPUCodeLibrary._sentry_cache_disable_inspectioncCs|t|jSr=)rJrrrrrrr@s zCPUCodeLibrary.get_llvm_strcCs|t|jj|jSr=)rJrr_tmZ emit_assemblyrrrrrrDszCPUCodeLibrary.get_asm_strNcKs|t|||fi|S)z= Get control-flow graph of the LLVM function )rJr<)rEr?r@rDrrrrBHszCPUCodeLibrary.get_function_cfgcCs|}t||S)aa Get the CFG of the disassembly of the ELF object at symbol mangled_name. Requires python package: r2pipe Requires radare2 binary on $PATH. Notebook rendering requires python package: graphviz Optionally requires a compiler toolchain (via pycc) to link the ELF to get better disassembly results. )r r )rEZ mangled_nameelfrrrget_disasm_cfgOs zCPUCodeLibrary.get_disasm_cfgc Csddlm}ddlm}ddlm}|||}td|D]?}|ddkr_t| dd d }td |D]&}|j s>q8td |j |d |d| |dd| |ddfq8q tdS)zw Dump the symbol table of an ELF file. Needs pyelftools (https://github.com/eliben/pyelftools) r)ELFFile) descriptions)BytesIOz ELF file:Zsh_typeZ SHT_SYMTABcSrr=r?)symrrrrgisz*CPUCodeLibrary._dump_elf..)keyz symbols:z/ - %r: size=%d, value=0x%x, type=%s, bind=%sst_sizeZst_valueZst_infor^bindN)Zelftools.elf.elffilerNZ elftools.elfrOiorPr-Z iter_sectionssortedZ iter_symbolsr?rZdescribe_symbol_typeZdescribe_symbol_bind) clsbufrNrOrPrsecsymbolsrRrrr _dump_elf\s,       zCPUCodeLibrary._dump_elfcCs:z|j}Wn tyYdSw|jrd|_||_dSdS)zB `ll_module` was compiled into object code `buf`. NT)rErrr r )rXr rYrErrr_object_compiled_hookws   z$CPUCodeLibrary._object_compiled_hookcCsDz|j}Wn tyYdSw|jr|jr |j}d|_|SdSdS)z> Return a cached object code for `ll_module`. N)rErrr )rXr rErYrrr_object_getbuffer_hooks   z%CPUCodeLibrary._object_getbuffer_hookcCs||jd|jfS)zX Serialize this library using its bitcode as the cached representation. bitcode)rr?r as_bitcoderrrrserialize_using_bitcodesz&CPUCodeLibrary.serialize_using_bitcodecCs(|||f}|jd|fS)z Serialize this library using its object code as the cached representation. We also include its bitcode for further inlining with other libraries. object)rr r3r`r?)rEdatarrrserialize_using_object_codes   z*CPUCodeLibrary.serialize_using_object_codec Cs|\}}}||}t||sJ|dkr!t||_||S|dkrF|\}}|||t||_||j j |j|St d|f)Nr_rbz!unsupported serialization kind %r) create_libraryrWrZ parse_bitcoderr<r rrr_engine_load_defined_symbolsr5) rXrstater?kindrcrEZ object_codeZshared_bitcoderrr _unserializes     zCPUCodeLibrary._unserializer=)rrrrGr#r-r3rrr7rr@rBr<rIrArJrrrBrM classmethodr\r]r^rardrj __classcell__rrrrrs: #        rc@s$eZdZddZddZddZdS)AOTCodeLibrarycCs||jj|jS)z Return this library as a native object (a bytestring) -- for example ELF under Linux. This function implicitly calls .finalize(). )rrrKZ emit_objectrrrrremit_native_objectsz!AOTCodeLibrary.emit_native_objectcCs||jS)zz Return this library as LLVM bitcode (a bytestring). This function implicitly calls .finalize(). )rrr`rrrr emit_bitcodes zAOTCodeLibrary.emit_bitcodecCdSr=rrrrrrGz!AOTCodeLibrary._finalize_specificN)rrrrnrorGrrrrrms rmc@seZdZddZddZdS)JITCodeLibrarycCs,||jj}||sdS|jj|S)a Generate native code for function named *name* and return a pointer to the start of the function (as an integer). This function implicitly calls .finalize(). Returns ------- pointer : int - zero (null) if no symbol of *name* is defined by this code library. - non-zero if the symbol is defined. r)rrrfis_symbol_definedget_function_address)rEr?eerrrget_pointer_to_functions  z&JITCodeLibrary.get_pointer_to_functioncCsL|j|j|jd|jjWddS1swYdS)NzFinalize object)r_scan_and_fix_unresolved_refsrrZ record_legacyrffinalize_objectrrrrrGs"z!JITCodeLibrary._finalize_specificN)rrrrvrGrrrrrrs rrc@s4eZdZdZdZddZddZddZd d Zd S) RuntimeLinkerzP For tracking unresolved symbols generated at runtime due to recursion. z.numba.unresolved$cCst|_t|_g|_dSr=)rZ UniqueDict _unresolvedr]_defined _resolvedrrrrrGs  zRuntimeLinker.__init__cCst|j}|jD]1}|j|r7|jt|d}||jrqtjd}t |}| |t |||j |<qdS)zr Scan and track all unresolved external symbols in the module and allocate memory for it. NZnrt_unresolved_abort)PREFIXr>r?r?rxrsr rrvctypesc_void_padd_global_mapping addressofrz)rEmoduleengineprefixrrRZabortfnptrrrrscan_unresolved_symbolss      z%RuntimeLinker.scan_unresolved_symbolscCs$|jD] }|js|j|jqdS)z5 Scan and track all defined symbols. N)rr.r{r;r?)rErrFrrrscan_defined_symbolss z"RuntimeLinker.scan_defined_symbolscsTfddjD}|D]}||}j|}||_j||fj|=q dS)z= Fix unresolved symbols if they are defined. csg|] }|jvr|qSr)r{).0r?rrr sz)RuntimeLinker.resolve..N)rzrtrr|r)rErpendingr?fnptrrrrrresolves   zRuntimeLinker.resolveN) rrrrr}rGrrrrrrrrys rycstfdd}|S)Ncs|jg|Ri|Sr=)_ee)rEargsrDoldrrwrapper)sz_proxy..wrapper) functoolswraps)rrrrr_proxy(src@sheZdZdZddZddZddZdd Zd d Ze e j j Z e e j j Z e e j jZe e j jZd S) JitEnginezWraps an ExecutionEngine to provide custom symbol tracking. Since the symbol tracking is incomplete (doesn't consider loaded code object), we are not putting it in llvmlite. cCs||_t|_dSr=)rr]_defined_symbols)rErurrrrG4s zJitEngine.__init__cCs ||jvS)z/Is the symbol defined in this session? )rrrrrrsAs zJitEngine.is_symbol_definedcCs.|j|jfD]}|jdd|DO_qdS)z(Extract symbols from the module cSsh|]}|js|jqSr)r.r?)rrrrr Js z2JitEngine._load_defined_symbols..N)rr>r)rEr2ZgsetsrrrrgFszJitEngine._load_defined_symbolscCs|||j|S)zXOverride ExecutionEngine.add_module to keep info about defined symbols. )rgr add_modulerErrrrrMs  zJitEngine.add_modulecCs|j|j|j||S)z`Override ExecutionEngine.add_global_mapping to keep info about defined symbols. )rr;r?rr)rEraddrrrrrTszJitEngine.add_global_mappingN)rrrrrGrsrgrrrrZExecutionEngineset_object_cacherxrtget_global_value_addressrrrrr/s   rc@sDeZdZdZeddZeddZeddZdd Z d d Z d S) CodegenaM Base Codegen class. It is expected that subclasses set the class attribute ``_library_class``, indicating the CodeLibrary class for the target. Subclasses should also initialize: ``self._data_layout``: the data layout for the target. ``self._target_data``: the binding layer ``TargetData`` for the target. cCre)zD Create a new empty module suitable for the target. NrrrrrrqrhzCodegen._create_empty_modulecCre)zu Add a module to the execution engine. Ownership of the module is transferred to the engine. NrrrrrrFwrhzCodegen._add_modulecCr)zJ The LLVM "target data" object for this codegen instance. ) _target_datarrrr target_data~rzCodegen.target_datacKs|j||fi|S)zb Create a :class:`CodeLibrary` object for use with this codegen instance. )_library_class)rEr?rDrrrreszCodegen.create_librarycCs|j||Sr=)rrj)rEZ serializedrrrunserialize_libraryszCodegen.unserialize_libraryN) rrrrrrrFrrrerrrrrrfs    rc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)rcCsBtd|_tt|||_d|j_t|_ | |jdS)NZglobal_codegen_module) initialize_llvmrrrrr _llvm_moduler?ry _rtlinker_init)rE module_namerrrrGs zCPUCodegen.__init__cCst|jgks Jdtjt}ttjd}| |_ | ||j di|}tj }tj|||d}tjr=|||_t||_|j|_t|j|_tjjrZd|_d|_nd|_d|_|j|jj|jjdS) NzModule isn't empty)r')use_lmmTFrr)listr>rTargetZ from_tripleget_process_tripler[rOPT_customize_tm_features _tm_features_customize_tm_optionscreate_target_machineZUSE_LLVMLITE_MEMORY_MANAGERZcreate_mcjit_compilerZENABLE_PROFILINGZenable_jit_eventsrKrrfrrrrZ is_opt_maxr*r+rrr]r^)rEZ llvm_modulerZ tm_optionstmrrrrrrs,      zCPUCodegen._initcCs,tt|}t|_|jr|j|_|Sr=) r4r5rrrrrrrrrrrrs  zCPUCodegen._create_empty_modulecKsp|dd}|jdi|}|}|dur+|dkr+tjdkr+|||tjr4| t ||fS)Nr(r$rr) pop _pass_builderZgetModulePassManagerrrZadd_loop_rotate_passZadd_instruction_combine_passZadd_jump_threading_passr,add_refprune_passr$)rErDr(r"pmrrrr)s  zCPUCodegen._module_pass_managercKs2|jdi|}|}tjr|t||fS)Nr)rZgetFunctionPassManagerrr,rr$)rErDr"rrrrrs  z!CPUCodegen._function_pass_managercKsH|dtj}|dtj}|dtj}t|jf|||d|}|S)Nr'r%r&)r'r%r&)rrrZLOOP_VECTORIZEZ SLP_VECTORIZErrK)rErD opt_levelr%r&r"rrrrs zCPUCodegen._pass_buildercCsVd}t|}t|}d|vsd|vrdSd|vr$t}td|ftd|f)z< Guard against some well-known LLVM bug(s). zo define double @func() { ret double 1.23e+01 } z12.3z1.23Nz1.0zLLVM will produce incorrect floating-point code in the current locale %s. Please read https://numba.readthedocs.io/en/stable/user/faq.html#llvm-locale-bug for more information.zUnexpected IR: %s )rrrlocale getlocalerrA)rEr8r2Zir_outlocrrrr:s zCPUCodegen._check_llvm_bugscCs|jj||jfS)zP Return a tuple unambiguously describing the codegen behaviour. )rr_get_host_cpu_namerrrrr magic_tuples zCPUCodegen.magic_tuplecCs.|j||j|j||j|jdSr=)rrrfrrrrrrrws z(CPUCodegen._scan_and_fix_unresolved_refscCsntd}|jj|}|j}z||}Wnty+tj|||d}d|_ Ynw| | ||S)NrQr0) r4ZIntTypeZ as_pointerrr}rZ get_globalKeyErrorZGlobalVariabler/Zbitcastload)rEZbuilderZfntyr?ZvoidptrZptrnameZllvm_modrrrrinsert_unresolved_refs   z CPUCodegen.insert_unresolved_refcCstjdur tStjSr=)rZCPU_NAMErZget_host_cpu_namerrrrr&s zCPUCodegen._get_host_cpu_namecCstjdurtjStSr=)rZ CPU_FEATURESget_host_cpu_featuresrrrr_get_host_cpu_features+s z!CPUCodegen._get_host_cpu_featuresN)rrrrGrrr)rrr:rrwrrrrrrrrs (  rc@s6eZdZdZeZd ddZddZddZd d Z dS) AOTCPUCodegenzp A codegen implementation suitable for Ahead-Of-Time compilation (e.g. generation of object files). NcCs|pd|_t||dSNr) _cpu_namerrG)rErcpu_namerrrrG9s zAOTCPUCodegen.__init__cCs<|j}|dkr |}||d<d|d<d|d<|j|d<dS)Nhostcpupicrelocr codemodelfeatures)rrr)rEoptionsrrrrr>sz#AOTCPUCodegen._customize_tm_optionscCrerrrrrrrGsz$AOTCPUCodegen._customize_tm_featurescCrpr=rrrrrrFLrqzAOTCPUCodegen._add_moduler=) rrrrrmrrGrrrFrrrrr1s  rc@s4eZdZdZeZddZddZddZdd Z d S) JITCPUCodegenzI A codegen implementation suitable for Just-In-Time compilation. cCs~||d<tjj}|drd}n |drd}nd}||d<d|d <|j|d <ttjj }d |j vr=d |d <dSdS) NrrZstaticppcrrrZ jitdefaultrrZjitT) rrrZfrom_default_tripler?r?rrZ pysignaturer parameters)rErrZ reloc_modelsigrrrrWs       z#JITCPUCodegen._customize_tm_optionscCs|Sr=)rrrrrrrsz$JITCPUCodegen._customize_tm_featurescCs|j|dSr=)rfrrrrrrFvrzJITCPUCodegen._add_modulecCs2|j|}tjd|}tt||d<dS)zrSet the environment address. Update the GlobalVariable named *env_name* to the address of *env*. rsrN)rfrr~r from_addressr)rEenv_nameenvZgvaddrZenvptrrrrset_env~s zJITCPUCodegen.set_envN) rrrrrrrrrrFrrrrrrPs rcCsttdS)z Safe to use multiple times. N)rZinitialize_native_targetZinitialize_native_asmprinterrrrrrs rcCsLzt}Wn tyYdSwtjs"|D] }|dr!d||<q|S)z~Get host CPU features using LLVM. The features may be modified due to user setting. See numba.config.ENABLE_AVX. rZavxF)rrrrZ ENABLE_AVXr?flatten)rrbrrrrs   r)6rrrrCr~rrZllvmlite.bindingrIrZ llvmlite.irr8r4abcrrZ numba.corerrrZnumba.core.llvm_bindingsrZnumba.core.runtime.nrtoptrZnumba.core.runtimer Znumba.core.compiler_lockr Znumba.core.errorsr Znumba.misc.inspectionr Znumba.misc.llvm_pass_timingsr frozensetrrr$r;rbr<rrrmrrryrrrrrrrrrrrrsT          6{<77*"8