U md9 @sXddlZddlmZmZmZmZmZmZmZddl m Z ddl m Z m Z ddZGdddeZd d Zd d Zd dZddddZd(ddZddZGddde jZeddddgZeddddd d!gZGd"d#d#e jZGd$d%d%e jZd&d'Zeege jj _!eege jj"_!ee jj"_#eege jj$_!eege jj%_!ege jj&_!ee jj&_#ege jj'_!e j(e jj'_#e j(eege jj)_!e j(ge jj*_!e j(e j+ge jj,_!ee jj,_#e j(e j+ege jj-_!ee jj-_#e j(e j+ge jj._!ee jj._#e j(e j+ge jj/_!ee jj/_#eeege jj0_!e j1e jj0_#e j1ge jj2_!ee jj2_#e j1ge jj3_!ee jj3_#e j1eeeeeeeeeg e jj4_!e j5e jj4_#e j5ge jj6_!e j5eege jj7_!e j5ege jj8_!e j5e j9ge jj:_!e j5e j;eeege jj<_!e j=e jj<_#e j=ge jj>_!ee jj>_#e j=ge jj?_!ee jj?_#e j=ge jj@_!e j5ge jjA_!e j(e jjA_#ge jjB_!ee jjB_#dS))N)POINTERc_char_p c_longlongc_intc_size_tc_void_p string_at)ffi)_decode_string_encode_stringc Cs4t"}tj|t|W5QRSQRXdS)a Return a target triple suitable for generating code for the current process. An example when the default triple from ``get_default_triple()`` is not be suitable is when LLVM is compiled for 32-bit but the process is executing in 64-bit mode. N)r OutputStringlibLLVMPY_GetProcessTriplestroutrQ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/llvmlite/binding/targets.pyget_process_triple s  rc@seZdZdZdddZdS) FeatureMapz Maps feature name to a boolean indicating the availability of the feature. Extends ``dict`` to add `.flatten()` method. Tcs>|rt|n t|}ddddfdd|DS)ap Args ---- sort: bool Optional. If True, the features are sorted by name; otherwise, the ordering is unstable between python session due to hash randomization. Defaults to True. Returns a string suitable for use as the ``features`` argument to ``Target.create_target_machine()``. +-)TF,c3s"|]\}}d||VqdS)z{0}{1}N)format).0kvflag_maprr *sz%FeatureMap.flatten..)sorteditemsiterjoin)selfsortiteratorrrrflattens  zFeatureMap.flattenN)T)__name__ __module__ __qualname____doc__r'rrrrrsrc Cstv}t}tj|s,|W5QRSddd}t|}|rn|dD] }|rL||d||dd<qL|W5QRSQRXdS)ac Returns a dictionary-like object indicating the CPU features for current architecture and whether they are enabled for this CPU. The key-value pairs are the feature name as string and a boolean indicating whether the feature is available. The returned value is an instance of ``FeatureMap`` class, which adds a new method ``.flatten()`` for returning a string suitable for use as the "features" argument to ``Target.create_target_machine()``. If LLVM has not implemented this feature or it fails to get the information, this function will raise a RuntimeError exception. TF)rrrrN)r r rr LLVMPY_GetHostCPUFeaturesrsplit)rZoutdictrcontentZfeatrrrget_host_cpu_features.s   r0c Cs4t"}tj|t|W5QRSQRXdS)zR Return the default target triple LLVM is configured to produce code for. N)r r r LLVMPY_GetDefaultTargetTriplerrrrrget_default_tripleGs  r2c Cs4t"}tj|t|W5QRSQRXdS)zm Get the name of the host's CPU, suitable for using with :meth:`Target.create_target_machine()`. N)r r r LLVMPY_GetHostCPUNamerrrrrget_host_cpu_namePs  r4COFFELFZMachO)r,cCs&|dkrt}tjt|}t|S)z~ Get the object format for the given *triple* string (or the default triple if omitted). A string is returned N)r2r r LLVMPY_GetTripleObjectFormatr _object_formats)tripleresrrrget_object_formatasr=cCsttjt|S)zE Create a TargetData instance for the given *layout* string. ) TargetDatar r LLVMPY_CreateTargetDatar )Zlayoutrrrcreate_target_datamsr@c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)r>z{ A TargetData provides structured access to a data layout. Use :func:`create_target_data` to create instances. c Cs@|jr dSt$}tj||t|W5QRSQRXdS)Nz)_closedr r r LLVMPY_CopyStringRepOfTargetDatarr$rrrr__str__zs  zTargetData.__str__cCs|j|dSN)_capiLLVMPY_DisposeTargetDatar$rrr_disposeszTargetData._disposecCstj||S)z1 Get ABI size of LLVM type *ty*. )r r LLVMPY_ABISizeOfType)r$tyrrr get_abi_sizeszTargetData.get_abi_sizecCs0tj|||}|dkr,td|t||S)zL Get byte offset of type's ty element at the given position zQCould not determined offset of {}th element of the type '{}'. Is it a structtype?)r r LLVMPY_OffsetOfElement ValueErrorrr)r$rKpositionoffsetrrrget_element_offsetszTargetData.get_element_offsetcCs(tj||}|dkr$td|f|S)zI Get ABI size of pointee type of LLVM pointer type *ty*. rMNot a pointer type: %s)r r LLVMPY_ABISizeOfElementType RuntimeErrorr$rKsizerrrget_pointee_abi_sizeszTargetData.get_pointee_abi_sizecCs(tj||}|dkr$td|f|S)zV Get minimum ABI alignment of pointee type of LLVM pointer type *ty*. rMrS)r r LLVMPY_ABIAlignmentOfElementTyperUrVrrrget_pointee_abi_alignmentsz$TargetData.get_pointee_abi_alignmentN) r(r)r*r+rDrIrLrRrXrZrrrrr>ts  r>defaultZstaticZpicZ dynamicnopic jitdefaultZsmallZkernelZmediumZlargec@s^eZdZdZeddZeddZeddZedd Z ed d Z d d Z dddZ dS)TargetcCst}||S)zB Create a Target instance for the default triple. )r2 from_triple)clsr;rrrfrom_default_tripleszTarget.from_default_triplec CsVtD}tj|d|}|s.tt|||}||_|W5QRSQRXdS)zK Create a Target instance for the given triple (a string). utf8N)r r r LLVMPY_GetTargetFromTripleencoderUr_triple)r`r;outerrtargetrrrr_s  zTarget.from_triplecCstj|}t|SrE)r r LLVMPY_GetTargetNamer r$srrrnames z Target.namecCstj|}t|SrE)r r LLVMPY_GetTargetDescriptionr rirrr descriptions zTarget.descriptioncCs|jSrE)rerHrrrr;sz Target.triplecCsd|j|jS)Nz)rrkrmrHrrrrDszTarget.__str__r7r[r\Fc Csd|krdksnt|tks&t|tks2t|j} tjdkrR|dkrR| d7} tj|t | t |t ||t |t |t |t |t | } | rt | St ddS)am Create a new TargetMachine for this target and the given options. Specifying codemodel='default' will result in the use of the "small" code model. Specifying codemodel='jitdefault' will result in the code model being picked based on platform bitness (32="small", 64="large"). The `printmc` option corresponds to llvm's `-print-machineinstrs`. The `jit` option should be set when the target-machine is to be used in a JIT engine. The `abiname` option specifies the ABI. RISC-V targets with hard-float needs to pass the ABI name to LLVM. rr8ntr\z-elfzCannot create target machineN) AssertionErrorRELOC CODEMODELreosrkr r LLVMPY_CreateTargetMachiner int TargetMachinerU) r$cpufeaturesoptZrelocZ codemodelZprintmcZjitZabinamer;tmrrrcreate_target_machines(   zTarget.create_target_machineN)r^r^r7r[r\FFr^) r(r)r*re classmethodrar_propertyrkrmr;rDrzrrrrr]s(     r]c@sVeZdZddZddZddZddZd d Zdd d Ze ddZ e ddZ dS)rucCs|j|dSrE)rFLLVMPY_DisposeTargetMachinerHrrrrI szTargetMachine._disposecCstj||dS)zW Register analysis passes for this target machine with a pass manager. N)r r LLVMPY_AddAnalysisPasses)r$pmrrradd_analysis_passes sz!TargetMachine.add_analysis_passescCstj||dS)z Set whether this target machine will emit assembly with human-readable comments describing control flow, debug information, and so on. N)r r #LLVMPY_SetTargetMachineAsmVerbosity)r$verboserrrset_asm_verbosityszTargetMachine.set_asm_verbositycCs|j|ddS)z Represent the module as a code object, suitable for use with the platform's linker. Returns a byte string. T use_object)_emit_to_memoryr$modulerrr emit_objectszTargetMachine.emit_objectcCst|j|ddS)z Return the raw assembler of the module, as a string. llvm.initialize_native_asmprinter() must have been called first. Fr)r rrrrr emit_assembly szTargetMachine.emit_assemblyFc Cstt,}tj||t||}|s0tt|W5QRXtj|}tj|}zt ||WStj |XdS)zReturns bytes of object code of the module. Args ---- use_object : bool Emit object code or (if False) emit assembly code. N) r r r LLVMPY_TargetMachineEmitToMemoryrtrUrLLVMPY_GetBufferStartLLVMPY_GetBufferSizeLLVMPY_DisposeMemoryBufferr)r$rrrfmbZbufptrZbufszrrrr(s    zTargetMachine._emit_to_memorycCsttj|SrE)r>r r LLVMPY_CreateTargetMachineDatarHrrr target_data>szTargetMachine.target_datac Cs6t$}tj||t|W5QRSQRXdSrE)r r r LLVMPY_GetTargetMachineTriplerrCrrrr;Bs zTargetMachine.tripleN)F) r(r)r*rIrrrrrr|rr;rrrrrus  rucCstjdkrdSdSdS)zG Returns True if SVML was enabled at FFI support compile time. rFTN)r r LLVMPY_HasSVMLSupportrrrrhas_svmlIsr)N)CrrctypesrrrrrrrZllvmlite.bindingr Zllvmlite.binding.commonr r rdictrr0r2r4r:r=r@Z ObjectRefr> frozensetrprqr]rurr rZargtypesr-restyper1r3r9r?ZLLVMTargetDataRefrBrGZ LLVMTypeRefrJrNrTrYrcZ LLVMTargetRefrhrlrsZLLVMTargetMachineRefr}rrZLLVMPassManagerRefr~Z LLVMModuleRefrZLLVMMemoryBufferRefrrrrrrrrrs$     5 YB