xKgddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl mZmZddlmZmZmZddlmZddlmZddlmZddlmZddlmZdd lmZdd l m!Z!e"gd Z#d Z$d Z%dZ&Gdde'Z(GddeZ)Gdde)Z*Gdde*Z+Gdde*Z,Gdde'Z-dZ.Gdde'Z/Gdd eZ0Gd!d"e0Z1Gd#d$e1Z2Gd%d&e1Z3d'Z4d(Z5y))N)abstractmethodABCMeta)utilsconfigcgutils)create_pass_manager_builder)remove_redundant_nrt_refct)rtsys)require_global_compiler_lock)NumbaInvalidConfigWarning)disassemble_elf_to_cfg)PassTimingsCollection)x86i386i486i586i686i786i886i986c:|jdd}|tvS)N-r)split_x86arch)triplearchs V/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/core/codegen.py_is_x86rs << Q D 8 c,tjjd}|syd}|D]>}|j} |t t j |jz}@|S#t$r tjd|tYkwxYw)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 )flagsvalitems r_parse_refprune_flagsr. s  & & , ,S 1E  Czz| 5 72//> >C J 5 MM3D8<3 5 5s+A**&BBcJ tjr6 ddl}ddlm ddlm}ddlm}ddlm ddl m ||d}|| fd}nt}td t|jd d ||td y#t$rd }t|wxYw) Nr) highlight)GasLexer) LlvmLexer)Terminal256Formatter)by_colorscheme)llvmasmc Nt|y)N)style)print)argr3r4r0lexers rprinterzdump..printerHs%iUW*1ABDErz0Please install pygments to see highlighted dumpszP================================================================================Pr)rHIGHLIGHT_DUMPSpygmentsr0pygments.lexersr1r2pygments.formattersr3numba.misc.dump_styler4 ImportError ValueErrorr9center) headerbodylangr? gas_lexer llvm_lexer lexer_mapr<msgr3r4r0r;s @@@@rdumprM8s  E  + = ? @ IdOE E E (O &--C ! DM (O) "DCS/ ! "s B B"c:eZdZdZdZ ddZd dZdZdZy) _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. c ||_||_||_|j|}t j ||_||_yN)cresnamepy_func get_functionr%get_function_cfgdotkwargs)selfrRrSrTrXfns r__init__z _CFG.__init__[s@     t $&&r* rNc  KLMNOPQRSTUVWXddlSddlWddl} ddl} ddlm} ddlm} ddlm } ddl m }d}| ||||||||}| || Kd }||||K||d Nd Nd <dNd<dNd<dNd<dNd<dNd<dNd<dNd<dNd<dNd<d Nd!<|jj|j}|jj}Wfd"}||USfd#}||j|j|}Wj!d$}Wj!d%}Wj!d&Vd'}Wj!|TWj!d(}Wj!d)}Wj!d*}Wj!d+}Wj!d,}Wj!d-}Wj!d.} Wj!d/}!d0}"d1Li}#i}$Kj"r| j%|j&\XQ| j(|j+d2d3}%Sj-|%j/d45}&| j1|&j3d6}'d}(|'d7D]E})d8\PO|)d9}*|)d:}+|)d;},|+|#|,<|*d}-g}.d|>|>\PO|jJr|jM|=rNd}?n|jNr|jM|=rNd}?n|jPr|jM|=rNd}?n|jRr| jM|=rNd}?nn|jTr|!jM|=rNd}?nK|jVr|jM|=rNd}?n(|jXr|jM|=rNd!}?nNdD}?|?NdDus|rC|"L|=D]$}@|.jARjC|?M|@&z|/r |.jAdGjC|/d3jE|.}A|ArdH|(dIAdJ}BdKjC|B}*nd3}*|j[|+|*LHdM|'vr/|'dMD]'}C|Cd;},|Cj?dNd}D|CdO|CdP|Df|$|,<)|$j]D]6\},}C|#|Cd<}E|#|Cd}F|CdQ}G|GEdRGzz }E|j_EF8|rlg}HNj]D]&\}I}JHjAdSjC|J|I(|j[dTdUjCd3jEHL|||ja|||V|j/dW5S)Xz "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)pythonlineinfoct|tr|jD]}t|||yt|trd|j D]P\}}||jvrt d|zt|trt|||@d}t |||fzyt|tr1|D]+}||jvrt d|zt||d-yd}t |t|z)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__setattrdictitemsrDsettype)_configkwargattrkvrLr-s r parse_configz)_CFG.pretty_printer..parse_configs %&#,,DGT51-E4(!KKMDAq 0 00()F)JKK!!T*A.D(1v66*E3'!D7#3#33()F)MNNt4 " B tE{!233rcy)Nwhiter|rrz%_CFG.pretty_printer..srorangemarkeryellowrigreentruebrredfalsebrcyanra turquoiserb lightpinkraise lightseagreenrepurplereturn rosybrownrgtomatorhci}jd}|jD]D}|j|}||j}|*t |dk(sJ|d||d<F|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+.*)r)compile splitlinesmatchgroupslen)llvm_strmdmetadata_entryxrgres r get_metadataz)_CFG.pretty_printer..get_metadataszBZZ(BCN((*&,,Q/$ A}"1v{*{#$Q41Q4 +Ircd}t||kDr%d|d}tj|t|d|}j ||}|j d|j ddd t |z |S) NzCFG output filename "z9" exceeds maximum supported length, it will be truncated.)filenameTB)rankdirnodenonez%s)shapefontsize)rr)r*r Digraphrvstr)rSfnamercmaxwstrfgvs r init_digraphz)_CFG.pretty_printer..init_digraphsD5zD /w7BC d$=>et  4% 0A FF4F FF6$X2FF GHrz .*{(.*)}.*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],?.*c2tj|ddS)Nxz... )widthsubsequent_indent)textwrapwrap)ss rrz!_CFG.pretty_printer..wraps==#H HrcVd}t||kDr)tt|}dj|d||}t j |}|j dd}|j dd}|j dd}|j d d }|j d d }|S) Ni,z{}...z\{z{z\}z}\z\%z%!z!)rrhashformathtmlescapereplace)rnhss rcleanz"_CFG.pretty_printer..cleansA1vza\$++AbqE26 AA %*A %*A $(A #w'A #w'AHrz\l...dot_json)rutf-8objects)rlabelrS_gvidrrz\l|z<{})TFr{zF{}defaultzF{}c j|}|1|j}|t|dk(sJ||d}j|d}|j|}|t|jdk(sJ||j\}}|k7s|k7rjr7d} | |||fz} j d| } |j | jrFt|dzz } | dz} j d| } |j | ||fSyyyyy)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 %srri) rrrgetrjrappendriint)l new_linesmatchedrr debug_dataldlinecolmfmt mark_linelnlidx source_line _interleavercol_spancscur_colcur_line firstlinenofmtlocation_entryrmetadata_markersrc_codes rmetadata_interleavez0_CFG.pretty_printer..metadata_interleaveis| *//2&(A}"1v{-A-{!"1%'VVFD%9 %1"0!5!5j!AB!~'*299;'71'< @b @'<,.IIK c$(8#3sg~'2';';/N48FD#;N4N -0ZZ8 h8=i8H.J(1(8(8(<'2'9'9 034yK!O/L6>tax6H -0ZZ8 h8=k8J.L(1(8(8(<+/9$4%8F . 2 %'rz {}z z
z<{}>)redgestailportheadtailrz:%sz;{}Keyzn<{}
Key:
>)rviewrsvg)1graphvizrjsoninspectllvmliter] numba.typedr^typesr_ collectionsr`rRrUrS get_llvm_strrrigetsourcelinesrTrVrSourcepipeloadsdecoderrrrrfrrrjoinr#rjrasearchrbrerdrcrgrhrrqedgerender)YrYrr render_formatr0 interleavestrip_irshow_keyrrrr%r^r_r`_default _highlightryrZrrrr 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_bytesjzonidcobjrrSgvidlinesr port_liner sliced_linesports ports_tokenstdfmttbl_datacolorstoktargetvaluecolor fmtheaderrrupdated_lineinfocolourrdattabrtprrportkey_tabrwrxrrrrrrrrrrrrrrsY @@@@@@@@@@@@@@rpretty_printerz_CFG.pretty_printercsM *$)+$H+3,4+3,4-56>3;> &XI  46 Z+[*- )8 8 8 9 8 "8 !7 '9 8 "- ' YY # #DII .99))+ ( # $ DIIx 8 ZZ - N3**%<=4 M2JJ?@ ZZ 01 ZZ 01 jj0 **%9:::n5::CDJJ12  I  ,   $+$:$:4<<$H !Hk &"%%b)11(B?YYw',,J,? zz*++G45 ?C!' HgLEv;Dw,>q,A ]F&&8??1+=[F''I,<,rSptc_names rr[zCodeLibrary.__init__sF  nn--.a ~Q?!6x!@ "rcR|jt|jdkDS)Nr)_ensure_finalizedrrGr2s rhas_dynamic_globalszCodeLibrary.has_dynamic_globalss$  4(()A--rc|jSrQ)rFr2s rrecorded_timingszCodeLibrary.recorded_timingss%%%rc|jS)z9 The codegen object owning this library. )rCr2s rr>zCodeLibrary.codegen"s }}rc|jSrQ)rDr2s rrSzCodeLibrary.name)s zzrc6d|jt|fzS)Nz)rSidr2s rr5zCodeLibrary.__repr__-s%BtH(===rc8|jrtd|y)Nz)operation impossible on finalized object ) _finalized RuntimeErrorr2s r_raise_if_finalizedzCodeLibrary._raise_if_finalized0s# ??"& )* * rc>|js|jyyrQ)rSfinalizer2s rrJzCodeLibrary._ensure_finalized5s MMOrc\|j|jj|}|S)zC Create an LLVM IR module for use by this library. )rUrC_create_empty_modulerYrS ir_modules rcreate_ir_modulezCodeLibrary.create_ir_module9s*   "MM66t< rcy)zk Add a library for linking into this library, without losing the original library. Nr|rYlibrarys radd_linking_libraryzCodeLibrary.add_linking_libraryArcy)zC Add an LLVM IR module's contents to this library. Nr|)rYr[s r add_ir_modulezCodeLibrary.add_ir_moduleHrarcy)z Finalize the library. After this call, nothing can be added anymore. Finalization involves various stages of code optimization and linking. Nr|r2s rrWzCodeLibrary.finalizeNrarcy)z5 Return the function named ``name``. Nr|rYrSs rrUzCodeLibrary.get_functionVrarcy)zA Get the human-readable form of the LLVM module. Nr|r2s rrzCodeLibrary.get_llvm_str\rarcy)z2 Get the human-readable assembly. Nr|r2s r get_asm_strzCodeLibrary.get_asm_strbrarc.d|_d|_d|_y)NTF)_object_caching_enabled_compiled_object _compiledr2s renable_object_cachingz!CodeLibrary.enable_object_cachingls'+$ $rc|jstd||jtd||jS)Nobject caching not enabled in zno compiled object yet for )rkrDrlrTr2s r_get_compiled_objectz CodeLibrary._get_compiled_objectqs?++4IJ J  ($HI I$$$rc|jstd||jrtd|||_d|_y)Nrpzlibrary already compiled: T)rkrDrmrl_disable_inspection)rYr"s r_set_compiled_objectz CodeLibrary._set_compiled_objectxs>++4IJ J >>tEF F %#' rN)r8r9r:r;rSrkrsrr[propertyrKrMr>rSr5rUrJr\rr`rcrWrUrrirnrqrtr|rrr=r=s J## #C#..&& >*              %(rr=) metaclassceZdZfdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZddZdZedZedZedZdZdZedZxZS)CPUCodeLibraryc*t|||g|_tjt |j j|j|_ tj|j|j_d|_ yrQ) superr[_linking_librariesr%parse_assemblyrrCrYrS _final_modulernormalize_ir_text_shared_module)rYr>rSrEs rr[zCPUCodeLibrary.__init__sn $'"$..  22499= >@")";";DII"F"rc|jj|_|jj|5}|jD]f}d|j }|j j|5|j|j||jdddh dddy#1swY|xYw#1swYyxYw)zP Internal: run function-level optimizations inside *ll_module*. zFunction passes on N) rC _data_layout data_layout_function_pass_manager functionsrSrFrecord initializerunrW)rY ll_modulefpmfuncrws r_optimize_functionsz"CPUCodeLibrary._optimize_functionss !% : :  ]] 1 1) <"++)$))7++2215NN$GGDMLLN65,= < 65 = ,>?    T7 + !  $tyy0$2C2C2Ev N    *D,<,<,> F rc#hK|j}|jD]}|jr|yw)zj Get all functions defined in the library. The library must have been finalized. N)r}rr)rYrrZs rget_defined_functionsz$CPUCodeLibrary.get_defined_functions's/   --B$$ s(22c8|jj|SrQ)r}rUrfs rrUzCPUCodeLibrary.get_function1s!!..t44rcH|jrtjdyy)Nz@Inspection disabled for cached code. Invalid result is returned.)rsr)r*r2s r _sentry_cache_disable_inspectionz/CPUCodeLibrary._sentry_cache_disable_inspection4s!  # # MM8 9 $rcL|jt|jSrQ)rrr}r2s rrzCPUCodeLibrary.get_llvm_str9s --/4%%&&rc|jt|jjj |j SrQ)rrrC_tm emit_assemblyr}r2s rrizCPUCodeLibrary.get_asm_str=s4 --/4==$$2243E3EFGGrc >|jt|||fi|S)z= Get control-flow graph of the LLVM function )rrO)rYrSrTrXs rrVzCPUCodeLibrary.get_function_cfgAs$ --/D$2622rc:|j}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. )rqr )rY mangled_nameelfs rget_disasm_cfgzCPUCodeLibrary.get_disasm_cfgHs'')%c<88rc ddlm}ddlm}ddlm}|||}t d|jD]}|ddk(s t|jd }t d |D]j}|jst d |jj|d |d |j|dd|j|ddfzlt y)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_SYMTABc|jSrQrS)syms rr}z*CPUCodeLibrary._dump_elf..bsSXXr)keyz symbols:z/ - %r: size=%d, value=0x%x, type=%s, bind=%sst_sizest_valuest_inforsbindN)elftools.elf.elffiler elftools.elfriorr9 iter_sectionssorted iter_symbolsrSrdescribe_symbol_typedescribe_symbol_bind) clsbufrrrrsecsymbolsrs r _dump_elfzCPUCodeLibrary._dump_elfUs 1- GCL ! k??$C9~- !1!1!39MNn%"C88 K XX__. ^ _)>>s9~f?UV)>>s9~f?UV  # % rcr |j}|jrd|_||_yy#t$rYywxYw)zB `ll_module` was compiled into object code `buf`. NT)rr(rkrmrl)rrrrYs r_object_compiled_hookz$CPUCodeLibrary._object_compiled_hookpsE  &&D  ' '!DN$'D ! (   s * 66c |j}|jr"|jr|j}d|_|Syy#t$rYywxYw)z> Return a cached object code for `ll_module`. N)rr(rkrl)rrrYrs r_object_getbuffer_hookz%CPUCodeLibrary._object_getbuffer_hook}sZ  &&D  ' 'D,A,A''C$(D !J-B '   s = A A cp|j|jd|jjfS)zX Serialize this library using its bitcode as the cached representation. bitcode)rJrSr} as_bitcoder2s rserialize_using_bitcodez&CPUCodeLibrary.serialize_using_bitcodes0   9d&8&8&C&C&EFFrc|j|j|jjf}|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)rJrqrrrS)rYdatas rserialize_using_object_codez*CPUCodeLibrary.serialize_using_object_codesI  ))+,,.99;= 8T**rc|\}}}|j|}t||sJ|dk(r,tj||_|j |S|dk(r|\}}|j |j|tj||_|j |jjj|j|Std|)Nrrzunsupported serialization kind ) create_libraryrlr% parse_bitcoder}rrnrtrrC_engine_load_defined_symbolsrD) rr>staterSkindrrY object_codeshared_bitcodes r _unserializezCPUCodeLibrary._unserializes dD%%d+$$$$ 9 !#!1!1$!7D   ' ' )K X *. 'K  & & (  % %k 2"$"2"2>"BD   ' ' ) MM ! ! 7 78K8K LKDJK KrrQ)r8r9r:r[rrrr`rcrrWrrrrrUrrrirVr classmethodrrrrrr __classcell__)rEs@rrxrxs## <"!F0(.&86 :G459 'H3 94 ( (  G +LLrrxceZdZdZdZdZy)AOTCodeLibraryc|j|jjj|jS)z Return this library as a native object (a bytestring) -- for example ELF under Linux. This function implicitly calls .finalize(). )rJrCr emit_objectr}r2s remit_native_objectz!AOTCodeLibrary.emit_native_objects1  }}  ,,T-?-?@@rcV|j|jjS)zz Return this library as LLVM bitcode (a bytestring). This function implicitly calls .finalize(). )rJr}rr2s r emit_bitcodezAOTCodeLibrary.emit_bitcodes%  !!,,..rcyrQr|r2s rrz!AOTCodeLibrary._finalize_specific rN)r8r9r:rrrr|rrrrsA/ rrceZdZdZdZy)JITCodeLibraryc|j|jj}|j|sy|jjj |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)rJrCris_symbol_definedget_function_address)rYrSees rget_pointer_to_functionz&JITCodeLibrary.get_pointer_to_functionsK  ]] " "##D)==((==dC Crc|jj|j|jj d5|jj j dddy#1swYyxYw)NzFinalize object)rC_scan_and_fix_unresolved_refsr}rFrrfinalize_objectr2s rrz!JITCodeLibrary._finalize_specificsQ 33D4F4FG  # # * *+< = MM ! ! 1 1 3> = =s %A//A8N)r8r9r:rrr|rrr r s D*4rr c,eZdZdZdZdZdZdZdZy) RuntimeLinkerzP For tracking unresolved symbols generated at runtime due to recursion. z.numba.unresolved$cbtj|_t|_g|_yrQ)r UniqueDict _unresolvedrr_defined _resolvedr2s rr[zRuntimeLinker.__init__s# ++- rc|j}|jD]}|jj|s|jt |d}|j |jrSt jjd}tj|}|j|tj|||j|<y)zr Scan and track all unresolved external symbols in the module and allocate memory for it. Nnrt_unresolved_abort)PREFIXrrSrrr r r_rctypesc_void_padd_global_mapping addressofr)rYmoduleengineprefixrrabortfnptrs rscan_unresolved_symbolsz%RuntimeLinker.scan_unresolved_symbolss ))Bww!!&)ggc&kl+++BGG4--??@VWoog.))"f.>.>s.CD(+  %*rc|jD]4}|jr|jj|j6y)z5 Scan and track all defined symbols. N)rrrrrS)rYr!rZs rscan_defined_symbolsz"RuntimeLinker.scan_defined_symbolss3""B$$ !!"''*#rc|jDcgc]}||jvs|}}|D]S}|j|}|j|}||_|jj ||f|j|=Uycc}w)z= Fix unresolved symbols if they are defined. N)rrr r"rr)rYr"rSpendingfnptrr%s rresolvezRuntimeLinker.resolves %)$4$4N$4D 8M4$4ND//5E""4(CCI NN ! !4+ .  &Os BBN) r8r9r:r;rr[r&r(r,r|rrrrs""F ,&+'rrcBtjfd}|S)Nc0|jg|i|SrQ)_ee)rYargsrXolds rwrapperz_proxy..wrapper"s488-d-f--r) functoolswraps)r1r2s` r_proxyr5!s#__S.. NrceZdZdZdZdZdZdZdZe e jjZ e e jjZ e e jjZe e jjZy) 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. c0||_t|_yrQ)r/rr_defined_symbols)rYrs rr[zJitEngine.__init__-s!$rc||jvS)z/Is the symbol defined in this session? )r9rfs rr zJitEngine.is_symbol_defined:st,,,,rc|j|jfD];}|xj|Dchc]}|js |jc}zc_=ycc}w)z(Extract symbols from the module N)rrr9rrS)rYrgsetsrs rrzJitEngine._load_defined_symbols?s\mmS%9%9:E  ! !&@"-/->->')gg&@ @ !;&@sA cZ|j||jj|S)zXOverride ExecutionEngine.add_module to keep info about defined symbols. )rr/ add_modulerYr!s rr>zJitEngine.add_moduleFs' ""6*xx""6**rc|jj|j|jj ||S)z`Override ExecutionEngine.add_global_mapping to keep info about defined symbols. )r9rrSr/r)rYraddrs rrzJitEngine.add_global_mappingMs3 !!"''*xx**2t44rN)r8r9r:r;r[r rr>rr5r%ExecutionEngineset_object_cacherr get_global_value_addressr|rrr7r7(s &- @+5b00AABR//??@O!""4"4"I"IJ% 33 rr7cLeZdZdZedZedZedZdZ dZ y)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. cy)zD Create a new empty module suitable for the target. Nr|rfs rrYzCodegen._create_empty_modulejrarcy)zu Add a module to the execution engine. Ownership of the module is transferred to the engine. Nr|r?s rrzCodegen._add_moduleprarc|jS)zJ The LLVM "target data" object for this codegen instance. ) _target_datar2s r target_datazCodegen.target_dataws    rc *|j||fi|S)zb Create a :class:`CodeLibrary` object for use with this codegen instance. )_library_class)rYrSrXs rrzCodegen.create_library~s #t""4888rc:|jj||SrQ)rMr)rY serializeds runserialize_libraryzCodegen.unserialize_librarys""//jAArN) r8r9r:r;rrYrrurKrrPr|rrrFrF_sN    !! 9BrrFcTeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy )r?ctd|_tjt |j ||_d|j _t|_ |j|j y)Nglobal_codegen_module) initialize_llvmrr%r|rrY _llvm_modulerSr _rtlinker_init)rY module_names rr[zCPUCodegen.__init__s_ -- ))+6 79!8& 4$$%rct|jgk(sJdtjj tj }t tj}|j|_ |j||jd i|}tj}tj|||}tjr|j!||_t%||_|j(|_t-|j*|_tjj0rd}d}nd}d}|j3|d|d |_|j3|_|j&j9|j:j<|j:j>y) NzModule isn't empty)opt)use_lmmTFrcheap)loop_vectorize slp_vectorizerZcostr|) listrr%Target from_tripleget_process_triplerprOPT_customize_tm_features _tm_features_customize_tm_optionscreate_target_machineUSE_LLVMLITE_MEMORY_MANAGERcreate_mcjit_compilerENABLE_PROFILINGenable_jit_eventsrr7rrKrJrr is_opt_max_module_pass_managerrrrCrMrr) rY llvm_moduler! tm_optionstmr[r"loopvect opt_levels rrWzCPUCodegen._initsrK001R7M9MM7&&r'<'<'>?fjj)  779 "":. )V ) ) 7J 744))+r7K  " "  $ $ & ( ".. 1 12 :: HI HI338BG8A9@4B 224 %%d&9&9&O&O&*&9&9&P&P Rrctjtj|}t j |_|jr|j|_|SrQ) rrrr~r%rdrrrrZs rrYzCPUCodegen._create_empty_modulesIMM'";";D"AB 002    $($5$5I !rc $tj}|jtj|jj ||j dd}|jdi|5}|j|ddd|H|dk(rCtjdk7r0|j|j|jtjr|jt!|S#1swY~xYw)Nr`r]rr|)r%create_module_pass_manageradd_target_library_infordradd_analysis_passespop_pass_manager_builderpopulaterreadd_loop_rotate_passadd_instruction_combining_passadd_jump_threading_passradd_refprune_passr.)rYrXpmr`pmbs rrozCPUCodegen._module_pass_managers  * * , ""2#8#8#:; $$R(zz&$' 'T ' ' 1& 1S LL 2  FJJ!O  # # %  - - /  & & (  $ $  !6!8 9 2 1s 7DDc ^tj|}|j|j|jj ||j di|5}|j|dddtjr|jt|S#1swY4xYw)Nr|) r%create_function_pass_managerrxrrryr{r|rrrr.)rYrprXrrs rrz!CPUCodegen._function_pass_managers  , ,[ 9 "";#5#56 $$R( 'T ' ' 1& 1S LL 2  $ $  !6!8 9 2 1s B##B,c |jdtj}|jdtj}|jdtj}t d|||d|}|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... rZr^r_)rZr^r_r|)rzrreLOOP_VECTORIZE SLP_VECTORIZEr)rYrXrtr^r_rs rr{z CPUCodegen._pass_manager_buildersmJJufjj1 $4f6K6KL ?F4H4HI )4i9G8E4-34  rcd}tj|}t|}d|vsd|vryd|vr#tj}t d|dt d|d ) z< Guard against some well-known LLVM bug(s). zo define double @func() { ret double 1.23e+01 } z12.3z1.23Nz1.0zFLLVM will produce incorrect floating-point code in the current locale zh. Please read https://numba.readthedocs.io/en/stable/user/faq.html#llvm-locale-bug for more information.zUnexpected IR:  )r%r|rlocale getlocalerTr)rYrrir_outlocs rrzCPUCodegen._check_llvm_bugssr #S V v/  F?""$C    v?@@rcd|jj|j|jfS)zP Return a tuple unambiguously describing the codegen behaviour. )rUr_get_host_cpu_namergr2s r magic_tuplezCPUCodegen.magic_tuples1!!(($*A*A*C!!# #rc|jj||j|jj||jj |jyrQ)rVr&rr(r,r?s rrz(CPUCodegen._scan_and_fix_unresolved_refs#sB ..vt||D ++F3 t||,rcptjdj}|jj|z}|j } |j |}|j|j||jS#t$r"tj|||}d|_ YYwxYw)Nrr) rIntType as_pointerrVrr! get_globalKeyErrorGlobalVariablerbitcastload)rYbuilderfntyrSvoidptrptrnamellvm_modr+s rinsert_unresolved_refz CPUCodegen.insert_unresolved_ref(s..#..0..''$.>> '''0E w||E2DOO4EFF  '))(G'JE&EM 's B (B54B5cjtjtjStjSrQ)rCPU_NAMEr%get_host_cpu_namer2s rrzCPUCodegen._get_host_cpu_name4s+??*$$& &__ &rcVtjtjStSrQ)r CPU_FEATURESget_host_cpu_featuresr2s r_get_host_cpu_featuresz!CPUCodegen._get_host_cpu_features9s#    *&& &$&&rN)r8r9r:r[rWrYrorr{rrrrrrr|rrr?r?sB&-R^*(A<#- G& 'rr?c.eZdZdZeZddZdZdZdZ y) AOTCPUCodegenzp A codegen implementation suitable for Ahead-Of-Time compilation (e.g. generation of object files). NcF|xsd|_tj||yNr) _cpu_namer?r[)rYrXcpu_names rr[zAOTCPUCodegen.__init__Gs!RD+.rc|j}|dk(r|j}||d<d|d<d|d<|j|d<y)Nhostcpupicrelocr codemodelfeatures)rrrg)rYoptionsrs rrhz#AOTCPUCodegen._customize_tm_optionsLsN>> v ..0H! ( "// rcyrr|r2s rrfz$AOTCPUCodegen._customize_tm_featuresUsrcyrQr|r?s rrzAOTCPUCodegen._add_moduleZrrrQ) r8r9r:r;rrMr[rhrfrr|rrrr?s" $N/ 0  rrc,eZdZdZeZdZdZdZdZ y) JITCPUCodegenzI A codegen implementation suitable for Just-In-Time compilation. c|j|d<tjjj}|j drd}n|j drd}nd}||d<d|d <|j |d <tjtjj}d |jvrd |d <yy) Nrrstaticppcrrr jitdefaultrrjitT) rr%rbfrom_default_triplerSrrgr pysignatureri parameters)rYrr reloc_modelsigs rrhz#JITCPUCodegen._customize_tm_optionses002yy,,.33 ??5 !"K __U #K#K&+ #//  ? ?@ CNN "!GEN #rc"|jSrQ)rr2s rrfz$JITCPUCodegen._customize_tm_featuress**,,rc:|jj|yrQ)rr>r?s rrzJITCPUCodegen._add_modules 'rc|jj|}tjdzj |}tjt ||d<y)zrSet the environment address. Update the GlobalVariable named *env_name* to the address of *env*. rrN)rrDrr from_addressrQ)rYenv_nameenvgvaddrenvptrs rset_envzJITCPUCodegen.set_envsG 66x@//A%33F;OOBsG,q rN) r8r9r:r;r rMrhrfrrr|rrrr^s"$N"6-(-rrc|tjtjtjy)z Safe to use multiple times. N)r%rinitialize_native_targetinitialize_native_asmprinterr|rrrTrTs%MMO!##%rc tj}tjs|D]}|j dsd||<|j S#t $rYywxYw)z~Get host CPU features using LLVM. The features may be modified due to user setting. See numba.config.ENABLE_AVX. avxFr)r%rr ENABLE_AVXrflattenrT)rrws rrrsd "++-  <<&"'HQK !! sA A A )6r)r3rrrrrllvmlite.bindingr]r% llvmlite.irrrabcrr numba.corerrrnumba.core.llvm_bindingsrnumba.core.runtime.nrtoptr numba.core.runtimer numba.core.compiler_lockr numba.core.errorsr numba.misc.inspectionr numba.misc.llvm_pass_timingsr frozensetrrr.rMrrOr=rxrr rr5r7rFr?rrrTrr|rrrs  '--@@$A78> & ' 06p6pf x(Gx(vpL[pLf  ^ 24^4:5'F5'n4 4 n'B'BTs's'l J >5-J5-p&"r