g)ddlmZmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZmZmZejjdZdZGddejZGddejZejejeegejj_ejejj_ejejeegejj_eejj_ejejgejj_ejej egejj!_ejgejj"_ejgejj#_ej$ejj#_gejj%_eejj%_ejegejj&_eejj&_ejegejj'_eejj'_ejej(gejj)_Gd 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_filectj5}tj|||}|st t | dddn #1swxYwYd|_t||S)zZ Create a MCJIT ExecutionEngine from the given *module* and *target_machine*. NT)module)r OutputStringlibLLVMPY_CreateMCJITCompiler RuntimeErrorstr_ownedExecutionEngine)rtarget_machineouterrengines V/granary/edev/quantum/lib/python3.11/site-packages/llvmlite/binding/executionengine.pycreate_mcjit_compilerr s   ,v33 NF,, ,s6{{++ + ,,,,,,,,,,,,,,,, !N 6& 1 1 11sAA!!A%(A%cntj}|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)errnos rcheck_jit_executionr$sA G 6 6 8 8E zzeG zceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ed Zd ZdZdZddZdZdZdZdS)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. Nct|g|_d|_d|_tj||dS)z; Module ownership is transferred to the EE NT)set_modules_tdrr ObjectRef__init__)selfptrrs rr,zExecutionEngine.__init__0s?VH    tS)))))r%chtj||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-names rget_function_addressz$ExecutionEngine.get_function_address9s( w00t{{77K7KLLLr%chtj||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. r0)rrLLVMPY_GetGlobalValueAddressr2r3s rget_global_value_addressz(ExecutionEngine.get_global_value_addressAs( w33D$++g:N:NOOOr%cHtj|||dSN)rrLLVMPY_AddGlobalMapping)r-gvaddrs radd_global_mappingz"ExecutionEngine.add_global_mappingIs" ''b$77777r%c||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-rs r add_modulezExecutionEngine.add_moduleMs[ T] " "@AA A   v...  &!!!!!r%cDtj|dS)z} Make sure all modules owned by the execution engine are fully processed and "usable" for execution. N)rrLLVMPY_FinalizeObjectr-s rfinalize_objectzExecutionEngine.finalize_objectWs %%d+++++r%cDtj|dS)zW Run static constructors which initialize module-level static objects. N)rrLLVMPY_RunStaticConstructorsrFs rrun_static_constructorsz'ExecutionEngine.run_static_constructors^s  ,,T22222r%cDtj|dS)zh Run static destructors which perform module-level cleanup of static resources. N)rrLLVMPY_RunStaticDestructorsrFs rrun_static_destructorsz&ExecutionEngine.run_static_destructorsds ++D11111r%ctj5}tj|||rt t | dddn #1swxYwY|j|d|_dS)z1 Ownership of module is returned NF) rrrLLVMPY_RemoveModulerrr)remover)r-rrs r remove_modulezExecutionEngine.remove_moduleks    06w**4@@ 0"3v;;/// 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 V$$$ s>AA#&A#c|j|jStj|}t j||_d|j_|jS)z; The TargetData for this execution engine. NT)r*rr#LLVMPY_GetExecutionEngineTargetDatar TargetDatar)r-r.s r target_datazExecutionEngine.target_datausJ 8 8Og99$??%c**xr%cDtj|}|S)z Enable JIT events for profiling of generated code. Return value indicates whether connection to profiling tool was successful. )rrLLVMPY_EnableJITEvents)r-rets renable_jit_eventsz!ExecutionEngine.enable_jit_eventss g,,T22 r%ct|tj}|jD])}t|jtj|kr|cS*dS)zH Find the ModuleRef corresponding to the given pointer. N)r rvaluer)_ptr)r- module_ptrr.rs r_find_module_ptrz ExecutionEngine._find_module_ptrsT:x((.m  FFK**0C77 8tr%ct|trtj|}t j||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_files radd_object_filezExecutionEngine.add_object_filesK h $ $ E"0::8DDH ))$99999r%c||_||_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_funcs rset_object_cachez ExecutionEngine.set_object_cachesG %0!'5$,T22 %%dD,>?????r%c|jdS|jj}|jj}|jj}t ||}||}|td||||dS)z( Low-level notify hook. N3object compilation notification for unknown module )rgcontentsr]buf_ptrbuf_lenr r^r)r-datar]rrrsbufrs r_raw_object_cache_notifyz(ExecutionEngine._raw_object_cache_notifys  $ , F]- -'-'))&&z22 >,:D* GHH H !!&#.....r%cX|jdS|jj}||}|t d|||}|Tt j|t||d_ t||d_ dSdS)z+ Low-level getbuffer hook. Nrpr) rhrqr]r^rrrLLVMPY_CreateByteStringlenrrrs)r-rtr]rrus r_raw_object_cache_getbufferz+ExecutionEngine._raw_object_cache_getbuffers  ' / F]- &&z22 >,:D* GHH H**622 ?!g==c3s88LLDGO!#hhDGOOO ?r%c|jD]}||j|j|jd|_|j|dSr:)r)detachr*clearrj_capiLLVMPY_DisposeExecutionEngine)r-mods r_disposezExecutionEngine._disposest=  C JJLLLL 8  HOO    ! 0066666r%)NN)__name__ __module__ __qualname____doc__rjr,r5r8r>rCrGrJrMrQpropertyrUrYr^rernrvrzrr%rrr)s?M***MMMPPP888""",,,333 222  X ::: @ @ @ @///$'''(77777r%rceZdZdZdZdZdS)rizN Internal: an ObjectCache instance for use within an ExecutionEngine. ctjtt|}tj||dSr:)rrLLVMPY_CreateObjectCache_notify_c_hook_getbuffer_c_hookr+r,)r-objr.s rr,z_ObjectCacheRef.__init__sAg..~/@/244 tS)))))r%c:|j|dSr:)r~LLVMPY_DisposeObjectCacherFs rrz_ObjectCacheRef._disposes ,,T22222r%N)rrrrr,rrr%rriris<*** 33333r%ric.eZdZdejfdefdefgZdS)_ObjectCacheDatar]rrrsN)rrrr LLVMModuleRefrr _fields_rr%rrr"s0 s() H HHHHr%rN)6ctypesrrrrrrr r r r r rllvmlite.bindingrrrrLLVMPY_LinkInMCJITrr$r+rrirLLVMTargetMachineRefrargtypesLLVMExecutionEngineRefrestyperOrA LLVMValueRefr;rErSLLVMTargetDataRefr!r1r7LLVMObjectFileRefrcr_ObjectCacheNotifyFunc_ObjectCacheGetBufferFuncrvrrzrrLLVMObjectCacheRefrrkrxrr%rrs****************************7666666666 2 2 2   p7p7p7p7p7cmp7p7p7f 3 3 3 3 3cm 3 3 3( GH/"+ .1-G"* GH($ '-#%! -0,F,/,<,4,6(+.*D)E&8+47:6K+368*35:*2 .!*-5!) 1$-08$,.!* y#4#*7+;#<#<>>%IdI&-g.>&?&?AA (',..--/11.D-F-6-8 ),/+A (.1.D-E!**-*D*-*@*B&+3',4h+?(((r%