U |e)@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_machineouterrenginer]/var/www/website-v5/atlas_env/lib/python3.8/site-packages/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#_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 ObjectFileRefZ 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)rMcontentsrHbuf_ptrbuf_lenr rIr)r'datarHrUrVbufrrrr_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. NrSr) rNrTrHrIrrrLLVMPY_CreateByteStringlenrUrV)r'rWrHrrXrrr_raw_object_cache_getbuffers   z+ExecutionEngine._raw_object_cache_getbuffercCsH|jD] }|q|jdk r(|j|jd|_|j|dSr1)r#detachr$clearrP_capiZLLVMPY_DisposeExecutionEngine)r'modrrr_disposes     zExecutionEngine._dispose)NN)__name__ __module__ __qualname____doc__rPr&r.r0r5r9r<r=r>rApropertyrCrErIrLrRrYr\rarrrrr)s(       rc@s eZdZdZddZddZdS)rOzN Internal: an ObjectCache instance for use within an ExecutionEngine. cCs"tjtt|}tj||dSr1)rrLLVMPY_CreateObjectCache_notify_c_hook_getbuffer_c_hookr%r&)r'objr(rrrr&s z_ObjectCacheRef.__init__cCs|j|dSr1)r_LLVMPY_DisposeObjectCacher;rrrrasz_ObjectCacheRef._disposeN)rbrcrdrer&rarrrrrOsrOc@s$eZdZdejfdefdefgZdS)_ObjectCacheDatarHrUrVN)rbrcrdr LLVMModuleRefrr_fields_rrrrrl"srlN)6ctypesrrrrrrrr r r r r llvmlite.bindingrrrrZLLVMPY_LinkInMCJITrr!r%rrOrmLLVMTargetMachineRefrargtypesLLVMExecutionEngineRefrestyper?r7 LLVMValueRefr2r:rBLLVMTargetDataRefrr*r/LLVMObjectFileRefrKrlZ_ObjectCacheNotifyFuncZ_ObjectCacheGetBufferFuncrYrhr\rirgLLVMObjectCacheRefrkrQrZrrrrs84