U +d)@sddlmZmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZmZmZejjddZddZGdddejZGd d d ejZejejeegejj_ejejj_ejejeegejj_eejj_ejejgejj_ejej egejj!_ejgejj"_ejgejj#_ej$ejj#_gejj%_eejj%_ejegejj&_eejj&_ejegejj'_eejj'_ejej(gejj)_Gd d d e Z*ed e ee*Z+ed e ee*Z,e+ej-Z.e,ej/Z0e+e,e gejj1_ej2ejj1_ej2gejj3_ejej2gejj4_eejj5_eegejj5_d S)) POINTERc_char_pc_boolc_void_pc_intc_uint64c_size_t CFUNCTYPE string_atcast py_object Structure)ffitargets object_filec CsFt&}tj|||}|s*tt|W5QRXd|_t||dS)zZ Create a MCJIT ExecutionEngine from the given *module* and *target_machine*. T)module)r OutputStringlibLLVMPY_CreateMCJITCompiler RuntimeErrorstr_ownedExecutionEngine)rZtarget_machineouterrZenginerD/tmp/pip-unpacked-wheel-stw2luzp/llvmlite/binding/executionengine.pycreate_mcjit_compiler s rcCs 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_executions  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%0s 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_address9sz$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_addressAsz(ExecutionEngine.get_global_value_addresscCstj|||dSN)rrLLVMPY_AddGlobalMapping)r&gvaddrrrradd_global_mappingIsz"ExecutionEngine.add_global_mappingcCs6||jkrtdtj||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_moduleMs  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_objectWszExecutionEngine.finalize_objectcCstj|dS)zW Run static constructors which initialize module-level static objects. N)rrZLLVMPY_RunStaticConstructorsr:rrrrun_static_constructors^sz'ExecutionEngine.run_static_constructorscCstj|dS)zh Run static destructors which perform module-level cleanup of static resources. N)rrZLLVMPY_RunStaticDestructorsr:rrrrun_static_destructorsdsz&ExecutionEngine.run_static_destructorsc CsFt"}tj|||r&tt|W5QRX|j|d|_dS)z1 Ownership of module is returned FN) rrrLLVMPY_RemoveModulerrr"remover)r&rrrrr remove_moduleks   zExecutionEngine.remove_modulecCs6|jdk r|jStj|}t||_d|j_|jS)z; The TargetData for this execution engine. NT)r#rr#LLVMPY_GetExecutionEngineTargetDatarZ TargetDatar)r&r'rrr target_dataus    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`|jdkrdS|jj}|jj}|jj}t||}||}|dkrPtd|f|||dS)z( Low-level notify hook. N5object compilation notification for unknown module %s)rLcontentsrFbuf_ptrbuf_lenr rGr)r&datarFrTrUbufrrrr_raw_object_cache_notifys   z(ExecutionEngine._raw_object_cache_notifycCsr|jdkrdS|jj}||}|dkr6td|f||}|dk rntj|t||d_ t||d_ dS)z+ Low-level getbuffer hook. NrRr) rMrSrFrGrrrLLVMPY_CreateByteStringlenrTrU)r&rVrFrrWrrr_raw_object_cache_getbuffers   z+ExecutionEngine._raw_object_cache_getbuffercCsH|jD] }|q|jdk r(|j|jd|_|j|dSr0)r"detachr#clearrO_capiZLLVMPY_DisposeExecutionEngine)r&modrrr_disposes     zExecutionEngine._dispose)NN)__name__ __module__ __qualname____doc__rOr%r-r/r4r8r;r<r=r@propertyrBrDrGrKrQrXr[r`rrrrr)s(       rc@s eZdZdZddZddZdS)rNzN Internal: an ObjectCache instance for use within an ExecutionEngine. cCs"tjtt|}tj||dSr0)rrLLVMPY_CreateObjectCache_notify_c_hook_getbuffer_c_hookr$r%)r&objr'rrrr%s z_ObjectCacheRef.__init__cCs|j|dSr0)r^LLVMPY_DisposeObjectCacher:rrrr`sz_ObjectCacheRef._disposeN)rarbrcrdr%r`rrrrrNsrNc@s$eZdZdejfdefdefgZdS)_ObjectCacheDatarFrTrUN)rarbrcr LLVMModuleRefrr_fields_rrrrrk"srkN)6ctypesrrrrrrrr r r r r Zllvmlite.bindingrrrrZLLVMPY_LinkInMCJITrr r$rrNrlZLLVMTargetMachineRefrargtypesZLLVMExecutionEngineRefrestyper>r6Z LLVMValueRefr1r9rAZLLVMTargetDataRefrr)r.ZLLVMObjectFileRefrJrkZ_ObjectCacheNotifyFuncZ_ObjectCacheGetBufferFuncrXrgr[rhrfZLLVMObjectCacheRefrjrPrYrrrrs84