a Df+@sddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZmZejjdddZddZGdd d ejZGd d d ejZejejeeegejj_ejejj_ejejeegejj_eejj_ejejgejj _ejej!egejj"_ejgejj#_ejgejj$_ej%ejj$_gejj&_eejj&_ejegejj'_eejj'_ejegejj(_eejj(_ejej)gejj*_Gd d d e Z+e de ee+Z,e de ee+Z-e,ej.Z/e-ej0Z1e,e-e gejj2_ej3ejj2_ej3gejj4_ejej3gejj5_eejj6_eegejj6_dS)N) POINTERc_char_pc_boolc_void_pc_intc_uint64c_size_t CFUNCTYPE string_atcast py_object Structure)ffitargets object_filecCsp|durtdv}t2}tj||||}|s@tt|Wdn1sT0Yd|_t ||dS)a Create a MCJIT ExecutionEngine from the given *module* and *target_machine*. *lmm* controls whether the llvmlite memory manager is used. If not supplied, the default choice for the platform will be used (``True`` on 64-bit ARM systems, ``False`` otherwise). N)arm64aarch64T)module) platformmachiner OutputStringlibLLVMPY_CreateMCJITCompiler RuntimeErrorstr_ownedExecutionEngine)rZtarget_machineZuse_lmmouterrZenginerm/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/llvmlite/binding/executionengine.pycreate_mcjit_compiler s   *r cCs tj}|dkrt|ddS)zp Check the system allows execution of in-memory JITted functions. An exception is raised otherwise. rzcannot allocate executable memory. This may be due to security restrictions on your system, such as SELinux or similar mechanisms.N)rr"LLVMPY_TryAllocateExecutableMemoryOSError)errnorrrcheck_jit_execution#s  r$c@seZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ eddZddZddZddZd%ddZdd Zd!d"Zd#d$ZdS)&rzAn ExecutionEngine owns all Modules associated with it. Deleting the engine will remove all associated modules. It is an error to delete the associated modules. NcCs*t|g|_d|_d|_tj||dS)z; Module ownership is transferred to the EE NT)set_modules_tdrr ObjectRef__init__)selfptrrrrrr)8s zExecutionEngine.__init__cCstj||dS)z Return the address of the function named *name* as an integer. It's a fatal error in LLVM if the symbol of *name* doesn't exist. ascii)rrLLVMPY_GetFunctionAddressencoder*namerrrget_function_addressAsz$ExecutionEngine.get_function_addresscCstj||dS)z Return the address of the global value named *name* as an integer. It's a fatal error in LLVM if the symbol of *name* doesn't exist. r,)rrLLVMPY_GetGlobalValueAddressr.r/rrrget_global_value_addressIsz(ExecutionEngine.get_global_value_addresscCstj|||dSN)rrLLVMPY_AddGlobalMapping)r*gvaddrrrradd_global_mappingQsz"ExecutionEngine.add_global_mappingcCs6||jvrtdtj||d|_|j|dS)zL Ownership of module is transferred to the execution engine z#module already added to this engineTN)r&KeyErrorrrLLVMPY_AddModuleradd)r*rrrr add_moduleTs  zExecutionEngine.add_modulecCstj|dS)z} Make sure all modules owned by the execution engine are fully processed and "usable" for execution. N)rrLLVMPY_FinalizeObjectr*rrrfinalize_object^szExecutionEngine.finalize_objectcCstj|dS)zW Run static constructors which initialize module-level static objects. N)rrZLLVMPY_RunStaticConstructorsr>rrrrun_static_constructorsesz'ExecutionEngine.run_static_constructorscCstj|dS)zh Run static destructors which perform module-level cleanup of static resources. N)rrZLLVMPY_RunStaticDestructorsr>rrrrun_static_destructorsksz&ExecutionEngine.run_static_destructorscCsZt,}tj|||r&tt|Wdn1s:0Y|j|d|_dS)z1 Ownership of module is returned NF) rrrLLVMPY_RemoveModulerrr&remover)r*rrrrr remove_modulers  * zExecutionEngine.remove_modulecCs6|jdur|jStj|}t||_d|j_|jS)z; The TargetData for this execution engine. NT)r'rr#LLVMPY_GetExecutionEngineTargetDatarZ TargetDatar)r*r+rrr target_data|s    zExecutionEngine.target_datacCstj|}|S)z Enable JIT events for profiling of generated code. Return value indicates whether connection to profiling tool was successful. )rrZLLVMPY_EnableJITEvents)r*retrrrenable_jit_eventss z!ExecutionEngine.enable_jit_eventscCs6t|tj}|jD]}t|jtj|kr|SqdS)zH Find the ModuleRef corresponding to the given pointer. N)r rvaluer&Z_ptr)r* module_ptrr+rrrr_find_module_ptrs    z ExecutionEngine._find_module_ptrcCs(t|trtj|}tj||dS)z Add object file to the jit. object_file can be instance of :class:ObjectFile or a string representing file system path N) isinstancerrZ ObjectFileRef from_pathrrLLVMPY_MCJITAddObjectFile)r*Zobj_filerrradd_object_files  zExecutionEngine.add_object_filecCs*||_||_t||_tj||jdS)z~ Set the object cache "notifyObjectCompiled" and "getBuffer" callbacks to the given Python functions. N)_object_cache_notify_object_cache_getbuffer_ObjectCacheRef _object_cacherrLLVMPY_SetObjectCache)r*Z notify_funcZgetbuffer_funcrrrset_object_caches z ExecutionEngine.set_object_cachecCs`|jdurdS|jj}|jj}|jj}t||}||}|durPtd|f|||dS)z( Low-level notify hook. N5object compilation notification for unknown module %s)rPcontentsrJbuf_ptrbuf_lenr rKr)r*datarJrXrYbufrrrr_raw_object_cache_notifys   z(ExecutionEngine._raw_object_cache_notifycCsr|jdurdS|jj}||}|dur6td|f||}|durntj|t||d_ t||d_ dS)z+ Low-level getbuffer hook. NrVr) rQrWrJrKrrrLLVMPY_CreateByteStringlenrXrY)r*rZrJrr[rrr_raw_object_cache_getbuffers   z+ExecutionEngine._raw_object_cache_getbuffercCsH|jD] }|q|jdur(|j|jd|_|j|dSr4)r&detachr'clearrS_capiZLLVMPY_DisposeExecutionEngine)r*modrrr_disposes     zExecutionEngine._dispose)NN)__name__ __module__ __qualname____doc__rSr)r1r3r8r<r?r@rArDpropertyrFrHrKrOrUr\r_rdrrrrr1s(       rc@s eZdZdZddZddZdS)rRzN Internal: an ObjectCache instance for use within an ExecutionEngine. cCs"tjtt|}tj||dSr4)rrLLVMPY_CreateObjectCache_notify_c_hook_getbuffer_c_hookr(r))r*objr+rrrr)s z_ObjectCacheRef.__init__cCs|j|dSr4)rbLLVMPY_DisposeObjectCacher>rrrrdsz_ObjectCacheRef._disposeN)rerfrgrhr)rdrrrrrRsrRc@s$eZdZdejfdefdefgZdS)_ObjectCacheDatarJrXrYN)rerfrgr LLVMModuleRefrr_fields_rrrrro*sro)N)7rctypesrrrrrrrr r r r r Zllvmlite.bindingrrrrZLLVMPY_LinkInMCJITr r$r(rrRrpZLLVMTargetMachineRefrargtypesZLLVMExecutionEngineRefrestyperBr:Z LLVMValueRefr5r=rEZLLVMTargetDataRefr!r-r2ZLLVMObjectFileRefrNroZ_ObjectCacheNotifyFuncZ_ObjectCacheGetBufferFuncr\rkr_rlrjZLLVMObjectCacheRefrnrTr]rrrrs8 3