o 4ήc)@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_filecCsZt}tj|||}|stt|Wdn1swYd|_t||dS)zZ Create a MCJIT ExecutionEngine from the given *module* and *target_machine*. NT)module)r OutputStringlibLLVMPY_CreateMCJITCompiler RuntimeErrorstr_ownedExecutionEngine)rtarget_machineouterrenginerG/tmp/pip-target-vg8gfxp4/lib/python/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__cCtj||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_address9z$ExecutionEngine.get_function_addresscCr*)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_addressAr1z(ExecutionEngine.get_global_value_addresscCstj|||dSN)rrLLVMPY_AddGlobalMapping)r(gvaddrrrradd_global_mappingIsz"ExecutionEngine.add_global_mappingcCs6||jvr tdtj||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_modulecCtj|dS)z} Make sure all modules owned by the execution engine are fully processed and "usable" for execution. N)rrLLVMPY_FinalizeObjectr(rrrfinalize_objectWzExecutionEngine.finalize_objectcCr=)zW Run static constructors which initialize module-level static objects. N)rrLLVMPY_RunStaticConstructorsr?rrrrun_static_constructors^sz'ExecutionEngine.run_static_constructorscCr=)zh Run static destructors which perform module-level cleanup of static resources. N)rrLLVMPY_RunStaticDestructorsr?rrrrun_static_destructorsdrAz&ExecutionEngine.run_static_destructorscCsZt}tj|||rtt|Wdn1swY|j|d|_dS)z1 Ownership of module is returned NF) rrrLLVMPY_RemoveModulerrr$remover)r(rrrrr remove_moduleks    zExecutionEngine.remove_modulecCs6|jdur|jStj|}t||_d|j_|jS)z; The TargetData for this execution engine. NT)r%rr#LLVMPY_GetExecutionEngineTargetDatar 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. )rrLLVMPY_EnableJITEvents)r(retrrrenable_jit_eventss z!ExecutionEngine.enable_jit_eventscCs6t|tj}|jD]}t|jtj|kr|Sq dS)zH Find the ModuleRef corresponding to the given pointer. N)r rvaluer$_ptr)r( module_ptrr)rrrr_find_module_ptrs  z ExecutionEngine._find_module_ptrcCs(t|tr tj|}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) isinstancerr ObjectFileRef from_pathrrLLVMPY_MCJITAddObjectFile)r(obj_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( notify_funcgetbuffer_funcrrrset_object_caches z ExecutionEngine.set_object_cachecCs`|jdurdS|jj}|jj}|jj}t||}||}|dur(td|f|||dS)z( Low-level notify hook. N5object compilation notification for unknown module %s)rYcontentsrQbuf_ptrbuf_lenr rRr)r(datarQrcrdbufrrrr_raw_object_cache_notifys   z(ExecutionEngine._raw_object_cache_notifycCsv|jdurdS|jj}||}|durtd|f||}|dur9tj|t||d_ t||d_ dSdS)z+ Low-level getbuffer hook. Nrar) rZrbrQrRrrrLLVMPY_CreateByteStringlenrcrd)r(rerQrrfrrr_raw_object_cache_getbuffers   z+ExecutionEngine._raw_object_cache_getbuffercCsH|jD]}|q|jdur|j|jd|_|j|dSr4)r$detachr%clearr\_capiLLVMPY_DisposeExecutionEngine)r(modrrr_disposes     zExecutionEngine._dispose)NN)__name__ __module__ __qualname____doc__r\r'r0r3r8r<r@rCrErHpropertyrKrNrRrXr`rgrjrprrrrr)s*        rc@s eZdZdZddZddZdS)r[zN 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)rmLLVMPY_DisposeObjectCacher?rrrrpsz_ObjectCacheRef._disposeN)rqrrrsrtr'rprrrrr[s r[c@s$eZdZdejfdefdefgZdS)_ObjectCacheDatarQrcrdN)rqrrrsr LLVMModuleRefrr_fields_rrrrr{"s r{N)6ctypesrrrrrrrr r r r r llvmlite.bindingrrrrLLVMPY_LinkInMCJITrr"r&rr[r|LLVMTargetMachineRefrargtypesLLVMExecutionEngineRefrestyperFr: LLVMValueRefr5r>rILLVMTargetDataRefrr,r2LLVMObjectFileRefrVr{_ObjectCacheNotifyFunc_ObjectCacheGetBufferFuncrgrwrjrxrvLLVMObjectCacheRefrzr]rhrrrrs84