B   ›\!sã @s„ddlmZmZmZddlZddlZddlZddlZddlZddl m m Z ddl m mZddlmZddlmZddlmZmZmZddlmZddlmZddlmZe ddd d d d d dgƒZ!dd„Z"dd„Z#Gdd„de$ƒZ%Gdd„de$ƒZ&Gdd„de&ƒZ'Gdd„de&ƒZ(Gdd„de$ƒZ)dd„Z*Gdd „d e$ƒZ+Gd!d"„d"e$ƒZ,Gd#d$„d$e,ƒZ-Gd%d&„d&e,ƒZ.d'd(„Z/d)d*„Z0dS)+é)Úprint_functionÚdivisionÚabsolute_importN)ÚconfigÚutilsÚcgutils)Úremove_redundant_nrt_refct)Úrtsys)Úrequire_global_compiler_lockÚx86Zi386Zi486Zi586Zi686Zi786Zi886Zi986cCs| d¡d}|tkS)Nú-r)ÚsplitÚ_x86arch)ÚtripleÚarch©rú4lib/python3.7/site-packages/numba/targets/codegen.pyÚ_is_x86srcCs$t| dd¡ƒt|ƒtdƒdS)NéPr zP================================================================================)ÚprintÚcenter)ÚheaderZbodyrrrÚdumpsrc@s*eZdZdZdd„Zd dd„Zdd „ZdS) Ú_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. cCs ||_dS)N)Údot)ÚselfrrrrÚ__init__*sz _CFG.__init__NFcCstj|j||dS)aC 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). )ÚfilenameÚview)ÚllZview_dot_graphr)rrrrrrÚdisplay-s z _CFG.displaycCs|jS)N)r)rrrrÚ__repr__8sz _CFG.__repr__)NF)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr r!rrrrr"s rc@s4eZdZdZdZdZdZdd„Zedd„ƒZ edd„ƒZ d d „Z d d „Z d d„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Z d5d6„Z!e"d7d8„ƒZ#e"d9d:„ƒZ$e"d;d<„ƒZ%d=d>„Z&d?d@„Z'e"dAdB„ƒZ(dCS)DÚ 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. FcCsN||_||_tƒ|_t t|j |j¡ƒ¡|_t   |j¡|j_ d|_ g|_ dS)N)Ú_codegenÚ_nameÚsetÚ_linking_librariesrÚparse_assemblyÚstrÚ_create_empty_moduleÚ _final_modulerÚnormalize_ir_textÚnameÚ_shared_moduleÚ_dynamic_globals)rÚcodegenr0rrrrGszCodeLibrary.__init__cCs| ¡t|jƒdkS)Nr)Ú_ensure_finalizedÚlenr2)rrrrÚhas_dynamic_globalsRszCodeLibrary.has_dynamic_globalscCs|jS)z9 The codegen object owning this library. )r')rrrrr3WszCodeLibrary.codegencCsd|jt|ƒfS)Nz)r(Úid)rrrrr!^szCodeLibrary.__repr__cCs|jrtd|fƒ‚dS)Nz+operation impossible on finalized object %r)Ú _finalizedÚ RuntimeError)rrrrÚ_raise_if_finalizedaszCodeLibrary._raise_if_finalizedcCs|js| ¡dS)N)r8Úfinalize)rrrrr4fszCodeLibrary._ensure_finalizedc CsP|jj|_|j |¡0}x(|jD]}| ¡| |¡| ¡q WWdQRXdS)zP Internal: run function-level optimizations inside *ll_module*. N)r'Ú _data_layoutÚ data_layoutÚ_function_pass_managerÚ functionsÚ initializeÚrunr;)rÚ ll_moduleZfpmÚfuncrrrÚ_optimize_functionsjs    zCodeLibrary._optimize_functionscCs |jj |j¡t|jƒ|_dS)zA Internal: optimize this library's final module. N)r'Ú_mpmrAr.r)rrrrÚ_optimize_final_modulexsz"CodeLibrary._optimize_final_modulecCs¤| ¡|jdk r|jS|j}g}d}x6|jD],}|d7}|js.|jtjjkr.|  |j ¡q.W|dkrtt d|fƒ‚|rš|  ¡}x|D]}d|  |¡_q†W||_|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 Nréz7library unfit for linking: no available functions in %sZ linkonce_odr)r4r1r.r?Úis_declarationÚlinkagerZLinkageÚexternalÚappendr0r9ZcloneÚ get_function)rÚmodZto_fixZnfuncsÚfnr0rrrÚ_get_module_for_linkings&     z#CodeLibrary._get_module_for_linkingcCs| ¡|j |¡}|S)zB Create a LLVM IR module for use by this library. )r:r'r-)rr0Ú ir_modulerrrÚcreate_ir_module¢s zCodeLibrary.create_ir_modulecCs| ¡|j |¡dS)zk Add a library for linking into this library, without losing the original library. N)r4r*Úadd)rÚlibraryrrrÚadd_linking_libraryªszCodeLibrary.add_linking_librarycCsN| ¡t|tjƒst‚t t|ƒ¡}t  |¡}|j |_ |  ¡|  |¡dS)zB Add a LLVM IR module's contents to this library. N) r:Ú isinstanceÚllvmirÚModuleÚAssertionErrorrr/r,rr+r0ÚverifyÚadd_llvm_module)rrPÚirrBrrrÚ add_ir_module²s zCodeLibrary.add_ir_modulecCs"| |¡t|ƒ}|j |¡dS)N)rDrr.Úlink_in)rrBrrrrZ¾s zCodeLibrary.add_llvm_modulecCsttƒ|j ¡| ¡tjr2td|j| ¡ƒx"|j D]}|j j |  ¡ddq:W|  ¡|j  ¡| ¡dS)z® Finalize the library. After this call, nothing can be added anymore. Finalization involves various stages of code optimization and linking. zFUNCTION OPTIMIZED DUMP %sT)ZpreserveN)r r'Ú_check_llvm_bugsr:rZ DUMP_FUNC_OPTrr(Ú get_llvm_strr*r.r]rOrFrYÚ_finalize_final_module)rrSrrrr;Äs   zCodeLibrary.finalizecCs0x*|jjD]}|j d¡r |j |j¡q WdS)Nznumba.dynamic.globals)r.Úglobal_variablesr0Ú startswithr2rK)rÚgvrrrÚ_finalize_dyanmic_globalsàs z%CodeLibrary._finalize_dyanmic_globalscCs<x6|jjD]*}|jr |j d¡r d}t| |j¡ƒ‚q WdS)NZ _ZN5numbazSymbol {} not linked properly)r.r?rHr0rbrXÚformat)rrNÚmsgrrrÚ_verify_declare_only_symbolsæsz(CodeLibrary._verify_declare_only_symbolscCsŽ| ¡| ¡t |¡|j_|j |j¡}|rs  z CodeLibrary._get_compiled_objectcCs8|jstd|fƒ‚|jr(td|fƒ‚||_d|_dS)Nz object caching not enabled in %szlibrary already compiled: %sT)ruryrwrvro)rÚvaluerrrÚ_set_compiled_objectEs z CodeLibrary._set_compiled_objectc CsÒddlm}ddlm}ddlm}|||ƒƒ}tdƒxŽ| ¡D]‚}|ddkrBt|  ¡dd „d }td ƒxT|D]L}|j s€qttd |j   ¡|d |d|  |dd¡|  |dd¡fƒqtWqBWtƒdS)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_SYMTABcSs|jS)N)r0)ÚsymrrrÚZsz'CodeLibrary._dump_elf..)Úkeyz symbols:z/ - %r: size=%d, value=0x%x, type=%s, bind=%sÚst_sizeZst_valueZst_infoÚtypeZbindN)Zelftools.elf.elffiler}Z elftools.elfr~ÚiorrZ iter_sectionsÚsortedZ iter_symbolsr0ÚdecodeZdescribe_symbol_typeZdescribe_symbol_bind) ÚclsÚbufr}r~rÚfZsecZsymbolsr€rrrÚ _dump_elfMs&       zCodeLibrary._dump_elfcCs6y |j}Wntk rdSX|jr2d|_||_dS)zB `ll_module` was compiled into object code `buf`. NT)rjÚAttributeErrorrurwrv)rˆrBr‰rrrrÚ_object_compiled_hookhs z!CodeLibrary._object_compiled_hookcCs@y |j}Wntk rdSX|jr<|jr<|j}d|_|SdS)z> Return a cached object code for `ll_module`. N)rjrŒrurv)rˆrBrr‰rrrÚ_object_getbuffer_hookus  z"CodeLibrary._object_getbuffer_hookcCs| ¡|jd|j ¡fS)zX Serialize this library using its bitcode as the cached representation. Úbitcode)r4r(r.Ú as_bitcode)rrrrÚserialize_using_bitcodeƒsz#CodeLibrary.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)r4rzrOrr()rÚdatarrrÚserialize_using_object_codeŠsz'CodeLibrary.serialize_using_object_codec Csž|\}}}| |¡}t||ƒs"t‚|dkrBt |¡|_| ¡|S|dkrŒ|\}}| ¡| |¡t |¡|_ | ¡|j j   |j ¡|St d|fƒ‚dS)Nrr’z!unsupported serialization kind %r)Úcreate_libraryrUrXrZ parse_bitcoder.r`rxr|r1r'Ú_engineÚ_load_defined_symbolsry) rˆr3Ústater0Zkindr“rZ object_codeZshared_bitcoderrrÚ _unserialize•s      zCodeLibrary._unserializeN))r"r#r$r%r8rurorÚpropertyr6r3r!r:r4rDrFrOrQrTr\rZr;rdrgr`rnrLrrr_rmrtrxrzr|Ú classmethodr‹rrŽr‘r”r™rrrrr&<sH  #      r&c@s$eZdZdd„Zdd„Zdd„ZdS)ÚAOTCodeLibrarycCs| ¡|jj |j¡S)z¤ Return this library as a native object (a bytestring) -- for example ELF under Linux. This function implicitly calls .finalize(). )r4r'rsZ emit_objectr.)rrrrÚemit_native_object®sz!AOTCodeLibrary.emit_native_objectcCs| ¡|j ¡S)zz Return this library as LLVM bitcode (a bytestring). This function implicitly calls .finalize(). )r4r.r)rrrrÚ emit_bitcode¸szAOTCodeLibrary.emit_bitcodecCsdS)Nr)rrrrrlÁsz!AOTCodeLibrary._finalize_specificN)r"r#r$rržrlrrrrrœ¬s  rœc@seZdZdd„Zdd„ZdS)ÚJITCodeLibrarycCs0| ¡|jj}| |¡sdS|jj |¡SdS)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. rN)r4r'r–Úis_symbol_definedÚget_function_address)rr0ÚeerrrÚget_pointer_to_functionÇs  z&JITCodeLibrary.get_pointer_to_functioncCs|j |j¡|jj ¡dS)N)r'Ú_scan_and_fix_unresolved_refsr.r–Úfinalize_object)rrrrrlÜsz!JITCodeLibrary._finalize_specificN)r"r#r$r£rlrrrrrŸÅsrŸc@s4eZdZdZdZdd„Zdd„Zdd„Zd d „Zd S) Ú RuntimeLinkerzP For tracking unresolved symbols generated at runtime due to recursion. z.numba.unresolved$cCst ¡|_tƒ|_g|_dS)N)rZ UniqueDictÚ _unresolvedr)Ú_definedÚ _resolved)rrrrrçs zRuntimeLinker.__init__cCsx|j}xl|jD]b}|j |¡r|jt|ƒd…}| |j¡r>qtj d¡}t   |¡}|  |t   |¡¡||j |<qWdS)zr Scan and track all unresolved external symbols in the module and allocate memory for it. NZnrt_unresolved_abort)ÚPREFIXrar0rbr5r r rSr£ÚctypesÚc_void_pÚadd_global_mappingZ addressofr§)rÚmoduleÚengineÚprefixrcr€ZabortfnÚptrrrrÚscan_unresolved_symbolsìs     z%RuntimeLinker.scan_unresolved_symbolscCs(x"|jD]}|js|j |j¡qWdS)z5 Scan and track all defined symbols. N)r?rHr¨rRr0)rr®rNrrrÚscan_defined_symbolsÿs z"RuntimeLinker.scan_defined_symbolscsX‡fdd„ˆjDƒ}x>|D]6}| |¡}ˆj|}||_ˆj ||f¡ˆj|=qWdS)z= Fix unresolved symbols if they are defined. csg|]}|ˆjkr|‘qSr)r¨)Ú.0r0)rrrú sz)RuntimeLinker.resolve..N)r§r¡r{r©rK)rr¯Úpendingr0Úfnptrr±r)rrÚresolves   zRuntimeLinker.resolveN) r"r#r$r%rªrr²r³r¸rrrrr¦ás r¦cst ˆ¡‡fdd„ƒ}|S)Ncsˆ|jf|ž|ŽS)N)Ú_ee)rÚargsÚkwargs)ÚoldrrÚwrappersz_proxy..wrapper)Ú functoolsÚwraps)r¼r½r)r¼rÚ_proxysrÀc@sheZdZdZdd„Zdd„Zdd„Zdd „Zd d „Ze e j j ƒZ e e j j ƒZ e e j jƒZe e j jƒZd S) Ú JitEnginez»Wraps 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ƒ|_dS)N)r¹r)Ú_defined_symbols)rr¢rrrr%s zJitEngine.__init__cCs ||jkS)z/Is the symbol defined in this session? )rÂ)rr0rrrr 2szJitEngine.is_symbol_definedcCs2x,|j|jfD]}|jdd„|DƒO_qWdS)z(Extract symbols from the module cSsh|]}|js|j’qSr)rHr0)r´rcrrrú ;sz2JitEngine._load_defined_symbols..N)r?rarÂ)rrMZgsetsrrrr—7szJitEngine._load_defined_symbolscCs| |¡|j |¡S)zXOverride ExecutionEngine.add_module to keep info about defined symbols. )r—r¹Ú add_module)rr®rrrrÄ>s zJitEngine.add_modulecCs|j |j¡|j ||¡S)z`Override ExecutionEngine.add_global_mapping to keep info about defined symbols. )rÂrRr0r¹r­)rrcZaddrrrrr­EszJitEngine.add_global_mappingN)r"r#r$r%rr r—rÄr­rÀrZExecutionEngineÚset_object_cacher¥r¡Úget_global_value_addressrrrrrÁ s    rÁc@sˆeZdZdd„Zdd„Zdd„Zedd„ƒZd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„ZdS) ÚBaseCPUCodegencCsBtƒd|_t t| |¡ƒ¡|_d|j_tƒ|_ |  |j¡dS)NZglobal_codegen_module) Úinitialize_llvmr<rr+r,r-Ú _llvm_moduler0r¦Ú _rtlinkerÚ_init)rÚ module_namerrrrXszBaseCPUCodegen.__init__cCs´t|jƒgkstdƒ‚tj t ¡¡}ttj d}|  ¡|_ |  |¡|j f|Ž}t ||¡}tjrl| ¡||_t|ƒ|_|j|_t|jƒ|_| ¡|_|j |jj|jj¡dS)NzModule isn't empty)Úopt)ÚlistrarXrÚTargetZ from_tripleÚget_process_tripleÚdictrÚOPTÚ_customize_tm_featuresÚ _tm_featuresÚ_customize_tm_optionsZcreate_target_machineZcreate_mcjit_compilerZENABLE_PROFILINGZenable_jit_eventsrsrÁr–Ú target_dataÚ _target_datar,r<Ú_module_pass_managerrErÅÚ_library_classrrŽ)rÚ llvm_moduleÚtargetZ tm_optionsZtmr¯rrrrËbs          zBaseCPUCodegen._initcCs,t t |¡¡}t ¡|_|jr(|j|_|S)N) ÚlcrWrr/rrÐrr<r=)rr0rPrrrr-xs  z#BaseCPUCodegen._create_empty_modulecCs|jS)zJ The LLVM "target data" object for this codegen instance. )r×)rrrrrÖszBaseCPUCodegen.target_datacCs | ||¡S)zb Create a :class:`CodeLibrary` object for use with this codegen instance. )rÙ)rr0rrrr•†szBaseCPUCodegen.create_librarycCs|j ||¡S)N)rÙr™)rZ serializedrrrÚunserialize_librarysz"BaseCPUCodegen.unserialize_libraryc Cs6t ¡}|j |¡| ¡}| |¡WdQRX|S)N)rZcreate_module_pass_managerrsÚadd_analysis_passesÚ_pass_manager_builderÚpopulate)rÚpmÚpmbrrrrØs   z#BaseCPUCodegen._module_pass_managerc Cs8t |¡}|j |¡| ¡}| |¡WdQRX|S)N)rZcreate_function_pass_managerrsrÞrßrà)rrÚrárârrrr>—s    z%BaseCPUCodegen._function_pass_managercCstjtjtjd}|S)a Create a PassManagerBuilder. Note: a PassManagerBuilder seems good only for one use, so you should call this method each time you want to populate a module or function pass manager. Otherwise some optimizations will be missed... )rÍZloop_vectorize)ÚlpZcreate_pass_manager_builderrrÒZLOOP_VECTORIZE)rrârrrrßžs z$BaseCPUCodegen._pass_manager_buildercCsZd}t |¡}t|ƒ}d|ks&d|kr*dSd|krHt ¡}td|fƒ‚td|fƒ‚dS)z< Guard against some well-known LLVM bug(s). zo define double @func() { ret double 1.23e+01 } z12.3z1.23Nz1.0z¯LLVM will produce incorrect floating-point code in the current locale %s. Please read http://numba.pydata.org/numba-doc/dev/user/faq.html#llvm-locale-bug for more information.zUnexpected IR: %s )rr+r,ÚlocaleZ getlocaler9rX)rr[rMZir_outZlocrrrr^«s  zBaseCPUCodegen._check_llvm_bugscCs|jj| ¡|jfS)zP Return a tuple unambiguously describing the codegen behaviour. )rÉrÚ_get_host_cpu_namerÔ)rrrrÚ magic_tupleÉs zBaseCPUCodegen.magic_tuplecCs.|j ||j¡|j |¡|j |j¡dS)N)rÊr²r–r³r¸)rr®rrrr¤Ðs z,BaseCPUCodegen._scan_and_fix_unresolved_refscCspt d¡ ¡}|jj|}|j}y| |¡}Wn*tk rXtj|||d}d|_ YnX|  |  |¡| ¡¡S)Né)r0rJ) rVZIntTypeZ as_pointerrÊrªr®Z get_globalÚKeyErrorZGlobalVariablerIZbitcastÚload)rZbuilderZfntyr0ZvoidptrZptrnameZllvm_modr·rrrÚinsert_unresolved_refÕs  z$BaseCPUCodegen.insert_unresolved_refcCstjdkrt ¡StjS)N)rZCPU_NAMErZget_host_cpu_name)rrrrråász!BaseCPUCodegen._get_host_cpu_namecCstjdk rtjStƒS)N)rZ CPU_FEATURESÚget_host_cpu_features)rrrrÚ_get_host_cpu_featuresæs z%BaseCPUCodegen._get_host_cpu_featuresN)r"r#r$rrËr-ršrÖr•rÝrØr>rßr^rær¤rêrårìrrrrrÇVs    rÇc@s6eZdZdZeZd dd„Zdd„Zdd„Zd d „Z dS) Ú AOTCPUCodegenzp A codegen implementation suitable for Ahead-Of-Time compilation (e.g. generation of object files). NcCs|pd|_t ||¡dS)NÚ)Ú _cpu_namerÇr)rrÌÚcpu_namerrrrós zAOTCPUCodegen.__init__cCs<|j}|dkr| ¡}||d<d|d<d|d<|j|d<dS)NZhostÚcpuZpicÚrelocÚdefaultÚ codemodelÚfeatures)rïrårÔ)rÚoptionsrðrrrrÕøsz#AOTCPUCodegen._customize_tm_optionscCsdS)Nrîr)rrrrrÓsz$AOTCPUCodegen._customize_tm_featurescCsdS)Nr)rr®rrrrkszAOTCPUCodegen._add_module)N) r"r#r$r%rœrÙrrÕrÓrkrrrrríës   ríc@s4eZdZdZeZdd„Zdd„Zdd„Zdd „Z d S) Ú JITCPUCodegenzI A codegen implementation suitable for Just-In-Time compilation. cCsR| ¡|d<tj ¡j}| d¡r(d}nd}||d<d|d<|j|d<d |d <dS) Nrñr ZstaticróròZ jitdefaultrôrõTZjitdebug)rårrÏZfrom_default_tripler0rbrÔ)rrörZ reloc_modelrrrrÕs    z#JITCPUCodegen._customize_tm_optionscCs| ¡S)N)rì)rrrrrÓ'sz$JITCPUCodegen._customize_tm_featurescCs|j |¡dS)N)r–rÄ)rr®rrrrk+szJITCPUCodegen._add_modulecCs2|j |¡}tjd |¡}t t|ƒ¡|d<dS)zrSet the environment address. Update the GlobalVariable named *env_name* to the address of *env*. rGrN)r–rÆr«r¬Z from_addressr7)rZenv_nameÚenvZgvaddrZenvptrrrrÚset_env3s zJITCPUCodegen.set_envN) r"r#r$r%rŸrÙrÕrÓrkrùrrrrr÷ s r÷cCst ¡t ¡t ¡dS)z Safe to use multiple times. N)rr@Zinitialize_native_targetZinitialize_native_asmprinterrrrrrÈ=srÈcCsTy t ¡}Wntk r dSXtjsHx|D]}| d¡r.d||<q.W| ¡SdS)z~Get host CPU features using LLVM. The features may be modified due to user setting. See numba.config.ENABLE_AVX. rîZavxFN)rrër9rZ ENABLE_AVXrbZflatten)rõÚkrrrrëEs    rë)1Z __future__rrrrpr¾rärhr«Zllvmlite.llvmpy.coreZllvmpyZcorerÜZllvmlite.llvmpy.passesZpassesrãZllvmlite.bindingZbindingrZ llvmlite.irr[rVZnumbarrrZnumba.runtime.nrtoptrZ numba.runtimer Znumba.compiler_lockr Ú frozensetrrrr’rr&rœrŸr¦rÀrÁrÇrír÷rÈrërrrrÚs@      r863