o ۾i@sdZddlZddlmZddlmZddlmZm Z ddl m Z ddZ dd Z d d ZGd d d eZGdddeZeZdS)zD Register external C functions necessary for Numba code generation. N)ir)utils intrinsics) _helperlibcCst|s t||dSdS)z1Add missing symbol into LLVM internal symtab N)lladdress_of_symbol add_symbol)symboladdrr H/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/externals.py_add_missing_symbols r cCs,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)cdllcastc_void_p)ctypesrrrgetattrmsvcrtvalue)r rrrfr r r _get_msvcrt_symbols  rc Cs@|}|d}|d}td}td}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)codegencreate_librarycreate_ir_modulerIntTypeConstant FunctionTypeFunctionargsappend_basic_block IRBuildertruncashrmuland_lshraddshlzextret add_ir_modulefinalize)contextrlibraryir_modi64i128 lower_mask_32_64fn_typefnabbbbuilderalblahbhrltrhrr r r compile_multi3sV                      rGc@eZdZdZddZdS) _InstallerFcCs|js ||d|_dSdS)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)selfr1r r r installls  z_Installer.installN)__name__ __module__ __qualname__rJrMr r r r rIhs rIc@rH)_ExternalMathFunctionszd Map the math functions from the C runtime library into the LLVM execution environment. cCstjdk}tj}tjdr|rtd}td|ntjdr0|r0td|dtd|d |rFt ||_ |j d }|sAJtd |t j D] }t|||qIdS) Nrwin32_ftol_ftol2linux __fixunsdfdifptoui __fixunssfdifptouifr__multi3)r MACHINE_BITSr c_helperssysplatform startswithrr rG _multi3_libget_pointer_to_functionr INTR_MATHrr)rLr1is32bitr\ftolptrfnamer r r rK|s      z"_ExternalMathFunctions._do_installN)rNrOrP__doc__rKr r r r rQvs rQ)rgr]llvmliterllvmlite.bindingbindingr numba.corerrnumbarr rrGobjectrIrQc_math_functionsr r r r s    I %