B Ꮚ\)@s*ddlmZmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZddlZddlmZmZmZejjddZdd ZGd d d ejZGd d d ejZejejeegejj_ej ejj_!ej ejeegejj"_eejj"_!ej ejgejj#_ej ej$egejj%_ej gejj&_ej gejj'_ej(ejj'_!gejj)_eejj)_!ej egejj*_e ejj*_!ej egejj+_e ejj+_!ej ej,gejj-_GdddeZ.e deee.Z/e deee.Z0e/ej1Z2e0ej3Z4e/e0egejj5_ej6ejj5_!ej6gejj7_ej ej6gejj8_eejj9_!ee gejj9_dS))print_functionabsolute_import) POINTERc_char_pc_boolc_void_pc_intc_uint64c_size_t CFUNCTYPE string_atcast py_object StructureN)ffitargets object_filec CsFt&}tj|||}|s*tt|WdQRXd|_t||dS)zZ Create a MCJIT ExecutionEngine from the given *module* and *target_machine*. NT)module)r OutputStringlibLLVMPY_CreateMCJITCompiler RuntimeErrorstr_ownedExecutionEngine)rZtarget_machineouterrZenginer?lib/python3.7/site-packages/llvmlite/binding/executionengine.pycreate_mcjit_compilers  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(3s 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_GetFunctionAddressencode)r)namerrrget_function_address<sz$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)r.rrrget_global_value_addressDsz(ExecutionEngine.get_global_value_addresscCstj|||dS)N)rrLLVMPY_AddGlobalMapping)r)ZgvZaddrrrradd_global_mappingLsz"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_modulePs  zExecutionEngine.add_modulecCstj|dS)z} Make sure all modules owned by the execution engine are fully processed and "usable" for execution. N)rrLLVMPY_FinalizeObject)r)rrrfinalize_objectZszExecutionEngine.finalize_objectcCstj|dS)zERun static constructors which initialize module-level static objects.N)rrZLLVMPY_RunStaticConstructors)r)rrrrun_static_constructorsasz'ExecutionEngine.run_static_constructorscCstj|dS)zNRun static destructors which perform module-level cleanup of static resources.N)rrZLLVMPY_RunStaticDestructors)r)rrrrun_static_destructorsesz&ExecutionEngine.run_static_destructorsc CsFt"}tj|||r&tt|WdQRX|j|d|_dS)z1 Ownership of module is returned NF) rrrLLVMPY_RemoveModulerrr%remover)r)rrrrr remove_moduleis   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_datass    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)Zretrrrenable_jit_eventss z!ExecutionEngine.enable_jit_eventscCs6t|tj}x$|jD]}t|jtj|kr|SqWdS)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 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. Nz5object compilation notification for unknown module %s)rHcontentsrCbuf_ptrbuf_lenr rDr)r)datarCrOrPbufrrrr_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. Nz5object compilation notification for unknown module %sr) rIrNrCrDrrrLLVMPY_CreateByteStringlenrOrP)r)rQrCrrRrrr_raw_object_cache_getbuffers    z+ExecutionEngine._raw_object_cache_getbuffercCsLx|jD] }|qW|jdk r,|j|jd|_|j|dS)N)r%detachr&clearrK_capiZLLVMPY_DisposeExecutionEngine)r)modrrr_disposes     zExecutionEngine._dispose)NN)__name__ __module__ __qualname____doc__rKr(r/r1r3r7r9r:r;r>propertyr@rArDrGrMrSrVr[rrrrr,s&      rc@s eZdZdZddZddZdS)rJzN Internal: an ObjectCache instance for use within an ExecutionEngine. cCs"tjtt|}tj||dS)N)rrLLVMPY_CreateObjectCache_notify_c_hook_getbuffer_c_hookr'r()r)objr*rrrr(sz_ObjectCacheRef.__init__cCs|j|dS)N)rYLLVMPY_DisposeObjectCache)r)rrrr[sz_ObjectCacheRef._disposeN)r\r]r^r_r(r[rrrrrJsrJc@s$eZdZdejfdefdefgZdS)_ObjectCacheDatarCrOrPN)r\r]r^r LLVMModuleRefrr Z_fields_rrrrrf srf):Z __future__rrZctypesrrrrrr r r r r rrwarningsrrrrZLLVMPY_LinkInMCJITrr#r'rrJrgZLLVMTargetMachineRefrZargtypesZLLVMExecutionEngineRefZrestyper<r5Z LLVMValueRefr2r8r?ZLLVMTargetDataRefr r,r0ZLLVMObjectFileRefrFrfZ_ObjectCacheNotifyFuncZ_ObjectCacheGetBufferFuncrSrbrVrcraZLLVMObjectCacheRefrerLrTrrrrsf80