\ K]c @`sddlmZmZmZddlZddlZddlZddlZddlZddl j j Z ddl j jZddljZddljZddlmZmZmZddlmZddlmZddlmZe ddd d d d d dgZ!dZ"dZ#de$fdYZ%de$fdYZ&de&fdYZ'de&fdYZ(de$fdYZ)dZ*de$fdYZ+de$fdYZ,d e,fd!YZ-d"e,fd#YZ.d$Z/d%Z0dS(&i(tprint_functiontdivisiontabsolute_importN(tconfigtutilstcgutils(tremove_redundant_nrt_refct(trtsys(trequire_global_compiler_locktx86ti386ti486ti586ti686ti786ti886ti986cC`s|jdd}|tkS(Nt-i(tsplitt_x86arch(ttripletarch((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_is_x86scC`s2t|jddt|tdddS(NiPRt=(tprinttcenter(theadertbody((s4lib/python2.7/site-packages/numba/targets/codegen.pytdumps t_CFGcB`s/eZdZdZdedZdZRS(s 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. cC`s ||_dS(N(tdot(tselfR((s4lib/python2.7/site-packages/numba/targets/codegen.pyt__init__*scC`stj|jd|d|S(sC 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). tfilenametview(tlltview_dot_graphR(RR!R"((s4lib/python2.7/site-packages/numba/targets/codegen.pytdisplay-s cC`s|jS(N(R(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt__repr__8sN(t__name__t __module__t__doc__R tNonetFalseR%R&(((s4lib/python2.7/site-packages/numba/targets/codegen.pyR"s  t CodeLibrarycB`sdeZdZeZeZeZdZedZ edZ dZ dZ dZ dZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!e"dZ#e"dZ$e"dZ%dZ&dZ'e"d Z(RS(!s 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. cC`sp||_||_g|_tjt|jj|j|_tj |j|j_ d|_ g|_ dS(N(t_codegent_namet_linking_librariesR#tparse_assemblytstrt_create_empty_modulet _final_moduleRtnormalize_ir_texttnameR*t_shared_modulet_dynamic_globals(RtcodegenR5((s4lib/python2.7/site-packages/numba/targets/codegen.pyR Gs   ! cC`s|jt|jdkS(Ni(t_ensure_finalizedtlenR7(R((s4lib/python2.7/site-packages/numba/targets/codegen.pythas_dynamic_globalsRs cC`s|jS(s9 The codegen object owning this library. (R-(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR8WscC`sd|jt|fS(Ns(R.tid(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR&^scC`s#|jrtd|fndS(Ns+operation impossible on finalized object %r(t _finalizedt RuntimeError(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_raise_if_finalizedas cC`s|js|jndS(N(R=tfinalize(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR9fs cC`sc|jj|_|jj|<}x2|jD]'}|j|j||jq.WWdQXdS(sP Internal: run function-level optimizations inside *ll_module*. N(R-t _data_layoutt data_layoutt_function_pass_managert functionst initializetrunR@(Rt ll_moduletfpmtfunc((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_optimize_functionsjs   cC`s,|jjj|jt|j|_dS(sA Internal: optimize this library's final module. N(R-t_mpmRFR3R(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_optimize_final_modulexscC`s|j|jdk r |jS|j}g}d}xM|jD]B}|d7}|j r?|jtjj kr?|j |j q?q?W|dkrt d|fn|r|j }x#|D]}d|j|_qWn||_|S(s: 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 iis7library unfit for linking: no available functions in %st linkonce_odrN(R9R6R*R3RDtis_declarationtlinkageR#tLinkagetexternaltappendR5R>tclonet get_function(Rtmodtto_fixtnfuncstfnR5((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_get_module_for_linkings&       cC`s |j|jj|}|S(sB Create a LLVM IR module for use by this library. (R?R-R2(RR5t ir_module((s4lib/python2.7/site-packages/numba/targets/codegen.pytcreate_ir_modules cC`s|j|jj|dS(sk Add a library for linking into this library, without losing the original library. N(R9R/RR(Rtlibrary((s4lib/python2.7/site-packages/numba/targets/codegen.pytadd_linking_librarys cC`sm|jt|tjs"ttjt|}tj |}|j |_ |j |j |dS(sB Add a LLVM IR module's contents to this library. N( R?t isinstancetllvmirtModuletAssertionErrorRR4R1R#R0R5tverifytadd_llvm_module(RRZtirRG((s4lib/python2.7/site-packages/numba/targets/codegen.pyt add_ir_modules   cC`s-|j|t|}|jj|dS(N(RJRR3tlink_in(RRG((s4lib/python2.7/site-packages/numba/targets/codegen.pyRcs  cC`st|jj|jtjrDtd|j|jnt }xI|j D]>}||krW|j ||j j |jdtqWqWW|j|j j|jdS(s Finalize the library. After this call, nothing can be added anymore. Finalization involves various stages of code optimization and linking. sFUNCTION OPTIMIZED DUMP %stpreserveN(RR-t_check_llvm_bugsR?Rt DUMP_FUNC_OPTRR.t get_llvm_strtsetR/taddR3RfRYtTrueRLRbt_finalize_final_module(RtseenR\((s4lib/python2.7/site-packages/numba/targets/codegen.pyR@s         cC`sCx<|jjD].}|jjdr |jj|jq q WdS(Nsnumba.dynamic.globals(R3tglobal_variablesR5t startswithR7RR(Rtgv((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_finalize_dyanmic_globalsscC`sWxP|jjD]B}|jr |jjdr d}t|j|jq q WdS(Nt _ZN5numbasSymbol {} not linked properly(R3RDRNR5RqRatformat(RRXtmsg((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_verify_declare_only_symbolsscC`s|j|jtj||j_|jj|j}|rWtj ||n|j t |_ t jrtd|j|jnt jr|j}|rtd|j|jqndS(s? Make the underlying LLVM module ready to use. sOPTIMIZED DUMP %ss ASSEMBLY %sN(RsRwtweakreftproxyR3t_CodeLibrary__libraryR-t _add_moduleRR@t_finalize_specificRmR=RtDUMP_OPTIMIZEDRR.Rjt DUMP_ASSEMBLYt get_asm_str(Rtcleanuptasm((s4lib/python2.7/site-packages/numba/targets/codegen.pyRns       cc`s2|j}x"|jD]}|js|VqqWdS(sj Get all functions defined in the library. The library must have been finalized. N(R3RDRN(RRURX((s4lib/python2.7/site-packages/numba/targets/codegen.pytget_defined_functionss  cC`s|jj|S(N(R3RT(RR5((s4lib/python2.7/site-packages/numba/targets/codegen.pyRTscC`s|jrtjdndS(Ns@Inspection disabled for cached code. Invalid result is returned.(t_disable_inspectiontwarningstwarn(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt _sentry_cache_disable_inspections cC`s|jt|jS(sA Get the human-readable form of the LLVM module. (RR1R3(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyRj"s cC`s&|jt|jjj|jS(s2 Get the human-readable assembly. (RR1R-t_tmt emit_assemblyR3(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR)s cC`s2|j|j|}tj|}t|S(s= Get control-flow graph of the LLVM function (RRTR#tget_function_cfgR(RR5RXR((s4lib/python2.7/site-packages/numba/targets/codegen.pyR0s cC`st|_d|_t|_dS(N(Rmt_object_caching_enabledR*t_compiled_objectR+t _compiled(R((s4lib/python2.7/site-packages/numba/targets/codegen.pytenable_object_caching=s  cC`sK|jstd|fn|jdkrDtd|fn|jS(Ns object caching not enabled in %ssno compiled object yet for %s(Rt ValueErrorRR*R>(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_get_compiled_objectBs  cC`sT|jstd|fn|jr>td|fn||_t|_dS(Ns object caching not enabled in %sslibrary already compiled: %s(RRRRRmR(Rtvalue((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_set_compiled_objectIs    c C`sddlm}ddlm}ddlm}|||}tdx|jD]}|ddkrYt|j dd }td xp|D]e}|j sqntd |j j |d |d |j |dd|j |ddfqWqYqYWtdS(sw Dump the symbol table of an ELF file. Needs pyelftools (https://github.com/eliben/pyelftools) i(tELFFile(t descriptions(tBytesIOs ELF file:tsh_typet SHT_SYMTABtkeycS`s|jS(N(R5(tsym((s4lib/python2.7/site-packages/numba/targets/codegen.pyt^ts symbols:s/ - %r: size=%d, value=0x%x, type=%s, bind=%stst_sizetst_valuetst_infottypetbindN(telftools.elf.elffileRt elftools.elfRtioRRt iter_sectionstsortedt iter_symbolsR5tdecodetdescribe_symbol_typetdescribe_symbol_bind( tclstbufRRRtftsectsymbolsR((s4lib/python2.7/site-packages/numba/targets/codegen.pyt _dump_elfQs&     'cC`sDy |j}Wntk r!dSX|jr@t|_||_ndS(sB `ll_module` was compiled into object code `buf`. N(RztAttributeErrorRRmRR(RRGRR((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_object_compiled_hookls    cC`sNy |j}Wntk r!dSX|jrJ|jrJ|j}d|_|SdS(s> Return a cached object code for `ll_module`. N(RzRRRR*(RRGRR((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_object_getbuffer_hookys    cC`s#|j|jd|jjfS(sX Serialize this library using its bitcode as the cached representation. tbitcode(R9R.R3t as_bitcode(R((s4lib/python2.7/site-packages/numba/targets/codegen.pytserialize_using_bitcodes cC`s8|j|j|jjf}|jd|fS(s Serialize this library using its object code as the cached representation. We also include its bitcode for further inlining with other libraries. tobject(R9RRYRR.(Rtdata((s4lib/python2.7/site-packages/numba/targets/codegen.pytserialize_using_object_codes  c C`s|\}}}|j|}t||s3t|dkr_tj||_|j|S|dkr|\}}|j|j|tj||_ |j|j j j |j |St d|fdS(NRRs!unsupported serialization kind %r(tcreate_libraryR^RaR#t parse_bitcodeR3RnRRR6R-t_enginet_load_defined_symbolsR( RR8tstateR5tkindRRt object_codetshared_bitcode((s4lib/python2.7/site-packages/numba/targets/codegen.pyt _unserializes        ()R'R(R)R+R=RRR tpropertyR;R8R&R?R9RJRLRYR[R]ReRcR@RsRwRnRRTRRjRRRRRt classmethodRRRRRR(((s4lib/python2.7/site-packages/numba/targets/codegen.pyR,<sH       #                tAOTCodeLibrarycB`s#eZdZdZdZRS(cC`s |j|jjj|jS(s Return this library as a native object (a bytestring) -- for example ELF under Linux. This function implicitly calls .finalize(). (R9R-Rt emit_objectR3(R((s4lib/python2.7/site-packages/numba/targets/codegen.pytemit_native_objects cC`s|j|jjS(sz Return this library as LLVM bitcode (a bytestring). This function implicitly calls .finalize(). (R9R3R(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt emit_bitcodes cC`sdS(N((R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR|s(R'R(RRR|(((s4lib/python2.7/site-packages/numba/targets/codegen.pyRs tJITCodeLibrarycB`seZdZdZRS(cC`s@|j|jj}|j|s)dS|jjj|SdS(s 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. iN(R9R-Rtis_symbol_definedtget_function_address(RR5tee((s4lib/python2.7/site-packages/numba/targets/codegen.pytget_pointer_to_functions   cC`s'|jj|j|jjjdS(N(R-t_scan_and_fix_unresolved_refsR3Rtfinalize_object(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR|s(R'R(RR|(((s4lib/python2.7/site-packages/numba/targets/codegen.pyRs t RuntimeLinkercB`s8eZdZdZdZdZdZdZRS(sP For tracking unresolved symbols generated at runtime due to recursion. s.numba.unresolved$cC`s(tj|_t|_g|_dS(N(Rt UniqueDictt _unresolvedRkt_definedt _resolved(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR s cC`s|j}x|jD]}|jj|r|jt|}|j|jrVqntjjd}t j |}|j |t j |||j |?s N(RDRpR(RRUtgsets((s4lib/python2.7/site-packages/numba/targets/codegen.pyR;scC`s|j||jj|S(sXOverride ExecutionEngine.add_module to keep info about defined symbols. (RRt add_module(RR((s4lib/python2.7/site-packages/numba/targets/codegen.pyRBs cC`s&|jj|j|jj||S(s`Override ExecutionEngine.add_global_mapping to keep info about defined symbols. (RRlR5RR(RRrtaddr((s4lib/python2.7/site-packages/numba/targets/codegen.pyRIs(R'R(R)R RRRRRR#tExecutionEnginetset_object_cacheRRtget_global_value_address(((s4lib/python2.7/site-packages/numba/targets/codegen.pyR$s    tBaseCPUCodegencB`seZdZdZdZedZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZRS(cC`s]td|_tjt|j||_d|j_t |_ |j |jdS(Ntglobal_codegen_module( tinitialize_llvmR*RAR#R0R1R2t _llvm_moduleR5Rt _rtlinkert_init(Rt module_name((s4lib/python2.7/site-packages/numba/targets/codegen.pyR \s   cC`st|jgks!tdtjjtj}tdtj }|j |_ |j ||j |}tj||}tjr|jn||_t||_|j|_t|j|_|j|_|jj|jj|jjdS(NsModule isn't emptytopt(tlistRpRaR#tTargett from_tripletget_process_tripletdictRtOPTt_customize_tm_featurest _tm_featurest_customize_tm_optionstcreate_target_machinetcreate_mcjit_compilertENABLE_PROFILINGtenable_jit_eventsRRRt target_datat _target_dataR1RAt_module_pass_managerRKRt_library_classRR(Rt llvm_modulettargett tm_optionsttmR((s4lib/python2.7/site-packages/numba/targets/codegen.pyRfs !     cC`sCtjtj|}tj|_|jr?|j|_n|S(N( tlcR`RR4R#RRRARB(RR5RZ((s4lib/python2.7/site-packages/numba/targets/codegen.pyR2|s  cC`s|jS(sJ The LLVM "target data" object for this codegen instance. (R (R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR scC`s|j||S(sb Create a :class:`CodeLibrary` object for use with this codegen instance. (R (RR5((s4lib/python2.7/site-packages/numba/targets/codegen.pyRscC`s|jj||S(N(R R(Rt serialized((s4lib/python2.7/site-packages/numba/targets/codegen.pytunserialize_libraryscC`sBtj}|jj||j}|j|WdQX|S(N(R#tcreate_module_pass_managerRtadd_analysis_passest_pass_manager_buildertpopulate(Rtpmtpmb((s4lib/python2.7/site-packages/numba/targets/codegen.pyR s  cC`sEtj|}|jj||j}|j|WdQX|S(N(R#tcreate_function_pass_managerRRRR(RR RR((s4lib/python2.7/site-packages/numba/targets/codegen.pyRCs cC`s"tjdtjdtj}|S(s 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... Rtloop_vectorize(tlptcreate_pass_manager_builderRRtLOOP_VECTORIZE(RR((s4lib/python2.7/site-packages/numba/targets/codegen.pyRs cC`sd}tj|}t|}d|ks9d|kr=dSd|krktj}td|fntd|fdS(s< Guard against some well-known LLVM bug(s). so define double @func() { ret double 1.23e+01 } s12.3s1.23Ns1.0sLLVM will produce incorrect floating-point code in the current locale %s. Please read http://numba.pydata.org/numba-doc/latest/user/faq.html#llvm-locale-bug for more information.sUnexpected IR: %s (R#R0R1tlocalet getlocaleR>Ra(RRdRUtir_outtloc((s4lib/python2.7/site-packages/numba/targets/codegen.pyRhs   cC`s|jj|j|jfS(sP Return a tuple unambiguously describing the codegen behaviour. (RRt_get_host_cpu_nameR(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt magic_tuplescC`s=|jj||j|jj||jj|jdS(N(RRRRR(RR((s4lib/python2.7/site-packages/numba/targets/codegen.pyRscC`stjdj}|jj|}|j}y|j|}Wn2tk rutj||d|}d|_ nX|j |j ||jS(NiR5RQ( R_tIntTypet as_pointerRRRt get_globaltKeyErrortGlobalVariableROtbitcasttload(RtbuildertfntyR5tvoidptrtptrnametllvm_modR((s4lib/python2.7/site-packages/numba/targets/codegen.pytinsert_unresolved_refs   cC`s tjdkrtjStjS(N(RtCPU_NAMER*R#tget_host_cpu_name(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR#scC`stjdk rtjStS(N(Rt CPU_FEATURESR*tget_host_cpu_features(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyt_get_host_cpu_featuress(R'R(R RR2RR RRR RCRRhR$RR1R#R6(((s4lib/python2.7/site-packages/numba/targets/codegen.pyRZs          t AOTCPUCodegencB`s;eZdZeZddZdZdZdZ RS(sp A codegen implementation suitable for Ahead-Of-Time compilation (e.g. generation of object files). cC`s#|p d|_tj||dS(NR(t _cpu_nameRR (RRtcpu_name((s4lib/python2.7/site-packages/numba/targets/codegen.pyR scC`sS|j}|dkr$|j}n||dR=t jitdefaultR?R@tjitdebug(R#R#Rtfrom_default_tripleR5RqRRm(RRARt reloc_model((s4lib/python2.7/site-packages/numba/targets/codegen.pyRs    cC`s |jS(N(R6(R((s4lib/python2.7/site-packages/numba/targets/codegen.pyR+scC`s|jj|dS(N(RR(RR((s4lib/python2.7/site-packages/numba/targets/codegen.pyR{/scC`sE|jj|}tjdj|}tjt||dRt ENABLE_AVXRqR+tflatten(R@tk((s4lib/python2.7/site-packages/numba/targets/codegen.pyR5Is   (1t __future__RRRRRRRxRtllvmlite.llvmpy.coretllvmpytcoreRtllvmlite.llvmpy.passestpassesRtllvmlite.bindingtbindingR#t llvmlite.irRdR_tnumbaRRRtnumba.runtime.nrtoptRt numba.runtimeRtnumba.compiler_lockRt frozensetRRRRRR,RRRRRRR7RBRR5(((s4lib/python2.7/site-packages/numba/targets/codegen.pyts<       u8 63