B   ›\Xã@s¾dZddlZddlZddlmZddlmZddlm Z m Z ddlm Z ddl m Z e jdkoje joje j Zd d „Zd d „Zd d„ZGdd„deƒZGdd„deƒZdd„Zdd„ZeƒZdS)zD Register external C functions necessary for Numba code generation. éN)Úir)ÚutilsÚconfig)Ú _helperlibé)Ú intrinsics)éécCst |¡st ||¡dS)z1Add missing symbol into LLVM internal symtab N)ÚllZaddress_of_symbolÚ add_symbol)ÚsymbolZaddr©r ú6lib/python3.7/site-packages/numba/targets/externals.pyÚ_add_missing_symbols rcCs,ddlm}m}m}t|j|ƒ}|||ƒjS)zr Under Windows, look up a symbol inside the C runtime and return the raw pointer value as an integer. r)ÚcdllÚcastÚc_void_p)ÚctypesrrrÚgetattrZmsvcrtÚvalue)r rrrÚfr r rÚ_get_msvcrt_symbols rc Cs@| ¡}| d¡}| d¡}t d¡}t d¡}t |d¡}t |d¡}t |d¡}t |||g¡} tj|| dd} | j\} } |   ¡} t  | ¡}|  | |¡}|  | |¡}|  |  | |¡|¡}|  |  | |¡|¡}|  | ||¡| ||¡¡}| ||¡}| ||¡}| ||  | ||¡| ||¡¡¡}| || ||¡¡}| ||¡}| ||¡}| ||¡}| ||  | ||¡| ||¡¡¡}| || ||¡¡}| || ||¡¡}| ||  | ||¡| ||¡¡¡}| ||  ||¡¡}| ||  ||¡¡}| ||¡}| || | ||¡|¡¡}| |¡| |¡| ¡|S)zo Compile the multi3() helper function used by LLVM for 128-bit multiplication on 32-bit platforms. Úmulti3é@é€lÿÿé )Úname)ÚcodegenÚcreate_libraryZcreate_ir_modulerZIntTypeZConstantÚ FunctionTypeZFunctionÚargsZappend_basic_blockZ IRBuilderZtruncZashrÚmulÚand_ZlshrÚaddZshlZzextZretZ add_ir_moduleÚfinalize)ÚcontextrÚlibraryÚir_modZi64Zi128Z lower_maskZ_32Z_64Zfn_typeÚfnÚaÚbZbbZbuilderZalZblZahZbhZrlÚtZrhÚrr r rÚcompile_multi3&sP                   r-c@seZdZdZdd„ZdS)Ú _InstallerFcCs|js| |¡d|_dS)z˜ Install the functions into LLVM. This only needs to be done once, as the mappings are persistent during the process lifetime. TN)Ú _installedÚ _do_install)Úselfr%r r rÚinstallss z_Installer.installN)Ú__name__Ú __module__Ú __qualname__r/r2r r r rr.osr.c@seZdZdZdd„ZdS)Ú_ExternalMathFunctionszd Map the math functions from the C runtime library into the LLVM execution environment. cCsätjdk}tj}tj d¡r4|r4tdƒ}td|ƒn,tj d¡r`|r`td|dƒtd|d ƒ|rŒt |ƒ|_ |j   d ¡}|s‚t ‚td |ƒxBt jD]8}tr¼| d ¡r¼t ||d |¡q”t |||¡q”Wtràt||ƒ|_dS)NrZwin32Z_ftolZ_ftol2ZlinuxZ __fixunsdfdiZfptouiZ __fixunssfdiZfptouifrZ__multi3ZfmodZfixed_)rZ MACHINE_BITSrÚ c_helpersÚsysÚplatformÚ startswithrrr-Z _multi3_libÚget_pointer_to_functionÚAssertionErrorrZ INTR_MATHÚ need_kb982107r r Ú set_fnclexZ _kb982107_lib)r1r%Zis32bitr7ZftolZptrÚfnamer r rr0ƒs&      z"_ExternalMathFunctions._do_installN)r3r4r5Ú__doc__r0r r r rr6}sr6cCs8|d}t dtj¡|ƒ}t|ƒ}| d¡}||ƒ|S)zl Install fnclex before fmod calls. Workaround for https://support.microsoft.com/en-us/kb/982107 r>NZfnclex)rZ CFUNCTYPErÚcompile_fnclexr;)r%r7Zptr_set_fnclexr(r&Z fnclex_ptrr r rr>©s  r>cCs:| ¡}| d¡}d}t ¡| t |¡¡| ¡|S)zm Compile a function that calls fnclex to workround https://support.microsoft.com/en-us/kb/982107 Zkb982107zT define void @fnclex() { call void asm sideeffect "fnclex", ""() ret void } )rrr Zinitialize_native_asmparserZadd_llvm_moduleZparse_assemblyr$)r%rr&r'r r rrA¸s rA)r@r8rZllvmliterZllvmlite.bindingZbindingr ZnumbarrrÚrZ PYVERSIONZIS_WIN32Z IS_32BITSr=rrr-Úobjectr.r6r>rAZc_math_functionsr r r rÚs$      I,