o 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)x86i386i486i586i686i786i886i986cCs|dd}|tvS)N-r)split_x86arch)triplearchrP/home/ubuntu/transcripts/venv/lib/python3.10/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 ) rLLVM_REFPRUNE_FLAGSrstripgetattrllRefPruneSubpassesupperAttributeErrorwarningswarnr )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/r0r,lexerrrprinterHs   zdump..printerzP================================================================================Pr)rHIGHLIGHT_DUMPSpygments ImportError ValueErrorr,pygments.lexersr-r.pygments.formattersr/numba.misc.dump_styler0r4center) headerbodylangr;msg gas_lexer llvm_lexer lexer_mapr8rr6rdump8s(         rIc@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_functionr"get_function_cfgdotkwargs)selfrLrMrNrRfnrrr__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__setattrdictitemsr=settype)_configkwargattrkvrEr*rrr parse_configs*         z)_CFG.pretty_printer..parse_configcSdS)Nwhiterrrrrz%_CFG.pretty_printer..orangemarkeryellowrcgreentruebrredfalsebrcyanr[ turquoiser\ lightpinkraise lightseagreenr_purplereturn rosybrownratomatorbcsbi}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_strmdmetadata_entryxrg)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)rankdirnodenonez%s)shapefontsize)rr&r'r Digraphrostr)rMfnamercmaxwstrf)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!)rrhashformathtmlescapereplace)rnhsrrrcleans        z"_CFG.pretty_printer..cleanz\l...dot_json)rutf-8objects)rlabelrM_gvidrrz\l|z<{})TFrtzF{}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. NrrrzMarker %s, Line %s, column %srxrc) rrrgetrdrappendrcint)l new_linesmatchedrrx debug_dataldlinecolmfmt mark_linelnlidx source_line) _interleavercol_spancscur_colcur_line firstlinenofmtlocation_entryrmetadata_markersrc_coderrmetadata_interleaveisB         z0_CFG.pretty_printer..metadata_interleavez {}zS%s
z<{}>)redgestailportheadtailrz:%sz;{}Keyzn<{}
Key:
>)rviewrsvg)1graphvizrjsoninspectllvmliterW numba.typedrXtypesrY collectionsrZrLrOrM get_llvm_strrrcgetsourcelinesrNrPrSourcepipeloadsdecoderrrrr`rrrjoinr rdr[searchr\r_r^r]rarbrrjedgerender)KrSrr render_formatr, interleavestrip_irshow_keyrrrr"rXrYrZ_default _highlightrrrTrrrr port_matchport_jmp_match location_expr dbg_value nrt_incref nrt_decref nrt_meminfoll_intrin_callsll_function_callll_raise ll_returnrnode_idsedge_idsraw_dot json_bytesjzonidcobjrrMgvidlinesr port_liner sliced_linesports ports_tokenstdfmttbl_datacolorstoktargetvaluecolor fmtheaderrrupdated_lineinfocolourrdattabrtprrportkey_tabrprqr)rrrrrrrrrrrrrrrpretty_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'. )rrrrNrr!rRr)rSrrrrawbytrrrdisplays  z _CFG.displaycCs|jdi|jdS)Nrrr#rSrrr _repr_svg_sz_CFG._repr_svg_cC|jSrK)rQr&rrr__repr__sz _CFG.__repr__)NNNTFFTrV)Nr"F) __name__ __module__ __qualname____doc__rUr!r%r'r)rrrrrJSs   rJc@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 CPUCodegenrMcCs>||_||_|jjd|jd}t||_g|_t|_dS)N()) _codegen_name __class__r*r _recorded_timings_dynamic_globalsrk _reload_init)rSr/rMptc_namerrrrUs   zCodeLibrary.__init__cCs|t|jdkS)Nr)_ensure_finalizedrr7r&rrrhas_dynamic_globalsszCodeLibrary.has_dynamic_globalscCr(rK)r6r&rrrrecorded_timings zCodeLibrary.recorded_timingscCr()z9 The codegen object owning this library. )r3r&rrrr/$zCodeLibrary.codegencCr(rK)r4r&rrrrM+r=zCodeLibrary.namecCsd|jt|fS)Nz)rMidr&rrrr)/szCodeLibrary.__repr__cCs|jr td|fdS)Nz+operation impossible on finalized object %r) _finalized RuntimeErrorr&rrr_raise_if_finalized2s zCodeLibrary._raise_if_finalizedcCs|js |dSdSrK)r@finalizer&rrrr:7s zCodeLibrary._ensure_finalizedcCs||j|}|S)zC Create an LLVM IR module for use by this library. )rBr3_create_empty_modulerSrM ir_modulerrrcreate_ir_module;s zCodeLibrary.create_ir_modulecCrs)zk Add a library for linking into this library, without losing the original library. NrrSlibraryrrradd_linking_libraryCrvzCodeLibrary.add_linking_librarycCrs)zC Add an LLVM IR module's contents to this library. Nr)rSrFrrr add_ir_moduleJrvzCodeLibrary.add_ir_modulecCrs)z Finalize the library. After this call, nothing can be added anymore. Finalization involves various stages of code optimization and linking. Nrr&rrrrCPrvzCodeLibrary.finalizecCrs)z5 Return the function named ``name``. NrrSrMrrrrOXrvzCodeLibrary.get_functioncCrs)zA Get the human-readable form of the LLVM module. Nrr&rrrr^rvzCodeLibrary.get_llvm_strcCrs)z2 Get the human-readable assembly. Nrr&rrr get_asm_strdrvzCodeLibrary.get_asm_strcCsd|_d|_d|_dS)NTF)_object_caching_enabled_compiled_object _compiledr&rrrenable_object_cachingns z!CodeLibrary.enable_object_cachingcCs2|js td|f|jdurtd|f|jS)N object caching not enabled in %szno compiled object yet for %s)rNr=rOrAr&rrr_get_compiled_objectss  z CodeLibrary._get_compiled_objectcCs8|js td|f|jrtd|f||_d|_dS)NrRzlibrary already compiled: %sT)rNr=rPrO_disable_inspection)rSrrrr_set_compiled_objectzs  z CodeLibrary._set_compiled_objectN)r*r+r,r-r@rNrTrrUpropertyr;r<r/rMr)rBr:rGrrJrKrCrOrrMrQrSrUrrrrr.sB            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|_ dSrK) superrU_linking_librariesr"parse_assemblyrr3rDrM _final_modulernormalize_ir_text_shared_module)rSr/rMr5rrrUs 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) r3 _data_layout data_layout functions_function_pass_managerrMr6recordrun)rS ll_modulefuncfpmpbrprrr_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)) r3_module_pass_manager _loopvect _opt_levelr6rdrer\rLLVM_REFPRUNE_PASSr)rS mpm_cheap mpb_cheapmpm_fullmpb_full cheap_name 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 Nrrz7library unfit for linking: no available functions in %s linkonce_odr)r:r^r\rbis_declarationlinkager"LinkageexternalrrMrAclonerO)rSmodto_fixnfuncsrTrMrrr_get_module_for_linkings*   z&CPUCodeLibrary._get_module_for_linkingcCs||j|dSrK)r:rZrrHrrrrJsz"CPUCodeLibrary.add_linking_librarycCsN|t|tjs Jtt|}t|}|j |_ | | |dSrK) rBrellvmirModulerr]rr"r[rMverifyadd_llvm_module)rSrFirrfrrrrKs zCPUCodeLibrary.add_ir_modulecCs(||tjs t|}|j|dSrK)rjrrsrr\link_in)rSrfrrrrs zCPUCodeLibrary.add_llvm_modulecCst|j|tjrtd|j|dt }|j D]}||vr<|j |j | ||jj|ddq ||j|dS)NzFUNCTION OPTIMIZED DUMP %sr1T)preserve)r r3_check_llvm_bugsrBr DUMP_FUNC_OPTrIrMrrkrZr8updateaddr\rrrzr_finalize_final_module)rSseenrIrrrrCs&      zCPUCodeLibrary.finalizecCs,|jjD]}|jdr|j|jqdS)Nznumba.dynamic.globals)r\global_variablesrM startswithr7r)rSrrrr_finalize_dynamic_globals s  z(CPUCodeLibrary._finalize_dynamic_globalscCs8|jjD]}|jr|jdrd}t||jqdS)N _ZN5numbazSymbol {} not linked properly)r\rbr|rMrAssertionErrorr)rSrTrErrr_verify_declare_only_symbolss 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 %sr1z ASSEMBLY %sr2N)rrweakrefproxyr\_CPUCodeLibrary__libraryr3 _add_modulerC_finalize_specificr@rDUMP_OPTIMIZEDrIrMr DUMP_ASSEMBLYrM)rScleanuprrrrs z%CPUCodeLibrary._finalize_final_moduleccs$|j}|jD]}|js|VqdS)zj Get all functions defined in the library. The library must have been finalized. N)r\rbr|)rSrrTrrrget_defined_functions2s z$CPUCodeLibrary.get_defined_functionscCs |j|SrK)r\rOrLrrrrO<s zCPUCodeLibrary.get_functioncCs|jr tddSdS)Nz@Inspection disabled for cached code. Invalid result is returned.)rTr&r'r&rrr _sentry_cache_disable_inspection?sz/CPUCodeLibrary._sentry_cache_disable_inspectioncCs|t|jSrK)rrr\r&rrrrDs zCPUCodeLibrary.get_llvm_strcCs|t|jj|jSrK)rrr3_tm emit_assemblyr\r&rrrrMHszCPUCodeLibrary.get_asm_strNcKs|t|||fi|S)z= Get control-flow graph of the LLVM function )rrJ)rSrMrNrRrrrrPLszCPUCodeLibrary.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. )rSr )rS mangled_nameelfrrrget_disasm_cfgSs 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:sh_type SHT_SYMTABcSr(rKrM)symrrrrumsz*CPUCodeLibrary._dump_elf..)keyz symbols:z/ - %r: size=%d, value=0x%x, type=%s, bind=%sst_sizest_valuest_inforlbindN)elftools.elf.elffiler elftools.elfriorr4 iter_sectionssorted iter_symbolsrMrdescribe_symbol_typedescribe_symbol_bind) clsbufrrrrsecsymbolsrrrr _dump_elf`s,       zCPUCodeLibrary._dump_elfcCs:z|j}Wn tyYdSw|jrd|_||_dSdS)zB `ll_module` was compiled into object code `buf`. NT)rr%rNrPrO)rrfrrSrrr_object_compiled_hook{s   z$CPUCodeLibrary._object_compiled_hookcCsDz|j}Wn tyYdSw|jr|jr |j}d|_|SdSdS)z> Return a cached object code for `ll_module`. N)rr%rNrO)rrfrSrrrr_object_getbuffer_hooks   z%CPUCodeLibrary._object_getbuffer_hookcCs||jd|jfS)zX Serialize this library using its bitcode as the cached representation. bitcode)r:rMr\ as_bitcoder&rrrserialize_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)r:rSrrrM)rSdatarrrserialize_using_object_codes   z*CPUCodeLibrary.serialize_using_object_codec Cs|\}}}||}t||sJ|dkr!t||_||S|dkrF|\}}|||t||_||j j |j|St d|f)Nrrz!unsupported serialization kind %r) create_libraryrer" parse_bitcoder\rrQrUr^r3_engine_load_defined_symbolsr=) rr/staterMkindrrS object_codeshared_bitcoderrr _unserializes     zCPUCodeLibrary._unserializerK)r*r+r,rUrjrzrrJrKrrCrrrrrOrrrMrPr classmethodrrrrrr __classcell__rrr_rrXs: #        rXc@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(). )r:r3r emit_objectr\r&rrremit_native_objectsz!AOTCodeLibrary.emit_native_objectcCs||jS)zz Return this library as LLVM bitcode (a bytestring). This function implicitly calls .finalize(). )r:r\rr&rrr emit_bitcodes zAOTCodeLibrary.emit_bitcodecCdSrKrr&rrrrz!AOTCodeLibrary._finalize_specificN)r*r+r,rrrrrrrrs rc@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)r:r3ris_symbol_definedget_function_address)rSrMeerrrget_pointer_to_functions  z&JITCodeLibrary.get_pointer_to_functioncCsL|j|j|jd|jjWddS1swYdS)NzFinalize object)r3_scan_and_fix_unresolved_refsr\r6 record_legacyrfinalize_objectr&rrrrs"z!JITCodeLibrary._finalize_specificN)r*r+r,rrrrrrrs rc@s4eZdZdZdZddZddZddZd d Zd S) RuntimeLinkerzP For tracking unresolved symbols generated at runtime due to recursion. z.numba.unresolved$cCst|_t|_g|_dSrK)r UniqueDict _unresolvedrk_defined _resolvedr&rrrrUs  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. Nnrt_unresolved_abort)PREFIXrrMrrrr rIrctypesc_void_padd_global_mapping addressofr)rSmoduleengineprefixrrabortfnptrrrrscan_unresolved_symbolss      z%RuntimeLinker.scan_unresolved_symbolscCs$|jD] }|js|j|jqdS)z5 Scan and track all defined symbols. N)rbr|rrrM)rSrrTrrrscan_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).0rMr&rr sz)RuntimeLinker.resolve..N)rrrrr)rSrpendingrMfnptrrrr&rresolves   zRuntimeLinker.resolveN) r*r+r,r-rrUrrrrrrrrs rcstfdd}|S)Ncs|jg|Ri|SrK)_ee)rSargsrRoldrrwrapper-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|_dSrK)rrk_defined_symbols)rSrrrrrU8s zJitEngine.__init__cCs ||jvS)z/Is the symbol defined in this session? )rrLrrrrEs zJitEngine.is_symbol_definedcCs.|j|jfD]}|jdd|DO_qdS)z(Extract symbols from the module cSsh|]}|js|jqSr)r|rM)rrrrr Ns z2JitEngine._load_defined_symbols..N)rbrr)rSrgsetsrrrrJszJitEngine._load_defined_symbolscCs|||j|S)zXOverride ExecutionEngine.add_module to keep info about defined symbols. )rr add_modulerSrrrrrQs  zJitEngine.add_modulecCs|j|j|j||S)z`Override ExecutionEngine.add_global_mapping to keep info about defined symbols. )rrrMrr)rSraddrrrrrXszJitEngine.add_global_mappingN)r*r+r,r-rUrrrrrr"ExecutionEngineset_object_cacherrget_global_value_addressrrrrr3s   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. cCrs)zD Create a new empty module suitable for the target. NrrLrrrrDurvzCodegen._create_empty_modulecCrs)zu Add a module to the execution engine. Ownership of the module is transferred to the engine. Nrrrrrr{rvzCodegen._add_modulecCr()zJ The LLVM "target data" object for this codegen instance. ) _target_datar&rrr target_datar>zCodegen.target_datacKs|j||fi|S)zb Create a :class:`CodeLibrary` object for use with this codegen instance. )_library_class)rSrMrRrrrrszCodegen.create_librarycCs|j||SrK)r r)rS serializedrrrunserialize_libraryszCodegen.unserialize_libraryN) r*r+r,r-rrDrrVr rrrrrrr js    r c@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)r0cCsBtd|_tt|||_d|j_t|_ | |jdS)Nglobal_codegen_module) initialize_llvmr`r"r[rrD _llvm_modulerMr _rtlinker_init)rS module_namerrrrUs 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)rn)use_lmmTFrr)listrr"Target from_tripleget_process_triplerirOPT_customize_tm_features _tm_features_customize_tm_optionscreate_target_machineUSE_LLVMLITE_MEMORY_MANAGERcreate_mcjit_compilerENABLE_PROFILINGenable_jit_eventsrrrr r rr` is_opt_maxrqrrrr rr)rS llvm_moduler tm_optionstmrrrrrrs,      zCPUCodegen._initcCs,tt|}t|_|jr|j|_|SrK) rrrr]r"rrr`rarErrrrDs  zCPUCodegen._create_empty_modulecKsp|dd}|jdi|}|}|dur+|dkr+tjdkr+|||tjr4| t ||fS)Nrorkrr) pop _pass_buildergetModulePassManagerrradd_loop_rotate_passadd_instruction_combine_passadd_jump_threading_passrsadd_refprune_passr+)rSrRroripmrrrrps  zCPUCodegen._module_pass_managercKs2|jdi|}|}tjr|t||fS)Nr)r*getFunctionPassManagerrrsr/r+)rSrRrir0rrrrcs  z!CPUCodegen._function_pass_managercKsH|dtj}|dtj}|dtj}t|jf|||d|}|S)Nrnrlrm)rnrlrm)r)rrLOOP_VECTORIZE SLP_VECTORIZErr)rSrR opt_levelrlrmrirrrr*s 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 )r"r[rlocale getlocalerAr)rSrrir_outlocrrrrs zCPUCodegen._check_llvm_bugscCs|jj||jfS)zP Return a tuple unambiguously describing the codegen behaviour. )rr_get_host_cpu_namerr&rrr magic_tuples zCPUCodegen.magic_tuplecCs.|j||j|j||j|jdSrK)rrrrrrrrrrs z(CPUCodegen._scan_and_fix_unresolved_refscCsntd}|jj|}|j}z||}Wnty+tj|||d}d|_ Ynw| | ||S)Nrr) rIntType as_pointerrrr get_globalKeyErrorGlobalVariabler}bitcastload)rSbuilderfntyrMvoidptrptrnamellvm_modrrrrinsert_unresolved_refs   z CPUCodegen.insert_unresolved_refcCstjdur tStjSrK)rCPU_NAMEr"get_host_cpu_namer&rrrr9*s zCPUCodegen._get_host_cpu_namecCstjdurtjStSrK)r CPU_FEATURESget_host_cpu_featuresr&rrr_get_host_cpu_features/s z!CPUCodegen._get_host_cpu_featuresN)r*r+r,rUrrDrprcr*rr:rrHr9rMrrrrr0s (  r0c@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_namer0rU)rSrcpu_namerrrrU=s zAOTCPUCodegen.__init__cCs<|j}|dkr |}||d<d|d<d|d<|j|d<dS)Nhostcpupicrelocr codemodelfeatures)rPr9r)rSoptionsrQrrrrBsz#AOTCPUCodegen._customize_tm_optionscCrsrOrr&rrrrKsz$AOTCPUCodegen._customize_tm_featurescCrrKrrrrrrPrzAOTCPUCodegen._add_modulerK) r*r+r,r-rr rUrrrrrrrrN5s  rNc@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) NrSrstaticppcrTrrU jitdefaultrVrWjitT) r9r"rfrom_default_triplerMrrr pysignaturer  parameters)rSrXr reloc_modelsigrrrr[s       z#JITCPUCodegen._customize_tm_optionscCs|SrK)rMr&rrrrvsz$JITCPUCodegen._customize_tm_featurescCs|j|dSrK)rrrrrrrzrzJITCPUCodegen._add_modulecCs2|j|}tjd|}tt||d<dS)zrSet the environment address. Update the GlobalVariable named *env_name* to the address of *env*. rrN)rr rr from_addressr?)rSenv_nameenvgvaddrenvptrrrrset_envs zJITCPUCodegen.set_envN) r*r+r,r-rr rrrrhrrrrrYTs rYcCsttdS)z Safe to use multiple times. N)r"initialize_native_targetinitialize_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. ravxF)r"rLrAr ENABLE_AVXrflatten)rWrprrrrLs   rL)6r&rr5rrrrllvmlite.bindingrWr" llvmlite.irrrabcrr numba.corerrrnumba.core.llvm_bindingsrnumba.core.runtime.nrtoptrnumba.core.runtimer numba.core.compiler_lockr numba.core.errorsr numba.misc.inspectionr numba.misc.llvm_pass_timingsr frozensetrrr+rIrrJr.rXrrrrrr r0rNrYrrLrrrrsT          6}>77*"8