B Ꮚ\v@sddlmZmZddlZddlZddlmZmZmZddddd d d Z ej fd d Z ej fddZ ej fddZ ddZddZddZejfddZddZGdddeZdS))print_functionabsolute_importN) instructionstypesvaluesgtlteqnegele)>=z<=csfdd}|S)Ncst|dfdd }|S)NcsH|j|jkr td|j|jf|j|j||f||}|||S)Nz,Operands must be the same type, got (%s, %s))type ValueErrorblock_insert)selflhsrhsnameflagsinstr)clsopnamer2lib/python3.7/site-packages/llvmlite/ir/builder.pywrappeds   z%_binop..wrap..wrapped)rr) functoolswraps)fnr)rrrrwrapsz_binop..wrapr)rrr#r)rrr_binops r$csfdd}|S)Ncst|dfdd }|S)Nrc s|j|jkr td|j|jf|j}t|tjs@td|ftd}|j}tt||g||g}| df|g|}|j |||g|d} | S)Nz,Operands must be the same type, got (%s, %s)z expected an integer type, got %srzllvm.%s.with.overflow)r) rr isinstancerIntType TypeErrormodule FunctionTypeZLiteralStructTypedeclare_intrinsiccall) rrrrZtyZbool_tymodZfntyr"ret)rrrr$s      z3_binop_with_overflow..wrap..wrapped)r)r r!)r"r)rrrr##sz"_binop_with_overflow..wrapr)rrr#r)rr_binop_with_overflow"s r.csfdd}|S)Ncst|dfdd }|S)Nrcs$|j|j|g|}|||S)N)rrr)roperandrr)rrrrr=s z%_uniop..wrap..wrapped)r)r r!)r"r)rrrrr#<sz_uniop..wrapr)rrr#r)rrr_uniop;s r0csfdd}|S)Ncst|dfdd }|S)Nrcs>t|jtjstd|j|j|jg}|||g|S)Nz expected an integer type, got %s)r%rrr&r'r(r*r+)rr/rr")rrrrJsz/_uniop_intrinsic..wrap..wrapped)r)r r!)r"r)rrrr#Isz_uniop_intrinsic..wrapr)rr#r)rr_uniop_intrinsicHs r1csfdd}|S)Ncst|dfdd }|S)Nrcsbt|jtjstd|j|jtdkr:td|j|j|j|jg}||||g|S)Nz expected an integer type, got %srzexpected an i1 type, got %s)r%rrr&r'r(r*r+)rr/flagrr")rrrrXs z9_uniop_intrinsic_with_flag..wrap..wrapped)r)r r!)r"r)rrrr#Wsz(_uniop_intrinsic_with_flag..wrapr)rr#r)rr_uniop_intrinsic_with_flagVs r3csfdd}|S)Ncst|dfdd }|S)Nrcs|j|jks|j|jkr2td|j|j|jfn"t|jtjtjfsTtd|j|j|j|j|jg}|||||g|S)Nz-expected types to be the same, got %s, %s, %sz'expected an floating point type, got %s) rr'r%rZ FloatTypeZ DoubleTyper(r*r+)rabcrr")rrrrhsz/_triop_intrinsic..wrap..wrapped)r)r r!)r"r)rrrr#gs z_triop_intrinsic..wrapr)rr#r)rr_triop_intrinsicfs r7csfdd}|S)Ncst|dfdd }|S)Nrcs.|j|kr|S|j|||}|||S)N)rrr)rvaltyprr)rrrrr|s   z&_castop..wrap..wrapped)r)r r!)r"r)rrrrr#{sz_castop..wrapr)rrr#r)rrr_castopzs r:cCs4t|dkr(d}d|d|d|gS||SdS)zReturns (label + suffix) or a truncated version if it's too long. Parameters ---------- label : str Label name suffix : str Label suffix 2rNz..)lenjoin)ZlabelsuffixZnheadrrr _label_suffixs r@c@seZdZdddZeddZeZeddZedd Zd d Z d d Z ddZ ddZ dddZ ejddZejddZejddZejdddZejdddZdd Zd!d"Zed#dd$d%Zed&dd'd(Zed)dd*d+Zed,dd-d.Zed/dd0d1Zed2dd3d4Zed5dd6d7Zed8dd9d:Zed;dddd?d@Z!edAddBdCZ"edDddEdFZ#edGddHdIZ$edJddKdLZ%edMddNdOZ&edPddQdRZ'edSddTdUZ(edVddWdXZ)e*dYddZd[Z+e*d\d d]d^Z,e*d_d d`daZ-e*dbd dcddZ.e*ded dfdgZ/e*dhd didjZ0ddkdlZ1ddmdnZ2dodpZ3ddqdrZ4ddsdtZ5dgfdudvZ6dgfdwdxZ7ddydzZ8e9d{dd|d}Z:e9d~dddZ;e9ddddZe9ddddZ?e9ddddZ@e9ddddZAe9ddddZBe9ddddZCe9ddddZDe9ddddZEe9ddddZFd ddZGd!ddZHd"ddZId#ddZJddZKddZLddZMddZNddZOddZPddZQddZRd$ddZSd%ddZTd&ddZUd'ddÄZVd(ddńZWd)ddDŽZXd*ddɄZYd+dd˄ZZd,dd̈́Z[d-ddτZ\d.ddфZ]d/ddӄZ^ddՄZ_d0ddׄZ`d1ddلZad2ddۄZbdd݄Zcd3dd߄ZdeedddZfeedddZgeedddZheidddZjeidddZkeldddZmdS(4 IRBuilderNcCs$||_|rt|jnd|_d|_dS)Nr)_blockr=r_anchordebug_metadata)rrrrr__init__szIRBuilder.__init__cCs|jS)z* The current basic block. )rB)rrrrrszIRBuilder.blockcCs|jjS)z' The current function. )rparent)rrrrfunctionszIRBuilder.functioncCs |jjjS)z% The current module. )rrFr()rrrrr(szIRBuilder.modulecCs|j|_|jj||_dS)z Position immediately before the given instruction. The current block is also changed to the instruction's basic block. N)rFrBrindexrC)rrrrrposition_beforeszIRBuilder.position_beforecCs |j|_|jj|d|_dS)z Position immediately after the given instruction. The current block is also changed to the instruction's basic block. rN)rFrBrrHrC)rrrrrposition_afterszIRBuilder.position_aftercCs||_d|_dS)z= Position at the start of the basic *block*. rN)rBrC)rrrrrposition_at_startszIRBuilder.position_at_startcCs||_t|j|_dS)z; Position at the end of the basic *block*. N)rBr=rrC)rrrrrposition_at_endszIRBuilder.position_at_endrcCs |j|S)z Append a basic block, with the given optional *name*, to the current function. The current block is not changed. The new block is returned. )rGappend_basic_block)rrrrrrMszIRBuilder.append_basic_blockc csF|j}|j}|dk r ||n ||z dVWd||XdS)z A context manager which temporarily positions the builder at the end of basic block *bb* (but before any terminator). N) basic_block terminatorrIrL)rrZ old_blocktermrrr goto_blocks   zIRBuilder.goto_blockc cs$||jj dVWdQRXdS)z} A context manager which temporarily positions the builder at the end of the function's entry block. N)rQrGZentry_basic_block)rrrrgoto_entry_blockszIRBuilder.goto_entry_blockccs*|||V|jjdkr&||dS)N)rLrNrObranch)rZbbenterZbbexitrrr_branch_helpers  zIRBuilder._branch_helperc cs|j}|jt|jdd}|jt|jdd}||||}|dk r^||rTddgnddg||| |VWdQRX||dS)a A context manager which sets up a conditional basic block based on the given predicate (a i1 value). If the conditional block is not explicitly terminated, a branch will be added to the next block. If *likely* is given, its boolean value indicates whether the predicate is likely to be true or not, and metadata is issued for LLVM's optimizers to account for that. z.if)rz.endifNcr)rNrMr@rcbranch set_weightsrTrL)rpredlikelybbbbifbbendbrrrrif_thens zIRBuilder.if_thenc cs|j}|jt|jdd}|jt|jdd}|jt|jdd}||||}|dk rr||rhddgnddg|||}|||} || fV||dS)a@ A context manager which sets up two conditional basic blocks based on the given predicate (a i1 value). A tuple of context managers is yield'ed. Each context manager acts as a if_then() block. *likely* has the same meaning as in if_then(). Typical use:: with builder.if_else(pred) as (then, otherwise): with then: # emit instructions for when the predicate is true with otherwise: # emit instructions for when the predicate is false z.if)rz.elsez.endifNrUr)rNrMr@rrVrWrTrL) rrXrYrZr[Zbbelser\r]ZthenZ otherwiserrrif_elses   zIRBuilder.if_elsecCsD|jdk r d|jkr |j|jd<|jj|j||jd7_dS)NZdbgr)rDZmetadatarBrinsertrC)rrrrrr2s zIRBuilder._insertcCs"|jjr t||||j_|S)N)rZ is_terminatedAssertionErrorrrO)rrPrrr_set_terminator8s  zIRBuilder._set_terminatorshlcCsdS)zC Left integer shift: name = lhs << rhs Nr)rrrrrrrrcBsz IRBuilder.shllshrcCsdS)zW Logical (unsigned) right integer shift: name = lhs >> rhs Nr)rrrrrrrrdIszIRBuilder.lshrashrcCsdS)zX Arithmetic (signed) right integer shift: name = lhs >> rhs Nr)rrrrrrrrePszIRBuilder.ashraddcCsdS)z@ Integer addition: name = lhs + rhs Nr)rrrrrrrrfWsz IRBuilder.addfaddcCsdS)zG Floating-point addition: name = lhs + rhs Nr)rrrrrrrrg^szIRBuilder.faddsubcCsdS)zC Integer subtraction: name = lhs - rhs Nr)rrrrrrrrhesz IRBuilder.subfsubcCsdS)zJ Floating-point subtraction: name = lhs - rhs Nr)rrrrrrrrilszIRBuilder.fsubmulcCsdS)zF Integer multiplication: name = lhs * rhs Nr)rrrrrrrrjssz IRBuilder.mulfmulcCsdS)zM Floating-point multiplication: name = lhs * rhs Nr)rrrrrrrrkzszIRBuilder.fmuludivcCsdS)zI Unsigned integer division: name = lhs / rhs Nr)rrrrrrrrlszIRBuilder.udivsdivcCsdS)zG Signed integer division: name = lhs / rhs Nr)rrrrrrrrmszIRBuilder.sdivfdivcCsdS)zG Floating-point division: name = lhs / rhs Nr)rrrrrrrrnszIRBuilder.fdivuremcCsdS)zJ Unsigned integer remainder: name = lhs % rhs Nr)rrrrrrrroszIRBuilder.uremsremcCsdS)zH Signed integer remainder: name = lhs % rhs Nr)rrrrrrrrpszIRBuilder.sremfremcCsdS)zH Floating-point remainder: name = lhs % rhs Nr)rrrrrrrrqszIRBuilder.fremorcCsdS)zB Bitwise integer OR: name = lhs | rhs Nr)rrrrrrror_sz IRBuilder.or_andcCsdS)zC Bitwise integer AND: name = lhs & rhs Nr)rrrrrrrand_szIRBuilder.and_xorcCsdS)zC Bitwise integer XOR: name = lhs ^ rhs Nr)rrrrrrrrvsz IRBuilder.xorZsaddcCsdS)zn Signed integer addition with overflow: name = {result, overflow bit} = lhs + rhs Nr)rrrrrrrsadd_with_overflowszIRBuilder.sadd_with_overflowZsmulcCsdS)zt Signed integer multiplication with overflow: name = {result, overflow bit} = lhs * rhs Nr)rrrrrrrsmul_with_overflowszIRBuilder.smul_with_overflowZssubcCsdS)zq Signed integer subtraction with overflow: name = {result, overflow bit} = lhs - rhs Nr)rrrrrrrssub_with_overflowszIRBuilder.ssub_with_overflowZuaddcCsdS)zp Unsigned integer addition with overflow: name = {result, overflow bit} = lhs + rhs Nr)rrrrrrruadd_with_overflowszIRBuilder.uadd_with_overflowZumulcCsdS)zv Unsigned integer multiplication with overflow: name = {result, overflow bit} = lhs * rhs Nr)rrrrrrrumul_with_overflowszIRBuilder.umul_with_overflowZusubcCsdS)zs Unsigned integer subtraction with overflow: name = {result, overflow bit} = lhs - rhs Nr)rrrrrrrusub_with_overflowszIRBuilder.usub_with_overflowcCsDt|jtjr&t|jd|jj}nt|jd}|j|||dS)zG Bitwise integer complement: name = ~value )r})r)r%rrZ VectorTyperConstantcountrv)rvaluerrrrrnot_szIRBuilder.not_cCs|jt|jd||dS)z= Integer negative: name = -value r)r)rhrr~r)rrrrrrnegsz IRBuilder.negcCsdy t|}Wn"tk r.td|fYnX|dkr@||}tj|j||||d}|||S)Nzinvalid comparison %r for icmp)z==z!=)r)_CMP_MAPKeyErrorrrZ ICMPInstrrr)rprefixcmpoprrroprrrr_icmps  zIRBuilder._icmpcCs|d||||S)z Signed integer comparison: name = lhs rhs where cmpop can be '==', '!=', '<', '<=', '>', '>=' s)r)rrrrrrrr icmp_signedszIRBuilder.icmp_signedcCs|d||||S)z Unsigned integer (or pointer) comparison: name = lhs rhs where cmpop can be '==', '!=', '<', '<=', '>', '>=' u)r)rrrrrrrr icmp_unsignedszIRBuilder.icmp_unsignedcCs@|tkrdt|}n|}tj|j|||||d}|||S)z Floating-point ordered comparison: name = lhs rhs where cmpop can be '==', '!=', '<', '<=', '>', '>=', 'ord', 'uno' o)rr)rr FCMPInstrrr)rrrrrrrrrrr fcmp_ordered!s  zIRBuilder.fcmp_orderedcCs@|tkrdt|}n|}tj|j|||||d}|||S)z Floating-point unordered comparison: name = lhs rhs where cmpop can be '==', '!=', '<', '<=', '>', '>=', 'ord', 'uno' r)rr)rrrrr)rrrrrrrrrrrfcmp_unordered0s  zIRBuilder.fcmp_unorderedcCs$tj|j||||d}|||S)zN Ternary select operator: name = cond ? lhs : rhs )r)rZ SelectInstrrr)rcondrrrrrrrselect?s zIRBuilder.selecttrunccCsdS)z_ Truncating integer downcast to a smaller type: name = (typ) value Nr)rrr9rrrrrLszIRBuilder.trunczextcCsdS)z` Zero-extending integer upcast to a larger type: name = (typ) value Nr)rrr9rrrrrSszIRBuilder.zextsextcCsdS)z` Sign-extending integer upcast to a larger type: name = (typ) value Nr)rrr9rrrrrZszIRBuilder.sextfptrunccCsdS)z` Floating-point downcast to a less precise type: name = (typ) value Nr)rrr9rrrrraszIRBuilder.fptruncfpextcCsdS)z^ Floating-point upcast to a more precise type: name = (typ) value Nr)rrr9rrrrrhszIRBuilder.fpextbitcastcCsdS)zZ Pointer cast to a different pointer type: name = (typ) value Nr)rrr9rrrrroszIRBuilder.bitcast addrspacecastcCsdS)z[ Pointer cast to a different address space: name = (typ) value Nr)rrr9rrrrrvszIRBuilder.addrspacecastfptouicCsdS)z\ Convert floating-point to unsigned integer: name = (typ) value Nr)rrr9rrrrr}szIRBuilder.fptouiuitofpcCsdS)z\ Convert unsigned integer to floating-point: name = (typ) value Nr)rrr9rrrrrszIRBuilder.uitofpfptosicCsdS)zZ Convert floating-point to signed integer: name = (typ) value Nr)rrr9rrrrrszIRBuilder.fptosisitofpcCsdS)zZ Convert signed integer to floating-point: name = (typ) value Nr)rrr9rrrrrszIRBuilder.sitofpptrtointcCsdS)zI Cast pointer to integer: name = (typ) value Nr)rrr9rrrrrszIRBuilder.ptrtointinttoptrcCsdS)zI Cast integer to pointer: name = (typ) value Nr)rrr9rrrrrszIRBuilder.inttoptrcCsb|dkr n8t|tjtjfr0t|jtjsBtnttd|}t |j |||}| ||S)zl Stack-allocate a slot for *size* elements of the given type. (default one element) N ) r%rZValuer~rrr&rarZ AllocaInstrrr)rr9sizerZalrrrallocas zIRBuilder.allocacCsHt|jtjs$td|jt|ft|j||}||_ | ||S)zf Load value from pointer, with optional guaranteed alignment: name = *ptr z5cannot load from value of type %s (%r): not a pointer) r%rr PointerTyper'strrZ LoadInstrralignr)rptrrrldrrrloads zIRBuilder.loadcCsjt|jtjs$td|jt|f|jj|jkrFtd|j|jft|j ||}||_ | ||S)ze Store value to pointer, with optional guaranteed alignment: *ptr = name z4cannot store to value of type %s (%r): not a pointerz(cannot store %s to %s: mismatching types) r%rrrr'rpointeerZ StoreInstrrrr)rrrrstrrrstores zIRBuilder.storecCsFt|jtjs$td|jt|ft|j||||}| ||S)zf Load value from pointer, with optional guaranteed alignment: name = *ptr z5cannot load from value of type %s (%r): not a pointer) r%rrrr'rrZLoadAtomicInstrrr)rrorderingrrrrrr load_atomics  zIRBuilder.load_atomiccCsht|jtjs$td|jt|f|jj|jkrFtd|j|jft|j ||||}| ||S)ze Store value to pointer, with optional guaranteed alignment: *ptr = name z4cannot store to value of type %s (%r): not a pointerz(cannot store %s to %s: mismatching types) r%rrrr'rrrZStoreAtomicInstrrr)rrrrrrrrr store_atomics zIRBuilder.store_atomiccCs t|jd||}|||S)zF Create a switch-case with a single *default* target. switch)rZ SwitchInstrrrb)rrdefaultZswtrrrrs zIRBuilder.switchcCs t|jd|g}|||S)z3 Unconditional branch to *target*. r])rBranchrrb)rtargetr]rrrrSs zIRBuilder.branchcCs$t|jd|||g}|||S)zV Conditional branch to *truebr* if *cond* is true, else to *falsebr*. r])rZConditionalBranchrrb)rrZtruebrZfalsebrr]rrrrV s   zIRBuilder.cbranchcCst|jd|}|||S)z3 Indirect branch to target *addr*. Z indirectbr)rZIndirectBranchrrb)rZaddrr]rrrbranch_indirects zIRBuilder.branch_indirectcCs|t|jdS)z7 Return from function without a value. zret void)rbrRetr)rrrrret_voidszIRBuilder.ret_voidcCs|t|jd|S)z> Return from function with the given *value*. r-)rbrrr)rrrrrr-"sz IRBuilder.retcCs t|jd|g}|||S)z0 Resume an in-flight exception. resume)rrrrb)r landingpadr]rrrr)s zIRBuilder.resumeFrc Cs(tj|j||||||d}|||S)zP Call function *fn* with *args*: name = fn(args...) )rcconvtailfastmath)rZ CallInstrrr)rr"argsrrrrinstrrrr+3s  zIRBuilder.callcCst||||}||||S)z# Inline assembler. )rZ InlineAsmr+)rftypeasmZ constraintrZ side_effectrrrrr=sz IRBuilder.asmcCs$t|g}||dd|gd|S)zo Load a register value into an LLVM value. Example: v = load_reg(IntType(32), "eax") rz={%s}F)rr)r)rreg_typereg_namerrrrrload_regDs zIRBuilder.load_regcCs,tt|g}||dd||gd|S)z Store an LLVM value inside a register Example: store_reg(Constant(IntType(32), 0xAAAAAAAA), IntType(32), "eax") rz{%s}T)rr)ZVoidTyper)rrrrrrrrr store_regLszIRBuilder.store_regc Cs(tj|j||||||d}|||S)N)rr)rZ InvokeInstrrrb) rr"rZ normal_toZ unwind_torrrrrrrinvokeTs zIRBuilder.invokecCs$tj|j||||d}|||S)zo Compute effective address (getelementptr): name = getelementptr ptr, )inboundsr)rZGEPInstrrr)rrindicesrrrrrrgep\s   z IRBuilder.gepcCs"tj|j|||d}|||S)z4 Returns the value at position idx. )r)rZExtractElementrr)rvectoridxrrrrrextract_elemenths zIRBuilder.extract_elementcCs$tj|j||||d}|||S)z Returns vector with vector[idx] replaced by value. The result is undefined if the idx is larger or equal the vector length. )r)rZ InsertElementrr)rrrrrrrrrinsert_elementps zIRBuilder.insert_elementcCs$tj|j||||d}|||S)a Constructs a permutation of elements from *vector1* and *vector2*. Returns a new vector in the same length of *mask*. * *vector1* and *vector2* must have the same element type. * *mask* must be a constant vector of integer types. )r)rZ ShuffleVectorrr)rZvector1Zvector2maskrrrrrshuffle_vectorzs zIRBuilder.shuffle_vectorcCs6t|ttfs|g}tj|j|||d}|||S)z= Extract member number *idx* from aggregate. )r)r%tuplelistrZ ExtractValuerr)raggrrrrrr extract_values  zIRBuilder.extract_valuecCs8t|ttfs|g}tj|j||||d}|||S)zI Insert *value* into member number *idx* from aggregate. )r)r%rrrZ InsertValuerr)rrrrrrrrr insert_values  zIRBuilder.insert_valuecCs tj|j||d}|||S)N)r)rZPhiInstrrr)rr9rrrrrphis z IRBuilder.phicCst|j}|||S)N)rZ Unreachablerrb)rrrrr unreachables  zIRBuilder.unreachablecCs&tj|j|||||d}|||S)N)r)rZ AtomicRMWrr)rrrr8rrrrrr atomic_rmws zIRBuilder.atomic_rmwc Cs8|dkr |n|}tj|j||||||d}|||S)a8 Atomic compared-and-set: atomic { old = *ptr success = (old == cmp) if (success) *ptr = val } name = { old, success } If failordering is `None`, the value of `ordering` is used. N)r)rZCmpXchgrr)rrZcmpr8rZ failorderingrrrrrcmpxchgs   zIRBuilder.cmpxchgcCs t|j|||}|||S)N)rZLandingPadInstrrr)rr9rZcleanuprrrrrs zIRBuilder.landingpadcCs|jd}|||gS)z? Optimizer hint: assume *cond* is always true. z llvm.assume)r(r*r+)rrr"rrrassumes zIRBuilder.assumecCs"tj|j|||d}|||S)z Add a memory barrier, preventing certain reorderings of load and/or store accesses with respect to other processors and devices. )r)rZFencerr)rrZ targetscoperrrrrfences zIRBuilder.fencez llvm.bswapcCsdS)zn Used to byte swap integer values with an even number of bytes (positive multiple of 16 bits) Nr)rrrrrbswapszIRBuilder.bswapzllvm.bitreversecCsdS)zh Reverse the bitpattern of an integer value; for example 0b10110110 becomes 0b01101101. Nr)rrrrr bitreverseszIRBuilder.bitreversez llvm.ctpopcCsdS)z; Counts the number of bits set in a value. Nr)rrrrrctpopszIRBuilder.ctpopz llvm.ctlzcCsdS)zC Counts the number of leading zeros in a variable. Nr)rrr2rrrctlzszIRBuilder.ctlzz llvm.cttzcCsdS)zD Counts the number of trailing zeros in a variable. Nr)rrr2rrrcttzszIRBuilder.cttzzllvm.fmacCsdS)z; Perform the fused multiply-add operation. Nr)rr4r5r6rrrfmasz IRBuilder.fma)N)r)N)N)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)r)Nr)rN)N)r)rNFr)r)r)r)rNF)Fr)r)r)r)r)r)r)r)Nr)rF)Nr)n__name__ __module__ __qualname__rEpropertyrrNrGr(rIrJrKrLrM contextlibcontextmanagerrQrRrTr^r_rrbr$rcrdrerfrgrhrirjrkrlrmrnrorprqrsrurvr.rwrxryrzr{r|rrrrrrrrr:rrrrrrrrrrrrrrrrrrrrSrVrrr-rr+rrrrrrrrrrrrrrrrrr1rrrr3rrr7rrrrrrAs                                     rA)Z __future__rrrr rrrrrZ Instructionr$r.r0r1r3r7Z CastInstrr:r@objectrArrrrs$