U
    +‰d ã                   @   sp  d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl	m
Z
 ddl	mZ e
 d¡Ze
 d¡Ze
 d¡Ze
 d	¡Ze
 d
¡Ze
 ¡ Ze
 ¡ Ze
 ¡ ZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZ dd„ Z!G dd„ deƒZ"G dd „ d eƒZ#G d!d"„ d"eƒZ$e%d#krle &¡  dS )$z
IR Construction Tests
é    Né   )ÚTestCase)Úir)Úbindingé   é   é    é@   c                   @   s„   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zd!dd„Z	d"dd„Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ ZdS )#ÚTestBasez!
    Utilities for IR tests.
    c                 C   s6   dd„ }d  t||ƒ¡}t dd|¡}|  ||¡ dS )zr
        Assert *pattern* is in *text*, ignoring any whitespace differences
        (including newlines).
        c                 S   s   |   ¡ s|  ¡ sd|  S | S )Nú\)ÚisalnumÚisspace)Úc© r   ú:/tmp/pip-unpacked-wheel-stw2luzp/llvmlite/tests/test_ir.pyÚescape&   s    z%TestBase.assertInText.<locals>.escapeÚ z\s+z\\s*N)ÚjoinÚmapÚreÚsubÚassertRegex)ÚselfÚpatternÚtextr   Úregexr   r   r   ÚassertInText    s    zTestBase.assertInTextc                 C   s&   dd„ t |ƒ ¡ D ƒ}|  ||¡ d S )Nc                 S   s   g | ]}|  ¡ ‘qS r   )Ústrip)Ú.0Úliner   r   r   Ú
<listcomp>0   s     z+TestBase.assert_ir_line.<locals>.<listcomp>)ÚstrÚ
splitlinesÚassertIn)r   r   ÚmodÚlinesr   r   r   Úassert_ir_line/   s    zTestBase.assert_ir_linec                 C   s   t  t|ƒ¡ d S ©N)ÚllvmÚparse_assemblyr!   ©r   r$   r   r   r   Úassert_valid_ir3   s    zTestBase.assert_valid_irc                 C   s<   t  t j|dd¡}|  |j|j¡ |  t|ƒt|ƒ¡ |S )zAssert that the IR object pickles and unpickles correctly.
        The IR string is equal and that their type is equal
        éÿÿÿÿ)Úprotocol)ÚpickleÚloadsÚdumpsÚassertIsÚ	__class__ÚassertEqualr!   )r   ZirobjectZ	newobjectr   r   r   Úassert_pickle_correctly6   s    z TestBase.assert_pickle_correctlyc                 C   s   t  ¡ S r'   )r   ÚModule©r   r   r   r   Úmodule?   s    zTestBase.moduleNÚmy_funcc              	   C   s4   |p
|   ¡ }t ttttt t¡f¡}t |||¡S r'   )r7   r   ÚFunctionTypeÚint32ÚdblÚPointerTypeÚFunction)r   r7   ÚnameZfntyr   r   r   ÚfunctionB   s
    ÿzTestBase.functionr   c                 C   s   |p
|   ¡ }| |¡S r'   )r?   Úappend_basic_block)r   Úfuncr>   r   r   r   ÚblockH   s    zTestBase.blockc                 C   s   g }|  |¡ d |¡S )Nr   )Údescrr   )r   ÚthingÚbufr   r   r   rC   L   s    
zTestBase.descrc                 C   s   t  |¡}| dd¡}|S )Nz
    z
  )ÚtextwrapÚdedentÚreplace)r   Úasmr   r   r   Ú_normalize_asmQ   s    
zTestBase._normalize_asmc                 C   s   |   |¡}|  ||¡ d S r'   )rJ   r3   )r   rC   rI   Úexpectedr   r   r   Úcheck_descrW   s    
zTestBase.check_descrc                 C   s   |   |  |¡|¡ d S r'   )rL   rC   )r   rB   rI   r   r   r   Úcheck_block[   s    zTestBase.check_blockc                 C   s*   |   |¡}| ¡ }|  | ¡ | ¡ ¡ d S r'   )rJ   Z_stringify_bodyr3   r   ©r   r7   rI   rK   Úactualr   r   r   Úcheck_module_body^   s    
zTestBase.check_module_bodyc                 C   s*   |   |¡}| ¡ }|  | ¡ | ¡ ¡ dS )z6
        Check module metadata against *asm*.
        N)rJ   Z_stringify_metadatar3   r   rN   r   r   r   Úcheck_metadatac   s    
zTestBase.check_metadatac                 C   sD   |   |¡}|  |¡}| d¡d  d¡d }|  | ¡ | ¡ ¡ d S )NÚ{é   Ú}r   )rJ   rC   Ú	partitionÚ
rpartitionr3   r   )r   rA   rI   rK   rO   r   r   r   Úcheck_func_bodyk   s    

zTestBase.check_func_body)Nr8   )Nr   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r&   r+   r4   r7   r?   rB   rC   rJ   rL   rM   rP   rQ   rW   r   r   r   r   r
      s   	

r
   c                   @   sˆ   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd S )!ÚTestFunctionz>i32 @"my_func"(i32 %".1", i32 %".2", double %".3", i32* %".4")c                 C   s0   |   ¡ }|  |¡ ¡ }|  | ¡ d| j ¡ d S )Nz
declare %s)r?   rC   r   r3   Úproto©r   rA   rI   r   r   r   Útest_declarev   s    zTestFunction.test_declarec                 C   s€   |   ¡ }|j d¡ |j d¡ d|j_tjtddd}t |  ¡ |d¡}||j_	|  
|¡ ¡ }|  |d| j ¡ |  |¡ d S )	NÚoptsizeÚalwaysinliner   r   T©Zvar_argZ__gxx_personality_v0z\declare %s alwaysinline optsize alignstack(16) personality i8 (...)* @"__gxx_personality_v0")r?   Ú
attributesÚaddZ
alignstackr   r9   Úint8r=   r7   ZpersonalityrC   r   r3   r]   r4   )r   rA   Ztp_persZpersrI   r   r   r   Útest_declare_attributes|   s    þÿz$TestFunction.test_declare_attributesc                 C   s†   |   ¡ }|jd  d¡ d|jd j_d|jd j_d|jd j_|jd  d¡ |j d	¡ |  |¡ 	¡ }|  
|d
¡ |  |¡ d S )Nr   Zzeroexté   r   é
   é   é   ZnonnullÚnoaliasz•declare noalias i32 @"my_func"(i32 zeroext %".1", i32 dereferenceable(5) dereferenceable_or_null(10) %".2", double %".3", i32* nonnull align 4 %".4"))r?   ÚargsÚadd_attributerc   ZdereferenceableZdereferenceable_or_nullÚalignZreturn_valuerC   r   r3   r4   r^   r   r   r   Útest_function_attributes   s    ÿz%TestFunction.test_function_attributesc                 C   sV   |   ¡ }|  |¡}| d| g ¡¡ |  |¡ ¡ }|  |d| j› d¡ |  |¡ d S )NÚdbgúdeclare z !dbg !0)	r7   r?   Úset_metadataÚadd_metadatarC   r   r3   r]   r4   ©r   r7   rA   rI   r   r   r   Útest_function_metadata   s    
ÿz#TestFunction.test_function_metadatac                 C   s@   |   ¡ }d|_|  |¡ ¡ }|  |d| j› d¡ |  |¡ d S )NÚ	a_sectionrq   z section "a_section")r?   ÚsectionrC   r   r3   r]   r4   r^   r   r   r   Útest_function_section©   s    ÿz"TestFunction.test_function_sectionc                 C   s\   |   ¡ }|  |¡}d|_| d| g ¡¡ |  |¡ ¡ }|  |d| j› d¡ |  	|¡ d S )Nrv   rp   rq   z section "a_section" !dbg !0)
r7   r?   rw   rr   rs   rC   r   r3   r]   r4   rt   r   r   r   Útest_function_section_meta´   s    
ÿz'TestFunction.test_function_section_metac                 C   sb   |   ¡ }|  |¡}|j d¡ | d| g ¡¡ |  |¡ ¡ }|  |d| j	› d¡ |  
|¡ d S )Nra   rp   rq   z alwaysinline !dbg !0)r7   r?   rc   rd   rr   rs   rC   r   r3   r]   r4   rt   r   r   r   Útest_function_attr_metaÁ   s    
ÿz$TestFunction.test_function_attr_metac                 C   sL   |   ¡ }|j d¡ d|_|  |¡ ¡ }|  |d| j› d¡ |  |¡ d S )Nr`   rv   rq   z optsize section "a_section")	r?   rc   rd   rw   rC   r   r3   r]   r4   r^   r   r   r   Útest_function_attr_sectionÎ   s    ÿz'TestFunction.test_function_attr_sectionc                 C   sh   |   ¡ }|  |¡}|j d¡ d|_| d| g ¡¡ |  |¡ ¡ }|  	|d| j
› d¡ |  |¡ d S )Nra   rv   rp   rq   z) alwaysinline section "a_section" !dbg !0)r7   r?   rc   rd   rw   rr   rs   rC   r   r3   r]   r4   rt   r   r   r   Útest_function_attr_section_metaÙ   s    
ÿz,TestFunction.test_function_attr_section_metac                 C   sT   |   ¡ }|j d¡ | d¡}t |¡}| ¡  |  |¡}|  |dj	| j
d¡ d S )Nra   Úmy_blockz            define {proto} alwaysinline
            {{
            my_block:
                ret void
            }}
            )r]   )r?   rc   rd   r@   r   Ú	IRBuilderÚret_voidrC   rL   Úformatr]   )r   rA   rB   ÚbuilderrI   r   r   r   Útest_defineç   s    



úzTestFunction.test_definec                 C   s¦   |   ¡ }t ¡ }| dtg¡}| d|tg¡}| d||tg¡}| d¡}|  |  |¡ ¡ d¡ |  |  |¡ ¡ d¡ |  |  |¡ ¡ d¡ |  |  |¡ ¡ d¡ d S )	Nú	llvm.powizllvm.memsetzllvm.memcpyzllvm.assumezD            declare double @"llvm.powi.f64"(double %".1", i32 %".2")zZ            declare void @"llvm.memset.p0i8.i32"(i8* %".1", i8 %".2", i32 %".3", i1 %".4")z`            declare void @"llvm.memcpy.p0i8.p0i8.i32"(i8* %".1", i8* %".2", i32 %".3", i1 %".4")z1            declare void @"llvm.assume"(i1 %".1"))	r7   re   Ú
as_pointerÚdeclare_intrinsicr;   r:   rL   rC   r   )r   r7   Zpint8ÚpowiÚmemsetZmemcpyÚassumer   r   r   Útest_declare_intrinsics÷   s    
z$TestFunction.test_declare_intrinsicsc                 C   s4   |   ¡ }| dtg¡}| dtg¡}|  ||¡ d S )Nrƒ   )r7   r…   r;   r1   )r   r7   r†   Zpowi2r   r   r   Útest_redeclare_intrinsic  s    z%TestFunction.test_redeclare_intrinsicc                 C   s   |   ¡ }|  |¡ d S r'   )r?   r4   )r   Úfnr   r   r   Útest_pickling  s    zTestFunction.test_picklingc              	   C   sJ   |   ¡ }|  |¡}|j d¡ d}|  t|¡ |j d¡ W 5 Q R X d S )Nra   ú$Can't have alwaysinline and noinlineÚnoinline©r7   r?   rc   rd   ÚassertRaisesRegexÚ
ValueError©r   r7   rA   Úmsgr   r   r   Ú%test_alwaysinline_noinline_disallowed  s    
z2TestFunction.test_alwaysinline_noinline_disallowedc              	   C   sJ   |   ¡ }|  |¡}|j d¡ d}|  t|¡ |j d¡ W 5 Q R X d S )NrŽ   r   ra   r   r’   r   r   r   Ú%test_noinline_alwaysinline_disallowed  s    
z2TestFunction.test_noinline_alwaysinline_disallowedN)rX   rY   rZ   r]   r_   rf   ro   ru   rx   ry   rz   r{   r|   r‚   r‰   rŠ   rŒ   r”   r•   r   r   r   r   r\   r   s    	r\   c                   @   s„   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS ) ÚTestIRc                 C   s8   |   ¡ }| tdƒtdƒg¡ |  d|¡ |  |¡ d S )Né{   é*   z!0 = !{ i32 123, i8 42 })r7   rs   r:   re   r&   r+   r*   r   r   r   Útest_unnamed_metadata'  s    zTestIR.test_unnamed_metadatac                 C   s‚   |   ¡ }| tdƒdg¡}| tdƒ|g¡}| tdƒ|g¡}|  ||¡ | |||g¡ |  d|¡ |  d|¡ |  d|¡ d S )Nr—   ÚkerneléÈ  z!0 = !{ i32 123, !"kernel" }z!1 = !{ i64 456, !0 }z!2 = !{ !0, !1, !1 })r7   rs   r:   Úint64r1   r&   )r   r$   Úm0Úm1Úm2r   r   r   Útest_unnamed_metadata_2.  s    zTestIR.test_unnamed_metadata_2c                 C   sX   |   ¡ }| tdƒtdƒgtdƒgtdƒgg¡ |  d|¡ |  d|¡ |  d|¡ d S )Nr—   r›   i  z!0 = !{ i32 456 }z!1 = !{ i32 789 }z!2 = !{ i32 123, !0, !1, !0 })r7   rs   r:   r&   r*   r   r   r   Útest_unnamed_metadata_3>  s
    (zTestIR.test_unnamed_metadata_3c                 C   s$   |   ¡ }| dg¡ |  d|¡ d S )Nz"\$z!0 = !{ !"\22\5c$" })r7   rs   r&   r*   r   r   r   Útest_metadata_stringF  s    zTestIR.test_metadata_stringc              	   C   sœ   |   ¡ }| tdƒg¡}| tdƒg¡}| d¡}| |¡ | |¡ | |¡ |  d|¡ |  |¡ |  || 	d¡¡ |  
t¡ | 	d¡ W 5 Q R X d S )Nr—   r›   Úfooz!foo = !{ !0, !1, !0 }Úbar)r7   rs   r:   rœ   Úadd_named_metadatard   r&   r+   r1   Zget_named_metadataÚassertRaisesÚKeyError)r   r$   r   rž   Znmdr   r   r   Útest_named_metadataL  s    




zTestIR.test_named_metadatac                 C   s¦   |   ¡ }| tdƒg¡}| d|¡ | dtdƒg¡ | ddg¡ | dg ¡ |  d|¡ |  d|¡ |  d|¡ |  d	|¡ |  d
|¡ |  d|¡ |  |¡ d S )Nr—   r£   r›   rš   r¤   z!foo = !{ !0, !1, !2 }z!0 = !{ i32 123 }z!1 = !{ i64 456 }z!2 = !{ !"kernel" }z!bar = !{ !3 }z
!3 = !{  })r7   rs   r:   r¥   rœ   r&   r+   )r   r$   r   r   r   r   Útest_named_metadata_2\  s    zTestIR.test_named_metadata_2c                 C   sf   |   ¡ }| t ¡ d ƒg¡ |  d|¡ |  |¡ |   ¡ }| d tdƒg¡ |  d|¡ |  |¡ d S )Nz!0 = !{ i32* null }r—   z!0 = !{ null, i32 123 })r7   rs   r:   r„   r&   r+   r*   r   r   r   Útest_metadata_nulll  s    
zTestIR.test_metadata_nullc              	   C   sÄ   |   ¡ }| ddddœ¡}| dd| d g¡i¡}|jdt d¡|d	d
ddœdd}|jdd|d|d|dœdd t|ƒ}|  d|¡ |  d|¡ |  d|¡ |  d|¡ |  d|¡ |  |¡ d S )NÚDIFiler£   r¤   ©ÚfilenameÚ	directoryZDISubroutineTypeÚtypesÚDICompileUnitÚDW_LANG_PythonZARTIQr   T)ÚlanguageÚfileZproducerZruntimeVersionZisOptimized©Zis_distinctZDISubprogramr8   é   F)r>   r³   r   ÚtypeZisLocalÚunitú/!0 = !DIFile(directory: "bar", filename: "foo")z!1 = !{ null }z!!2 = !DISubroutineType(types: !1)zy!3 = distinct !DICompileUnit(file: !0, isOptimized: true, language: DW_LANG_Python, producer: "ARTIQ", runtimeVersion: 0)zd!4 = distinct !DISubprogram(file: !0, isLocal: false, line: 11, name: "my_func", type: !2, unit: !3))r7   Úadd_debug_infors   r   ÚDITokenr!   r&   r+   )r   r$   Údi_fileZdi_func_typeZdi_compileunitÚstrmodr   r   r   Útest_debug_infox  sR    þ 
þûúúù
ÿýýzTestIR.test_debug_infoc                 C   s²   |   ¡ }| ddddœ¡}| ddddœ¡}| ddddœ¡}|jddddœdd}|  ||¡ |  t||||hƒd¡ t|ƒ}|  d|¡ |  d	|¡ |  d
|¡ |  |¡ d S )Nr«   r£   r¤   r¬   Tr´   rj   r¸   z/!1 = !DIFile(directory: "foo", filename: "bar")z8!2 = distinct !DIFile(directory: "bar", filename: "foo"))r7   r¹   r1   r3   Úlenr!   r&   r+   )r   r$   Zdi1Zdi2Zdi3Zdi4r¼   r   r   r   Útest_debug_info_2¥  sF    ÿÿÿÿÿÿÿýÿÿÿzTestIR.test_debug_info_2c              
   C   s„  |   ¡ }t |t ¡ d¡}t t ¡ d¡|_| dddt d¡dœ¡}| d| d	i ¡|jd
|j|ddœdddœ¡}| 	d|¡ t
|ƒ}|  d|¡ |  d|¡ |  d|¡ |  d|¡ |  d|¡ t d¡dƒdt d¡dƒg}t d¡dƒdt d¡dƒg}| |¡}| |¡}| d¡}| |¡ | |¡ | ddddœ¡}	|jdt d ¡|	t d!¡| |g¡d"œdd}
| d#|
¡ d S )$NÚgvarr˜   ZDIBasicTypeÚfloatr   ZDW_ATE_float)r>   ÚsizeÚencodingZDIGlobalVariableExpressionZDIExpressionZDIGlobalVariableT)r>   r¶   ZisDefinitionr´   )ÚexprÚvarrp   zB!0 = !DIBasicType(encoding: DW_ATE_float, name: "float", size: 32)z!1 = !DIExpression()zK!2 = distinct !DIGlobalVariable(isDefinition: true, name: "gvar", type: !0)z3!3 = !DIGlobalVariableExpression(expr: !1, var: !2)z2@"gvar" = global float 0x4045000000000000, !dbg !3rS   zDwarf Versionri   zDebug Info Versionrj   zllvm.module.flagsr«   r£   r¤   r¬   r°   r±   Z	FullDebug)r²   r³   ZemissionKindÚglobalszllvm.dbg.cu)r7   r   ÚGlobalVariableÚ	FloatTypeÚConstantÚinitializerr¹   rº   r>   rr   r!   r&   ÚIntTypers   r¥   rd   )r   r$   rÀ   Zdi_floatZdi_gvarr¼   ZdverZdiverÚflagsr»   Zdi_cur   r   r   Útest_debug_info_gvarÄ  sj    ý
ýüþÿÿÿÿ




þ
üûzTestIR.test_debug_info_gvarc                 C   sP   |   ¡ }| dddi¡ t|ƒ}d tdd„ d ¡ ƒ¡}|  d|› d	|¡ d S )
NZDILocalVariabler>   u   aâˆ†r   c                 S   s   d| d›S )Nr   Z02xr   )Úxr   r   r   Ú<lambda>  ó    z7TestIR.test_debug_info_unicode_string.<locals>.<lambda>u   âˆ†z!0 = !DILocalVariable(name: "az"))r7   r¹   r!   r   r   Úencoder&   )r   r$   r¼   r>   r   r   r   Útest_debug_info_unicode_string  s
    z%TestIR.test_debug_info_unicode_stringc                 C   sŽ   |   ¡ }t |t t ¡ g ¡d¡}t | d¡¡}t ttg¡}tj|dddd}| 	|tdƒg¡ | 
¡  d}|  |t|ƒ¡ |  |¡ d S )	Nr£   r   ú
mov $1, $2ú=r,rT©Zside_effectr—   ú8call i32 asm sideeffect "mov $1, $2", "=r,r" ( i32 123 ))r7   r   r=   r9   ÚVoidTyper~   r@   r:   Z	InlineAsmÚcallr   r   r!   r+   )r   r$   r£   r   ÚasmtyrI   Úpatr   r   r   Útest_inline_assembly  s    zTestIR.test_inline_assemblyc                 C   s„   |   ¡ }t |t t ¡ g ¡d¡}t | d¡¡}t ttg¡}|j|ddtdƒgdd | 	¡  d}|  
|t|ƒ¡ |  |¡ d S )	Nr£   r   rÓ   rÔ   r—   TrÕ   rÖ   )r7   r   r=   r9   r×   r~   r@   r:   rI   r   r   r!   r+   )r   r$   r£   r   rÙ   rÚ   r   r   r   Útest_builder_asm  s    zTestIR.test_builder_asmc                 C   sn   |   ¡ }t |t t ¡ g ¡d¡}t | d¡¡}| t d¡d¡ | 	¡  d}|  
|t|ƒ¡ |  |¡ d S )Nr£   r   r	   Úraxzcall i64 asm "", "={rax}")r7   r   r=   r9   r×   r~   r@   Zload_regrË   r   r   r!   r+   ©r   r$   r£   r   rÚ   r   r   r   Útest_builder_load_reg(  s    zTestIR.test_builder_load_regc                 C   st   |   ¡ }t |t t ¡ g ¡d¡}t | d¡¡}| tdƒt 	d¡d¡ | 
¡  d}|  |t|ƒ¡ |  |¡ d S )Nr£   r   r—   r	   rÝ   z0call void asm sideeffect "", "{rax}" ( i64 123 ))r7   r   r=   r9   r×   r~   r@   Z	store_regrœ   rË   r   r   r!   r+   rÞ   r   r   r   Útest_builder_store_reg2  s    zTestIR.test_builder_store_regN)rX   rY   rZ   r™   r    r¡   r¢   r¨   r©   rª   r½   r¿   rÍ   rÒ   rÛ   rÜ   rß   rà   r   r   r   r   r–   %  s   -D	
r–   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestGlobalValuesc              	   C   s¦   |   ¡ }t |t t ¡ g ¡d¡}t |t t ¡ g ¡d¡ t |t ¡ d¡}|  | d¡|¡ |  | d¡|¡ |  	t
¡ | d¡ W 5 Q R X |  t|ƒd¡ d S )Nr£   r¤   Ú
globdoubleZkkkz2<ir.GlobalVariable 'globdouble' of type 'double*'>)r7   r   r=   r9   r×   rÇ   Ú
DoubleTyper3   Z
get_globalr¦   r§   Úrepr)r   r$   r£   râ   r   r   r   Útest_globals_access?  s    
ÿz$TestGlobalValues.test_globals_accessc                 C   s€   |   ¡ }t t ¡ g ¡}t ||d¡}t ||d¡}t |t ¡ d¡}|  t|j	ƒt||fƒ¡ |  t|j
ƒt|||fƒ¡ dS )zr
        Accessing functions and global values through Module.functions
        and Module.global_values.
        r£   r¤   râ   N)r7   r   r9   r×   r=   rÇ   rã   r3   ÚsetZ	functionsZglobal_values)r   r$   Zftyr£   r¤   râ   r   r   r   Ú#test_functions_global_values_accessL  s    z4TestGlobalValues.test_functions_global_values_accessc                 C   s  |   ¡ }t |td¡}tj|tddd}t |td¡}tdƒ|_t |td¡}d|_t |td	¡}d
|_tj|tddd}d|_t |td¡}d
|_tdƒ|_d|_	t |td¡}	d
|	_tdƒ|	_d|	_
t |td¡}
d
|
_tdƒ|
_d|
_	d|
_
|  |d¡ dS )z7
        IR serialization of global variables.
        ÚaÚbr˜   ©Ú	addrspacer   r—   ÚdTÚeZinternalÚfr›   Úgr   ÚhZ	h_sectionÚir   Z	i_sectionaß              @"a" = external global i8
            @"b" = external addrspace(42) global i8
            @"c" = global i32 123
            @"d" = external constant i32
            @"e" = internal global i32 undef
            @"f" = external unnamed_addr addrspace(456) global i32
            @"g" = internal global i32 123, align 16
            @"h" = internal global i32 123, section "h_section"
            @"i" = internal global i32 456, section "i_section", align 8
            N)r7   r   rÇ   re   r:   rÊ   Úglobal_constantÚlinkageZunnamed_addrrn   rw   rP   )r   r$   rè   ré   r   rì   rí   rî   rï   rð   rñ   r   r   r   Útest_global_variables_irY  s2    



z)TestGlobalValues.test_global_variables_irc                 C   s   |   ¡ }|  |¡ d S r'   )r7   r4   r*   r   r   r   Útest_pickle„  s    zTestGlobalValues.test_pickleN)rX   rY   rZ   rå   rç   rô   rõ   r   r   r   r   rá   =  s   +rá   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
Ú	TestBlockc                 C   s4   |   ¡ }tj|dd}|  |j|¡ |  |j¡ d S )NÚstart©Úparentr>   )r?   r   ÚBlockr1   rù   ÚassertFalseÚis_terminated©r   rA   rB   r   r   r   Útest_attributes‹  s    zTestBlock.test_attributesc                 C   sD   | j dd}|  |  |¡d¡ |j ddg¡ |  |  |¡d¡ d S )Nr}   ©r>   z
my_block:
rè   ré   zmy_block:
  a
  b
)rB   r3   rC   ZinstructionsÚextend)r   rB   r   r   r   Ú
test_descr‘  s    zTestBlock.test_descrc                 C   s”   | j dd}t |¡}|jjd d… \}}| ||d¡}| ||d¡}| ||d¡ t ||j	d||fd¡}|  
|d	¡ | ||¡ |  
|d
¡ d S )Nr}   rÿ   rS   r   rì   rí   Úsdivrî   z¥            my_block:
                %"c" = add i32 %".1", %".2"
                %"d" = sub i32 %".1", %".2"
                %"e" = mul i32 %"d", %".2"
            z¥            my_block:
                %"c" = add i32 %".1", %".2"
                %"f" = sdiv i32 %"c", %".2"
                %"e" = mul i32 %"f", %".2"
            )rB   r   r~   r?   rl   rd   r   ÚmulÚInstructionr¶   rM   rH   )r   rB   r   rè   ré   r   rì   rî   r   r   r   Útest_replace—  s    
zTestBlock.test_replacec                 C   s*   |   ¡ }tj|dd}|  t|ƒd¡ dS )z4
        Blocks should have a useful repr()
        r÷   rø   z"<ir.Block 'start' of type 'label'>N)r?   r   rú   r3   rä   rý   r   r   r   Ú	test_repr­  s    zTestBlock.test_reprN)rX   rY   rZ   rþ   r  r  r  r   r   r   r   rö   ‰  s   rö   c                   @   s´  e Zd ZdZdZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd1d2„ Zd3d4„ Zd5d6„ Zd7d8„ Zd9d:„ Z d;d<„ Z!d=d>„ Z"d?d@„ Z#dAdB„ Z$dCdD„ Z%dEdF„ Z&dGdH„ Z'dIdJ„ Z(dKdL„ Z)dMdN„ Z*dOdP„ Z+dQdR„ Z,dSdT„ Z-dUdV„ Z.dWdX„ Z/dYdZ„ Z0d[d\„ Z1d]d^„ Z2d_d`„ Z3dadb„ Z4dcdd„ Z5dedf„ Z6dgdh„ Z7didj„ Z8dkS )lÚTestBuildInstructionszN
    Test IR generation of LLVM instructions through the IRBuilder class.
    i   c                 C   sX   | j dd}t |¡}|jjd d… \}}| ||d¡}|  |d¡ |  t|ƒd¡ d S )Nr}   rÿ   rS   ÚreszP            my_block:
                %"res" = add i32 %".1", %".2"
            z}<ir.Instruction 'res' of type 'i32', opname 'add', operands (<ir.Argument '.1' of type i32>, <ir.Argument '.2' of type i32>)>)	rB   r   r~   r?   rl   rd   rM   r3   rä   )r   rB   r   rè   ré   Úinstr   r   r   Útest_simple½  s    

ÿz!TestBuildInstructions.test_simplec              	   C   sx  | j dd}t |¡}|jjd d… \}}}| ||d¡ | ||d¡ | ||d¡ | ||d¡ | 	||d¡ | 
||d	¡ | ||d
¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ | ||d¡ |  t¡}| ||¡ W 5 Q R X |  t|jƒd¡ |  |j¡ |  |d¡ d S )Nr}   rÿ   rj   r   rì   rí   rî   rï   rð   rñ   ÚjÚkÚlÚmÚnÚoÚpÚqÚrÚsÚtz1Operands must be the same type, got (i32, double)aD              my_block:
                %"c" = add i32 %".1", %".2"
                %"d" = fadd i32 %".1", %".2"
                %"e" = sub i32 %".1", %".2"
                %"f" = fsub i32 %".1", %".2"
                %"g" = mul i32 %".1", %".2"
                %"h" = fmul i32 %".1", %".2"
                %"i" = udiv i32 %".1", %".2"
                %"j" = sdiv i32 %".1", %".2"
                %"k" = fdiv i32 %".1", %".2"
                %"l" = urem i32 %".1", %".2"
                %"m" = srem i32 %".1", %".2"
                %"n" = frem i32 %".1", %".2"
                %"o" = or i32 %".1", %".2"
                %"p" = and i32 %".1", %".2"
                %"q" = xor i32 %".1", %".2"
                %"r" = shl i32 %".1", %".2"
                %"s" = ashr i32 %".1", %".2"
                %"t" = lshr i32 %".1", %".2"
            )rB   r   r~   r?   rl   rd   Úfaddr   Úfsubr  ÚfmulÚudivr  ÚfdivÚuremÚsremÚfremÚor_Úand_ÚxorÚshlÚashrÚlshrr¦   r‘   r3   r!   Ú	exceptionrû   rü   rM   )r   rB   r   rè   ré   ÚffÚcmr   r   r   Útest_binopsÌ  s8    
ÿz!TestBuildInstructions.test_binopsc                 C   sb   | j dd}t |¡}|jjd d… \}}|j||ddd |j||ddd	gd |  |d
¡ d S )Nr}   rÿ   rS   r   )Únuw©rÌ   rì   r(  Znswz†            my_block:
                %"c" = add nuw i32 %".1", %".2"
                %"d" = sub nuw nsw i32 %".1", %".2"
            )rB   r   r~   r?   rl   rd   r   rM   ©r   rB   r   rè   ré   r   r   r   Útest_binop_flagsý  s    
z&TestBuildInstructions.test_binop_flagsc                 C   sb   | j dd}t |¡}|jjd d… \}}|j||ddd |j||ddd	gd |  |d
¡ d S )Nr}   rÿ   rS   r   ©Úfastr)  rì   ÚninfÚnszzŠ            my_block:
                %"c" = fadd fast i32 %".1", %".2"
                %"d" = fsub ninf nsz i32 %".1", %".2"
            )rB   r   r~   r?   rl   r  r  rM   r*  r   r   r   Útest_binop_fastmath_flags  s    
z/TestBuildInstructions.test_binop_fastmath_flagsc                 C   sŽ   | j dd}t |¡}|jjd d… \}}| ||d¡ | ||d¡ | ||d¡ | ||d¡ | 	||d¡ | 
||d	¡ |  |d
¡ d S )Nr}   rÿ   rS   r   rì   rí   rî   rï   rð   að  my_block:
    %"c" = call {i32, i1} @"llvm.sadd.with.overflow.i32"(i32 %".1", i32 %".2")
    %"d" = call {i32, i1} @"llvm.smul.with.overflow.i32"(i32 %".1", i32 %".2")
    %"e" = call {i32, i1} @"llvm.ssub.with.overflow.i32"(i32 %".1", i32 %".2")
    %"f" = call {i32, i1} @"llvm.uadd.with.overflow.i32"(i32 %".1", i32 %".2")
    %"g" = call {i32, i1} @"llvm.umul.with.overflow.i32"(i32 %".1", i32 %".2")
    %"h" = call {i32, i1} @"llvm.usub.with.overflow.i32"(i32 %".1", i32 %".2")
            )rB   r   r~   r?   rl   Zsadd_with_overflowZsmul_with_overflowZssub_with_overflowZuadd_with_overflowZumul_with_overflowZusub_with_overflowrM   r*  r   r   r   Útest_binops_with_overflow  s    
z/TestBuildInstructions.test_binops_with_overflowc                 C   sl   | j dd}t |¡}|jjd d… \}}}| |d¡ | |d¡ | |d¡ |  |j	¡ |  
|d¡ d S )Nr}   rÿ   rj   rì   rí   rî   zœ            my_block:
                %"d" = sub i32 0, %".1"
                %"e" = xor i32 %".2", -1
                %"f" = fneg double %".3"
            )rB   r   r~   r?   rl   ÚnegÚnot_Úfnegrû   rü   rM   ©r   rB   r   rè   ré   r   r   r   r   Útest_unary_ops-  s    
z$TestBuildInstructions.test_unary_opsc                 C   s”   | j dd}t |¡}|jjd d… \}}t t d¡tj¡}t t d¡tj¡}| ||d¡}|  	|d¡ | 
||¡ | 
||¡ |  	|d¡ d S )Nr}   rÿ   rS   r   r   zN            my_block:
                %"c" = add i32 undef, undef
            zN            my_block:
                %"c" = add i32 %".1", %".2"
            )rB   r   r~   r?   rl   rÉ   rË   Ú	Undefinedrd   rM   Zreplace_usage)r   rB   r   rè   ré   Zundef1Zundef2r   r   r   r   Útest_replace_operand<  s    
z*TestBuildInstructions.test_replace_operandc              	   C   sR  | j dd}t |¡}|jjd d… \}}| d||d¡ | d||d¡ | d||d	¡ | d
||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d
||d¡ | d||d¡ | d||d¡ |  t¡ | d||d¡ W 5 Q R X |  t¡ | d||d¡ W 5 Q R X |  	|j
¡ |  |d¡ d S )Nr}   rÿ   rS   ú==r   ú!=rì   ú<rí   ú<=rî   ú>rï   ú>=rð   rñ   r  r  r  r  r  ÚunoÚzzr£   aj              my_block:
                %"c" = icmp eq i32 %".1", %".2"
                %"d" = icmp ne i32 %".1", %".2"
                %"e" = icmp ult i32 %".1", %".2"
                %"f" = icmp ule i32 %".1", %".2"
                %"g" = icmp ugt i32 %".1", %".2"
                %"h" = icmp uge i32 %".1", %".2"
                %"i" = icmp eq i32 %".1", %".2"
                %"j" = icmp ne i32 %".1", %".2"
                %"k" = icmp slt i32 %".1", %".2"
                %"l" = icmp sle i32 %".1", %".2"
                %"m" = icmp sgt i32 %".1", %".2"
                %"n" = icmp sge i32 %".1", %".2"
            )rB   r   r~   r?   rl   Úicmp_unsignedÚicmp_signedr¦   r‘   rû   rü   rM   r*  r   r   r   Útest_integer_comparisonsN  s*    
z.TestBuildInstructions.test_integer_comparisonsc              
   C   sd  | j dd}t |¡}|jjd d… \}}| d||d¡ | d||d¡ | d||d	¡ | d
||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d
||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ | d||d¡ |jd||dddd d!d"gd# |  |j¡ |  	|d$¡ d S )%Nr}   rÿ   rS   r9  r   r:  rì   r;  rí   r<  rî   r=  rï   r>  rð   rñ   r  r  r  r  r  ÚordÚur?  ÚvÚwrÎ   ÚoltÚyÚnnanr.  r/  Úarcpr-  r)  a{              my_block:
                %"c" = fcmp oeq i32 %".1", %".2"
                %"d" = fcmp one i32 %".1", %".2"
                %"e" = fcmp olt i32 %".1", %".2"
                %"f" = fcmp ole i32 %".1", %".2"
                %"g" = fcmp ogt i32 %".1", %".2"
                %"h" = fcmp oge i32 %".1", %".2"
                %"i" = fcmp ueq i32 %".1", %".2"
                %"j" = fcmp une i32 %".1", %".2"
                %"k" = fcmp ult i32 %".1", %".2"
                %"l" = fcmp ule i32 %".1", %".2"
                %"m" = fcmp ugt i32 %".1", %".2"
                %"n" = fcmp uge i32 %".1", %".2"
                %"u" = fcmp ord i32 %".1", %".2"
                %"v" = fcmp uno i32 %".1", %".2"
                %"w" = fcmp ord i32 %".1", %".2"
                %"x" = fcmp uno i32 %".1", %".2"
                %"y" = fcmp nnan ninf nsz arcp fast olt i32 %".1", %".2"
            )
rB   r   r~   r?   rl   Úfcmp_orderedÚfcmp_unorderedrû   rü   rM   r*  r   r   r   Útest_float_comparisonss  s0    
ÿz,TestBuildInstructions.test_float_comparisonsc                 C   sz   | j dd}t td¡}t |¡}|jjd d… \}}|j|||ddd |  |j	¡ | 
¡  |  |j	¡ |  |d¡ d S )	Nr}   rÿ   TrS   r   )rK  rJ  r)  z„            my_block:
                %"c" = select arcp nnan i1 true, i32 %".1", i32 %".2"
                unreachable
            )rB   r   rÉ   Úint1r~   r?   rl   Úselectrû   rü   ZunreachableÚ
assertTruerM   )r   rB   r  r   rè   ré   r   r   r   Útest_misc_ops   s    
z#TestBuildInstructions.test_misc_opsc                 C   s†   | j dd}t |¡}|jjd d… \}}|j d¡}|j d¡}|jtddd}| ||¡ | ||¡ |  	|j
¡ |  |d	¡ d S )
Nr}   rÿ   rS   Zb2Zb3Zmy_phir,  r)  zj            my_block:
                %"my_phi" = phi fast i32 [%".1", %"b2"], [%".2", %"b3"]
            )rB   r   r~   r?   rl   r@   Úphir:   Zadd_incomingrû   rü   rM   )r   rB   r   rè   ré   Zbb2Zbb3rS  r   r   r   Útest_phi¯  s    
zTestBuildInstructions.test_phic              	   C   sÔ  | j dd}t |¡}|jjd d… \}}}|jtdd}|jtddd}|jt|dd}d	|_|  	|j
t t¡¡ | ||¡}	|  	|	j
t ¡ ¡ | ||¡}
|  	|
j
t ¡ ¡ | |d
¡}|  	|j
t¡ |j||dd}|  	|j
t ¡ ¡ |j|ddd}|  	|j
t¡ |j||ddd}|  	|j
t ¡ ¡ |j|dddd}|  	|j
t¡ |  t¡ | ||¡ W 5 Q R X |  t¡ | |¡ W 5 Q R X |  t¡}| ||¡ W 5 Q R X |  	t|jƒd¡ |  |d¡ d S )Nr}   rÿ   rj   r   r˜   rì   )rÂ   r>   rí   r   rï   r   )rn   rñ   Úseq_cstri   )Úorderingrn   r  )rV  rn   r>   z.cannot store i32 to double*: mismatching typesa              my_block:
                %"c" = alloca i32
                %"d" = alloca i32, i32 42
                %"e" = alloca double, i32 %".1", align 8
                store double %".3", double* %"e"
                store i32 %".2", i32* %"c"
                %"g" = load i32, i32* %"c"
                store i32 %".2", i32* %"c", align 1
                %"i" = load i32, i32* %"c", align 1
                store atomic i32 %".2", i32* %"c" seq_cst, align 4
                %"k" = load atomic i32, i32* %"c" seq_cst, align 4
            )rB   r   r~   r?   rl   Úallocar:   r;   rn   r3   r¶   r<   Ústorer×   ÚloadZstore_atomicZload_atomicr¦   Ú	TypeErrorr!   r$  rM   )r   rB   r   rè   ré   Úzr   rì   rí   Úeerî   rï   rð   rñ   r  r  r&  r   r   r   Útest_mem_ops¾  s@    
ÿz"TestBuildInstructions.test_mem_opsc                 C   s~   | j dd}t |¡}|jjd d… \}}|jt t¡dd}|j|t 	td¡|gdd}|  
|jt t¡¡ |  |d¡ d S )Nr}   rÿ   rS   r   rg   rì   z            my_block:
                %"c" = alloca i32*
                %"d" = getelementptr i32*, i32** %"c", i32 5, i32 %".1"
            )rB   r   r~   r?   rl   rW  r<   r:   ÚgeprÉ   r3   r¶   rM   ©r   rB   r   rè   ré   r   rì   r   r   r   Útest_gepï  s    
zTestBuildInstructions.test_gepc           	      C   s˜   | j dd}t |¡}|jjd d… \}}t ¡ }t t|||tg¡}|j	|| ¡ dd}|j
|dd„ dD ƒdd}|  |jt |¡¡ |  |d	¡ d S )
Nr}   rÿ   rS   rì   c                 S   s   g | ]}t  t|¡‘qS r   ©r   rÉ   r:   ©r   rÎ   r   r   r   r      s     z<TestBuildInstructions.test_gep_castinstr.<locals>.<listcomp>©r   rj   rí   z×            my_block:
                %"d" = bitcast i32 %".1" to {i64, i8*, i8*, i8*, i64}*
                %"e" = getelementptr {i64, i8*, i8*, i8*, i64}, {i64, i8*, i8*, i8*, i64}* %"d", i32 0, i32 3
            )rB   r   r~   r?   rl   re   r„   ÚLiteralStructTyperœ   Úbitcastr^  r3   r¶   r<   rM   )	r   rB   r   rè   ré   Úint8ptrÚlsrì   rí   r   r   r   Útest_gep_castinstrý  s    
z(TestBuildInstructions.test_gep_castinstrc           
      C   s´   | j dd}t |¡}|jjd d… \}}d}t ¡ }t t|||tg¡}|j	||j|ddd}|j
|dd„ d	D ƒd
d}	|  |	jj|¡ |  |	jtj||d¡ |  |d¡ d S )Nr}   rÿ   rS   ri   rê   rì   c                 S   s   g | ]}t  t|¡‘qS r   ra  rb  r   r   r   r      s     zFTestBuildInstructions.test_gep_castinstr_addrspace.<locals>.<listcomp>rc  rí   zñ            my_block:
                %"d" = bitcast i32 %".1" to {i64, i8*, i8*, i8*, i64} addrspace(4)*
                %"e" = getelementptr {i64, i8*, i8*, i8*, i64}, {i64, i8*, i8*, i8*, i64} addrspace(4)* %"d", i32 0, i32 3
            )rB   r   r~   r?   rl   re   r„   rd  rœ   re  r^  r3   r¶   rë   r<   rM   )
r   rB   r   rè   ré   rë   rf  rg  rì   rí   r   r   r   Útest_gep_castinstr_addrspace  s    
z2TestBuildInstructions.test_gep_castinstr_addrspacec           	      C   sÐ   | j dd}t |¡}|jjd d… \}}d}|jtjt|ddd}|  t	|j
ƒd¡ |  |j
jj|¡ |j|t td¡|gd	d}|  |j
j|¡ |j|t td
¡gdd}|  |j
j|¡ |  |d¡ d S )Nr}   rÿ   rS   ri   rê   r   zi32 addrspace(4)**rg   rì   rh   rí   zý            my_block:
                %"c" = alloca i32 addrspace(4)*
                %"d" = getelementptr i32 addrspace(4)*, i32 addrspace(4)** %"c", i32 5, i32 %".1"
                %"e" = getelementptr i32, i32 addrspace(4)* %"d", i32 10
            )rB   r   r~   r?   rl   rW  r<   r:   r3   r!   r¶   Úpointeerë   r^  rÉ   rM   )	r   rB   r   rè   ré   rë   r   rì   rí   r   r   r   Útest_gep_addrspace!  s    
z(TestBuildInstructions.test_gep_addrspacec              	   C   sP  | j dd}t |¡}|jjd d… \}}t ttg¡}t t|g¡}t 	|t 	td¡t 	td¡f¡}|j
|ddd}|j||ddd}	|j|	t 	td	¡d
dd}
|  |	j|¡ |  |
j|¡ |j|dd}|j|dd}|j
|ddd}|j
|d
dd}|j
|ddd}|j
|ddd}|j||d
dd}|j||ddd}|  |jt¡ |  |j|¡ |  |jt¡ |  |jt¡ |  |j|¡ |  |j|¡ |  t¡ | 
|d¡ W 5 Q R X |  t¡ | 
|d¡ W 5 Q R X |  t¡ | 
|d¡ W 5 Q R X |  t¡ | ||d¡ W 5 Q R X |  t¡ | ||d
¡ W 5 Q R X |  |d¡ d S )Nr}   rÿ   rS   ri   Tr   r   rì   Fr   rí   Úptrr  r  r  )r   r   r  )r   r   r  r  r  )r   r   rg   aî              my_block:
                %"c" = extractvalue {i32, i1} {i32 4, i1 true}, 0
                %"d" = insertvalue {i32, i1} {i32 4, i1 true}, i32 %".1", 0
                %"e" = insertvalue {i32, i1} %"d", i1 false, 1
                %"ptr" = alloca {i8, {i32, i1}}
                %"j" = load {i8, {i32, i1}}, {i8, {i32, i1}}* %"ptr"
                %"k" = extractvalue {i8, {i32, i1}} %"j", 0
                %"l" = extractvalue {i8, {i32, i1}} %"j", 1
                %"m" = extractvalue {i8, {i32, i1}} %"j", 1, 0
                %"n" = extractvalue {i8, {i32, i1}} %"j", 1, 1
                %"o" = insertvalue {i8, {i32, i1}} %"j", {i32, i1} %"l", 1
                %"p" = insertvalue {i8, {i32, i1}} %"j", i32 %".1", 1, 0
            )rB   r   r~   r?   rl   rd  r:   rO  re   rÉ   Zextract_valueZinsert_valuer3   r¶   rW  rY  r¦   rZ  rM   )r   rB   r   rè   ré   Ztp_innerZtp_outerZc_innerr   rì   rí   Zp_outerr  r  r  r  r  r  r  r   r   r   Útest_extract_insert_value4  sL    

ÿz/TestBuildInstructions.test_extract_insert_valuec                 C   sþ   | j dd}t |¡}|jjd d… \}}}}|j|tdd}|j|tdd}|j	|tdd}	| 
|td¡}
| |
td¡}| |td	¡}| |td
¡}| |td¡}| |td¡}| |td¡}| |t t¡d¡}| |td¡}|  |j¡ |  |d¡ d S )Nr}   rÿ   ri   r   rì   rí   ÚfbÚfcrï   rð   ÚfdÚferñ   r  r  ah              my_block:
                %"c" = trunc i32 %".1" to i8
                %"d" = zext i8 %"c" to i32
                %"e" = sext i8 %"c" to i32
                %"fb" = fptrunc double %".3" to float
                %"fc" = fpext float %"fb" to double
                %"g" = fptoui double %".3" to i32
                %"h" = fptosi double %".3" to i8
                %"fd" = uitofp i32 %"g" to float
                %"fe" = sitofp i8 %"h" to double
                %"i" = ptrtoint i32* %".4" to i32
                %"j" = inttoptr i32 %"i" to i8*
                %"k" = bitcast i32 %".1" to float
            )rB   r   r~   r?   rl   Útruncre   Úzextr:   ÚsextÚfptruncÚfltÚfpextr;   ÚfptouiÚfptosiÚuitofpÚsitofpÚptrtointÚinttoptrr<   re  rû   rü   rM   )r   rB   r   rè   ré   Úfarl  r   rì   rí   rn  ro  rï   rð   rp  rq  rñ   r  r  r   r   r   Útest_cast_opsq  s"    
z#TestBuildInstructions.test_cast_opsc                 C   sh   | j dd}t |¡}|jjd d… \}}|jtdd}| d||dd¡}|  |j	t¡ |  
|d¡ d S )	Nr}   rÿ   rS   r   rd   Ú	monotonicrì   zˆ            my_block:
                %"c" = alloca i32
                %"d" = atomicrmw add i32* %"c", i32 %".1" monotonic
            )rB   r   r~   r?   rl   rW  r:   Z
atomic_rmwr3   r¶   rM   r_  r   r   r   Útest_atomicrmw’  s    
z$TestBuildInstructions.test_atomicrmwc                 C   sJ   | j dd}t |¡}|jjdd}| |¡ |  |j¡ |  |d¡ d S )Nr}   rÿ   ÚtargetzE            my_block:
                br label %"target"
            )	rB   r   r~   r?   r@   ÚbranchrQ  rü   rM   )r   rB   r   Z	bb_targetr   r   r   Útest_branchŸ  s    

z!TestBuildInstructions.test_branchc                 C   sd   | j dd}t |¡}|jjdd}|jjdd}| t td¡||¡ |  |j	¡ |  
|d¡ d S )Nr}   rÿ   Úb_trueÚb_falseFza            my_block:
                br i1 false, label %"b_true", label %"b_false"
            )rB   r   r~   r?   r@   ÚcbranchrÉ   rO  rQ  rü   rM   )r   rB   r   Úbb_trueÚbb_falser   r   r   Útest_cbranchª  s    
z"TestBuildInstructions.test_cbranchc                 C   s€   | j dd}t |¡}|jjdd}|jjdd}| t td¡||¡}| ddg¡ |  	|j
¡ |  |d¡ |  |jd	¡ d S )
Nr}   rÿ   r…  r†  Frg   r˜   zk            my_block:
                br i1 false, label %"b_true", label %"b_false", !prof !0
            zC            !0 = !{ !"branch_weights", i32 5, i32 42 }
            )rB   r   r~   r?   r@   r‡  rÉ   rO  Zset_weightsrQ  rü   rM   rQ   r7   )r   rB   r   rˆ  r‰  Úbrr   r   r   Útest_cbranch_weights¶  s    
z*TestBuildInstructions.test_cbranch_weightsc                 C   sv   | j dd}t |¡}|jjdd}|jjdd}| t |j|¡¡}| |¡ | |¡ |  |j	¡ |  
|d¡ d S )Nr}   rÿ   Zb_1Zb_2z€            my_block:
                indirectbr i8* blockaddress(@"my_func", %"b_1"), [label %"b_1", label %"b_2"]
            )rB   r   r~   r?   r@   Zbranch_indirectZBlockAddressZadd_destinationrQ  rü   rM   )r   rB   r   Zbb_1Zbb_2Z
indirectbrr   r   r   Útest_branch_indirectÆ  s    
ÿ

z*TestBuildInstructions.test_branch_indirectc                    sä   ‡ fdd„}ˆ j dd}t |¡}| ¡  ||dƒ ˆ j dd}t |¡}| tdƒ¡ ||dƒ ˆ j dd}t |¡}| ¡ }| d	|j d
¡¡ ||dƒ ˆ j dd}t |¡}| tdƒ¡}| d	|j d
¡¡ ||dƒ d S )Nc                    s   ˆ   | j¡ ˆ  | |¡ d S r'   )rQ  rü   rM   )rB   Zexpected_irr6   r   r   ÚcheckÖ  s    z1TestBuildInstructions.test_returns.<locals>.checkr}   rÿ   z;            my_block:
                ret void
            Zother_blockrg   z?            other_block:
                ret i32 5
            rp   r   zD            my_block:
                ret void, !dbg !0
            é   zE            my_block:
                ret i32 6, !dbg !0
            )	rB   r   r~   r   Úretr:   rr   r7   rs   )r   rŽ  rB   r   r	  r   r6   r   Útest_returnsÕ  s&    






z"TestBuildInstructions.test_returnsc           
      C   s¾   | j dd}t |¡}|jjd d… \}}|jjdd}|jjdd}|jjdd}|jjdd}| ||¡}	|	 t t	d¡|¡ |	 t t	d	¡|¡ |	 d|¡ |  
|j¡ |  |d
¡ d S )Nr}   rÿ   rS   ZonzeroZononeZontwoÚ	otherwiser   r   zœ            my_block:
                switch i32 %".1", label %"otherwise" [i32 0, label %"onzero" i32 1, label %"onone" i32 2, label %"ontwo"]
            )rB   r   r~   r?   rl   r@   ÚswitchZadd_caserÉ   r:   rQ  rü   rM   )
r   rB   r   rè   ré   Z	bb_onzeroZbb_ononeZbb_ontwoZbb_elseÚswr   r   r   Útest_switchý  s    
z!TestBuildInstructions.test_switchc                 C   sz  | j dd}t |¡}|jjd d… \}}t tttf¡}tjttfdd}t t	ttf¡}t 
|jj|d¡}t 
|jj|d¡}	t 
|jj|d¡}
| |||fd	¡ | |	||fd
¡ | |
||fd¡ |j|||fddd | |||fd¡}|j d¡ |j|||fddd |j|||fddd |j|||fddd |j|||fddd |j|||fddd |  |d¡ d S )Nr}   rÿ   rS   Trb   rî   rï   rð   Úres_fZres_gZres_hZ
res_f_fastÚfastcc)ÚcconvÚres_f_readonlyÚreadonlyZres_fastr-  )ÚfastmathZres_nnan_ninf)rJ  r.  Zres_noinlinerŽ   )ÚattrsZres_alwaysinlinera   Zres_noinline_ro)rŽ   rš  a          my_block:
            %"res_f" = call float @"f"(i32 %".1", i32 %".2")
            %"res_g" = call double (i32, ...) @"g"(i32 %".2", i32 %".1")
            %"res_h" = call half @"h"(i32 %".1", i32 %".2")
            %"res_f_fast" = call fastcc float @"f"(i32 %".1", i32 %".2")
            %"res_f_readonly" = call float @"f"(i32 %".1", i32 %".2") readonly
            %"res_fast" = call fast float @"f"(i32 %".1", i32 %".2")
            %"res_nnan_ninf" = call ninf nnan float @"f"(i32 %".1", i32 %".2")
            %"res_noinline" = call float @"f"(i32 %".1", i32 %".2") noinline
            %"res_alwaysinline" = call float @"f"(i32 %".1", i32 %".2") alwaysinline
            %"res_noinline_ro" = call float @"f"(i32 %".1", i32 %".2") noinline readonly
        )rB   r   r~   r?   rl   r9   rv  r:   r;   Úhlfr=   r7   rØ   rc   rd   rM   )r   rB   r   rè   ré   Útp_fZtp_gZtp_hrî   rï   rð   r™  r   r   r   Ú	test_call  s*    
zTestBuildInstructions.test_callc                 C   s|   | j dd}t |¡}t t ¡ t ¡ gd ¡}t |j|d¡}|jt	dd}|j 
d¡}| ||||f¡ |  |d¡ dS )	z9
        Function calls with metadata arguments.
        r}   rÿ   rj   zllvm.dbg.declarerè   r   z             my_block:
                %"a" = alloca i32
                call void @"llvm.dbg.declare"(metadata i32* %"a", metadata !0, metadata !0)
            N)rB   r   r~   r9   r×   ZMetaDataTyper=   r7   rW  r:   rs   rØ   rM   )r   rB   r   Zdbg_declare_tyZdbg_declarerè   ré   r   r   r   Útest_call_metadata4  s    
ýz(TestBuildInstructions.test_call_metadatac                 C   s¦   | j dd}t |¡}t t ¡ t ¡ tt ¡ f¡}t |jj	|d¡}|j
d  d¡ |jtdd}|jtdd}|j||t td¡|fd	d
dœd |  |d¡ d S )Nr}   rÿ   Úfunr   ÚsretÚretvalÚotherr˜   ©r¢  rk   rk   ©r   rS   ©Ú	arg_attrsz¸        my_block:
            %"retval" = alloca i32
            %"other" = alloca i32
            call void @"fun"(i32* noalias sret %"retval", i32 42, i32* noalias %"other")
        )rB   r   r~   r9   r×   r:   r„   r=   r?   r7   rl   rm   rW  rØ   rÉ   rM   )r   rB   r   Úfun_tyr¡  r£  r¤  r   r   r   Útest_call_attributesH  s$    
 ÿþýz*TestBuildInstructions.test_call_attributesc              	   C   sb   |   ¡ }t |¡}t t ¡ d¡}t |jj|d¡}|  t	¡ |j
|dddid W 5 Q R X d S )Nr   r¡  r   r¢  r§  )rB   r   r~   r9   r×   r=   r?   r7   r¦   r‘   rØ   )r   rB   r   r©  r¡  r   r   r   Útest_invalid_call_attributes`  s    
z2TestBuildInstructions.test_invalid_call_attributesc           	      C   sŽ   | j dd}t |¡}|jjd d… \}}t tttf¡}t |jj	|d¡}|jj
dd}|jj
dd}| |||f||d¡ |  |d¡ d S )	Nr}   rÿ   rS   rî   ÚnormalÚunwindr–  z£            my_block:
                %"res_f" = invoke float @"f"(i32 %".1", i32 %".2")
                    to label %"normal" unwind label %"unwind"
            )rB   r   r~   r?   rl   r9   rv  r:   r=   r7   r@   ÚinvokerM   )	r   rB   r   rè   ré   rž  rî   Ú	bb_normalÚ	bb_unwindr   r   r   Útest_invokei  s    
z!TestBuildInstructions.test_invokec           	      C   sÒ   | j dd}t |¡}t t ¡ t ¡ tt ¡ f¡}t |jj	|d¡}d|_
|jd  d¡ |jtdd}|jtdd}|jjd	d}|jjd
d}|j||t td¡|f||ddddddœd |  |d¡ d S )Nr}   rÿ   r¡  r—  r   r¢  r£  r¤  r¬  r­  r˜   r-  rŽ   r¥  rk   r¦  )r˜  r›  rœ  r¨  a	          my_block:
            %"retval" = alloca i32
            %"other" = alloca i32
            invoke fast fastcc void @"fun"(i32* noalias sret %"retval", i32 42, i32* noalias %"other") noinline
                to label %"normal" unwind label %"unwind"
        )rB   r   r~   r9   r×   r:   r„   r=   r?   r7   Zcalling_conventionrl   rm   rW  r@   r®  rÉ   rM   )	r   rB   r   r©  r¡  r£  r¤  r¯  r°  r   r   r   Útest_invoke_attributesx  s4    
 ÿþøz,TestBuildInstructions.test_invoke_attributesc              
   C   sš   | j dd}t |¡}| t tt ¡ g¡d¡}t |j	j
t ¡ d¡}d|_| t |¡¡ | t t t |jd¡|g¡¡¡ | |¡ |  |d¡ d S )Nr}   rÿ   ÚlpZ_ZTIiTr   zÖ            my_block:
                %"lp" = landingpad {i32, i8*}
                    catch i8** @"_ZTIi"
                    filter [1 x i8**] [i8** @"_ZTIi"]
                resume {i32, i8*} %"lp"
            )rB   r   r~   Z
landingpadrd  r:   re   r„   rÇ   r?   r7   rò   Z
add_clauseZCatchClauseZFilterClauserÉ   Ú	ArrayTyper¶   ÚresumerM   )r   rB   r   r³  Zint_typeinfor   r   r   Útest_landingpad™  s*    

ÿÿ
 ÿ ÿÿ
z%TestBuildInstructions.test_landingpadc                 C   sV   | j dd}t |¡}|jjd d… \}}|jd||dd}| |¡ |  |d¡ d S )Nr}   rÿ   rS   r=  r   z…            my_block:
                %"c" = icmp sgt i32 %".1", %".2"
                call void @"llvm.assume"(i1 %"c")
            )rB   r   r~   r?   rl   rB  rˆ   rM   r5  r   r   r   Útest_assume­  s    

z!TestBuildInstructions.test_assumec                 C   sŽ  | j dd}t |¡}|jjd d… \}}d|_d|_t |jd¡}t |tj	¡}t 
d¡}|j|||dƒdd}|j|||d	ƒd
d}|  |d¡ | d¡}| |¡ | |¡ t |d	dg¡}|j|||dd |  |d¡ | d¡}| |¡ | |¡ |j||dd |  |d¡ | d¡}| |¡ | |¡ |j||dƒdd}	|j||d	ƒdd}
|  |d¡ | | |	|
¡¡ |  |j¡ d S )NZinsert_blockrÿ   rS   rè   ré   r   r   Zvec1r   Zvec2z§insert_block:
    %"vec1" = insertelement <2 x i32> <i32 undef, i32 undef>, i32 %"a", i32 0
    %"vec2" = insertelement <2 x i32> %"vec1", i32 %"b", i32 1
            Zshuffle_blockZshufzŽ            shuffle_block:
                %"shuf" = shufflevector <2 x i32> %"vec2", <2 x i32> %"vec2", <2 x i32> <i32 1, i32 0>
            Z	add_blockÚsumz[            add_block:
                %"sum" = add <2 x i32> %"vec2", %"vec2"
            Zextract_blockZex1Zex2z¥            extract_block:
              %"ex1" = extractelement <2 x i32> %"vec2", i32 0
              %"ex2" = extractelement <2 x i32> %"vec2", i32 1
            )rB   r   r~   r?   rl   r>   Ú
VectorTyper¶   rÉ   r7  rË   Zinsert_elementrM   r@   rƒ  Úposition_at_endZshuffle_vectorrd   Zextract_elementr  r+   r7   )r   rB   r   rè   ré   ÚvectyÚvecZidxtyÚmaskr   rì   r   r   r   Útest_vector_ops¹  s<    










z%TestBuildInstructions.test_vector_opsc                 C   sJ   | j dd}t |¡}t td¡}|j|dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   z}            my_block:
                %"c" = call i64 @"llvm.bitreverse.i64"(i64 5)
                ret i64 %"c"
            )rB   r   r~   rÉ   rœ   Ú
bitreverser  rM   ©r   rB   r   rè   r   r   r   r   Útest_bitreverseó  s    

z%TestBuildInstructions.test_bitreversec              	   C   s\   | j dd}t |¡}t td¡}|  t¡}|j|dd W 5 Q R X |  dt	|j
ƒ¡ d S )Nr}   rÿ   rg   r   ú#expected an integer type, got float)rB   r   r~   rÉ   rv  r¦   rZ  r¿  r#   r!   r$  ©r   rB   r   rè   Úraisesr   r   r   Útest_bitreverse_wrongtypeÿ  s    
þz/TestBuildInstructions.test_bitreverse_wrongtypec              	   C   sÄ   | j dd}t |¡}|  t¡}| dd ¡ W 5 Q R X |  dt|jƒ¡ |  t¡}| d d¡ W 5 Q R X |  dt|jƒ¡ | dd ¡ | dd¡ | d	d¡ | d
¡ | 	¡  |  
|d¡ d S )Nr}   rÿ   r€  z#Invalid fence ordering "monotonic"!zInvalid fence ordering "None"!ÚacquireÚreleaseZsinglethreadZacq_relrU  zç            my_block:
                fence acquire
                fence syncscope("singlethread") release
                fence syncscope("singlethread") acq_rel
                fence seq_cst
                ret void
            )rB   r   r~   r¦   r‘   Zfencer#   r!   r$  r   rM   )r   rB   r   rÄ  r   r   r   Ú
test_fence
  s(    
þþ
z TestBuildInstructions.test_fencec                 C   sJ   | j dd}t |¡}t td¡}|j|dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   zx            my_block:
                %"c" = call i32 @"llvm.bswap.i32"(i32 5)
                ret i32 %"c"
            )rB   r   r~   rÉ   r:   Zbswapr  rM   rÀ  r   r   r   Ú
test_bswap%  s    

z TestBuildInstructions.test_bswapc                 C   sJ   | j dd}t |¡}t td¡}|j|dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   zx            my_block:
                %"c" = call i16 @"llvm.ctpop.i16"(i16 5)
                ret i16 %"c"
            )rB   r   r~   rÉ   Úint16Zctpopr  rM   rÀ  r   r   r   Ú
test_ctpop1  s    

z TestBuildInstructions.test_ctpopc                 C   sX   | j dd}t |¡}t td¡}t td¡}|j||dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   r   z}            my_block:
                %"c" = call i16 @"llvm.ctlz.i16"(i16 5, i1 1)
                ret i16 %"c"
            )	rB   r   r~   rÉ   rÊ  rO  Zctlzr  rM   r5  r   r   r   Ú	test_ctlz=  s    

zTestBuildInstructions.test_ctlzc                 C   sJ   | j dd}t |¡}t td¡}|j|dd}| |¡ |  |d¡ d S )Nr}   rÿ   g      @ré   z•            my_block:
                %"b" = call i16 @"llvm.convert.to.fp16.f32"(float 0x4014000000000000)
                ret i16 %"b"
            )rB   r   r~   rÉ   rv  Úconvert_to_fp16r  rM   r*  r   r   r   Útest_convert_to_fp16_f32J  s    

z.TestBuildInstructions.test_convert_to_fp16_f32c              	   C   s\   | j dd}t |¡}t td¡}|  t¡}|j|dd W 5 Q R X |  dt	|j
ƒ¡ d S )Nr}   rÿ   rg   ré   úexpected a float type, got i16)rB   r   r~   rÉ   rÊ  r¦   rZ  rÍ  r#   r!   r$  rÃ  r   r   r   Ú"test_convert_to_fp16_f32_wrongtypeV  s    
þz8TestBuildInstructions.test_convert_to_fp16_f32_wrongtypec                 C   sL   | j dd}t |¡}t td¡}|j|dtd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   ré   ©r>   Útozˆ            my_block:
                %"b" = call float @"llvm.convert.from.fp16.f32"(i16 5)
                ret float %"b"
            )	rB   r   r~   rÉ   rÊ  Úconvert_from_fp16rv  r  rM   r*  r   r   r   Útest_convert_from_fp16_f32a  s    

z0TestBuildInstructions.test_convert_from_fp16_f32c              	   C   s\   | j dd}t |¡}t td¡}|  t¡}|j|dd W 5 Q R X |  dt	|j
ƒ¡ d S )Nr}   rÿ   ç      @ré   zexpected a float return type©rB   r   r~   rÉ   rv  r¦   rZ  rÓ  r#   r!   r$  rÃ  r   r   r   Ú!test_convert_from_fp16_f32_notypem  s    
þz7TestBuildInstructions.test_convert_from_fp16_f32_notypec              	   C   s^   | j dd}t |¡}t td¡}|  t¡}|j|dtd W 5 Q R X |  dt	|j
ƒ¡ d S )Nr}   rÿ   rÕ  ré   rÑ  zexpected an i16 type, got floatrÖ  rÃ  r   r   r   Ú$test_convert_from_fp16_f32_wrongtypex  s    
þz:TestBuildInstructions.test_convert_from_fp16_f32_wrongtypec              	   C   s^   | j dd}t |¡}t td¡}|  t¡}|j|dtd W 5 Q R X |  	dt
|jƒ¡ d S )Nr}   rÿ   rÕ  ré   rÑ  rÏ  )rB   r   r~   rÉ   rv  r¦   rZ  rÓ  rÊ  r#   r!   r$  rÃ  r   r   r   Ú%test_convert_from_fp16_f32_wrongtype2ƒ  s    
þz;TestBuildInstructions.test_convert_from_fp16_f32_wrongtype2c                 C   sX   | j dd}t |¡}t td¡}t td¡}|j||dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   r   z}            my_block:
                %"c" = call i64 @"llvm.cttz.i64"(i64 5, i1 1)
                ret i64 %"c"
            )	rB   r   r~   rÉ   rœ   rO  Úcttzr  rM   r5  r   r   r   Ú	test_cttzŽ  s    

zTestBuildInstructions.test_cttzc              	   C   sj   | j dd}t |¡}t td¡}t td¡}|  t¡}|j||dd W 5 Q R X |  	dt
|jƒ¡ d S )Nr}   rÿ   rg   rj   r   zexpected an i1 type, got i32)rB   r   r~   rÉ   rœ   r:   r¦   rZ  rÚ  r#   r!   r$  ©r   rB   r   rè   ré   rÄ  r   r   r   Útest_cttz_wrongflag›  s    
þz)TestBuildInstructions.test_cttz_wrongflagc              	   C   sj   | j dd}t |¡}t td¡}t td¡}|  t¡}|j||dd W 5 Q R X |  	dt
|jƒ¡ d S )Nr}   rÿ   rg   r   r   rÂ  )rB   r   r~   rÉ   rv  rO  r¦   rZ  rÚ  r#   r!   r$  rÜ  r   r   r   Útest_cttz_wrongtype§  s    
þz)TestBuildInstructions.test_cttz_wrongtypec                 C   sf   | j dd}t |¡}t td¡}t td¡}t td¡}|j|||dd}| |¡ |  |d¡ d S )Nr}   rÿ   rg   r   rS   ÚfmazÅ            my_block:
                %"fma" = call float @"llvm.fma.f32"(float 0x4014000000000000, float 0x3ff0000000000000, float 0x4000000000000000)
                ret float %"fma"
            )rB   r   r~   rÉ   rv  rß  r  rM   )r   rB   r   rè   ré   r   rß  r   r   r   Útest_fma³  s    

zTestBuildInstructions.test_fmac              	   C   sx   | j dd}t |¡}t td¡}t td¡}t td¡}|  t¡}|j|||dd W 5 Q R X |  dt	|j
ƒ¡ d S )Nr}   rÿ   rg   r   rS   rß  z(expected an floating point type, got i32)rB   r   r~   rÉ   r:   r¦   rZ  rß  r#   r!   r$  ©r   rB   r   rè   ré   r   rÄ  r   r   r   Útest_fma_wrongtypeÁ  s    
þz(TestBuildInstructions.test_fma_wrongtypec              	   C   sx   | j dd}t |¡}t td¡}t td¡}t td¡}|  t¡}|j|||dd W 5 Q R X |  	dt
|jƒ¡ d S )Nr}   rÿ   rg   r   rS   rß  z7expected types to be the same, got float, double, float)rB   r   r~   rÉ   rv  r;   r¦   rZ  rß  r#   r!   r$  rá  r   r   r   Útest_fma_mixedtypesÎ  s    
þz)TestBuildInstructions.test_fma_mixedtypesN)9rX   rY   rZ   r[   ZmaxDiffr
  r'  r+  r0  r1  r6  r8  rC  rN  rR  rT  r]  r`  rh  ri  rk  rm  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  ¶  sl   1%-1=!($	!:r  c                   @   sh   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )ÚTestBuilderMiscz=
    Test various other features of the IRBuilder class.
    c                 C   s\   | j dd}t |¡}|  |j|j¡ |  |jtj¡ |  |j|jj¡ |  |jtj	¡ d S )Nr÷   rÿ   )
rB   r   r~   r1   r?   rù   ÚassertIsInstancer=   r7   r5   ©r   rB   r   r   r   r   rþ   á  s    
zTestBuilderMisc.test_attributesc              
   C   sð   | j dd}t |¡}|jjd d… \}}| ||d¡ |jdd}| |¡N | ||d¡ | 	¡  | 
||d¡ W 5 Q R X | ||d¡ | |¡ W 5 Q R X | ||d	¡ | |¡ | ||d
¡ W 5 Q R X |  |d¡ |  |d¡ d S )Nr}   rÿ   rS   r   r£   rì   rí   rî   rï   rð   z¦            my_block:
                %"c" = add i32 %".1", %".2"
                %"e" = sub i32 %".1", %".2"
                %"g" = mul i32 %".1", %".2"
            zÄ            foo:
                %"d" = fadd i32 %".1", %".2"
                %"f" = fsub i32 %".1", %".2"
                %"h" = fmul i32 %".1", %".2"
                br label %"foo"
            )rB   r   r~   r?   rl   rd   r@   Z
goto_blockr  Zgoto_entry_blockr   r  rƒ  r  r  rM   )r   rB   r   rè   ré   Zbb_newr   r   r   Útest_goto_blocké  s     

zTestBuilderMisc.test_goto_blockc              	   C   s°   | j dd}t |¡}t td¡}| ||d¡}| |¡}| ||d¡ W 5 Q R X |  |j |¡ | ||d¡}| |¡ | ||d¡ | |¡ W 5 Q R X |  	|j
d¡ d S )	NÚonerÿ   r   rè   ré   r   rì   aÝ              one:
                %"a" = add i1 0, 0
                br i1 %"a", label %"one.if", label %"one.endif"
            one.if:
                %"b" = add i1 0, 0
                br label %"one.endif"
            one.endif:
                %"c" = add i1 0, 0
                br i1 %"c", label %"one.endif.if", label %"one.endif.endif"
            one.endif.if:
                %"d" = add i1 0, 0
                br label %"one"
            one.endif.endif:
            )rB   r   r~   rÉ   rO  rd   Úif_thenr1   rƒ  rW   r?   )r   rB   r   r[  rè   Zbbendr   r   r   r   Útest_if_then  s    
zTestBuilderMisc.test_if_thenc              
   C   s’   | j dd}t |¡}t td¡}| ||d¡}| |¡8 | ||d¡}| |¡ | ||d¡ W 5 Q R X W 5 Q R X | ¡  |  |j	d¡ d S )Nrè  rÿ   r   rè   ré   r   aÐ              one:
                %"a" = add i1 0, 0
                br i1 %"a", label %"one.if", label %"one.endif"
            one.if:
                %"b" = add i1 0, 0
                br i1 %"b", label %"one.if.if", label %"one.if.endif"
            one.endif:
                ret void
            one.if.if:
                %"c" = add i1 0, 0
                br label %"one.if.endif"
            one.if.endif:
                br label %"one.endif"
            )
rB   r   r~   rÉ   rO  rd   ré  r   rW   r?   )r   rB   r   r[  rè   ré   r   r   r   Útest_if_then_nested$  s    
"z#TestBuilderMisc.test_if_then_nestedc              
   C   s¬   d}| j |d}t |¡}t td¡}| ||d¡}| |¡8 | ||d¡}| |¡ | ||d¡ W 5 Q R X W 5 Q R X | ¡  |  |j	dj
||d d… d	 d
¡ d S )NZPLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongLongrÿ   r   rè   ré   r   a              {full_label}:
                %"a" = add i1 0, 0
                br i1 %"a", label %"{label}.if", label %"{label}.endif"
            {label}.if:
                %"b" = add i1 0, 0
                br i1 %"b", label %"{label}.if.if", label %"{label}.if.endif"
            {label}.endif:
                ret void
            {label}.if.if:
                %"c" = add i1 0, 0
                br label %"{label}.if.endif"
            {label}.if.endif:
                br label %"{label}.endif"
            é   z..)Ú
full_labelÚlabel)rB   r   r~   rÉ   rO  rd   ré  r   rW   r?   r€   )r   rí  rB   r   r[  rè   ré   r   r   r   Útest_if_then_long_label?  s    
" òz'TestBuilderMisc.test_if_then_long_labelc                    s<   ‡ fdd„}|dƒ}ˆ   |jd¡ |dƒ}ˆ   |jd¡ d S )Nc              	      sL   ˆ j dd}t |¡}t td¡}|j|| d W 5 Q R X ˆ  |d¡ |S )Nrè  rÿ   r   ©Úlikelyzp                one:
                    br i1 0, label %"one.if", label %"one.endif", !prof !0
                )rB   r   r~   rÉ   rO  ré  rM   )rñ  rB   r   r[  r6   r   r   rŽ  [  s    

z2TestBuilderMisc.test_if_then_likely.<locals>.checkTúC            !0 = !{ !"branch_weights", i32 99, i32 1 }
            FúC            !0 = !{ !"branch_weights", i32 1, i32 99 }
            ©rQ   r7   ©r   rŽ  r   r   r6   r   Útest_if_then_likelyZ  s
    z#TestBuilderMisc.test_if_then_likelyc              
   C   sä   | j dd}t |¡}t td¡}| ||d¡}| |¡F\}}| | ||d¡ W 5 Q R X | | ||d¡ W 5 Q R X W 5 Q R X | |¡<\}}| | |¡ W 5 Q R X | | ¡  W 5 Q R X W 5 Q R X |  	|j
d¡ d S )Nrè  rÿ   r   rè   ré   r   a)              one:
                %"a" = add i1 0, 0
                br i1 %"a", label %"one.if", label %"one.else"
            one.if:
                %"b" = add i1 0, 0
                br label %"one.endif"
            one.else:
                %"c" = add i1 0, 0
                br label %"one.endif"
            one.endif:
                br i1 %"a", label %"one.endif.if", label %"one.endif.else"
            one.endif.if:
                br label %"one"
            one.endif.else:
                ret void
            one.endif.endif:
            )rB   r   r~   rÉ   rO  rd   Úif_elserƒ  r   rW   r?   )r   rB   r   r[  rè   Úthenr’  r   r   r   Útest_if_elseo  s    
"zTestBuilderMisc.test_if_elsec                    s<   ‡ fdd„}|dƒ}ˆ   |jd¡ |dƒ}ˆ   |jd¡ d S )Nc              
      s„   ˆ j dd}t |¡}t td¡}|j|| d<\}}| | |¡ W 5 Q R X | | ¡  W 5 Q R X W 5 Q R X ˆ  |j	d¡ |S )Nrè  rÿ   r   rð  zý                one:
                    br i1 0, label %"one.if", label %"one.else", !prof !0
                one.if:
                    br label %"one"
                one.else:
                    ret void
                one.endif:
                )
rB   r   r~   rÉ   rO  r÷  rƒ  r   rW   r?   )rñ  rB   r   r[  rø  r’  r6   r   r   rŽ  ”  s    
	z2TestBuilderMisc.test_if_else_likely.<locals>.checkTrò  Fró  rô  rõ  r   r6   r   Útest_if_else_likely“  s
    z#TestBuilderMisc.test_if_else_likelyc           	      C   s  |   ¡ }t ¡ }t td¡}|jdd}|jdd}|jdd}| |¡ | ||d¡ | |¡ | ||d¡ | ||d¡ | |¡ | ||d	¡}| ||d
¡ | |¡ | ||d¡}| 	|¡ | ||d¡ | 
|¡ | ||d¡ |  |d¡ |  |d¡ |  |d¡ dS )zI
        Test IRBuilder.position_{before,after,at_start,at_end}.
        r   rè  rÿ   ÚtwoZthreerè   r  r  r  r  ré   r  r   z‰            one:
                %"a" = add i32 0, 0
                %"c" = add i32 0, 0
                %"b" = add i32 0, 0
            zÑ            two:
                %"o" = add i32 0, 0
                %"q" = add i32 0, 0
                %"p" = add i32 0, 0
                %"m" = add i32 0, 0
                %"n" = add i32 0, 0
            z            three:
            N)r?   r   r~   rÉ   r:   r@   Zposition_at_startrd   rº  Zposition_afterZposition_beforerM   )	r   rA   r   r[  Zbb_oneZbb_twoZbb_threer  ré   r   r   r   Útest_positioning°  s.    





z TestBuilderMisc.test_positioningc           	      C   s´   |   ¡ }t ¡ }|jdd}| |¡ t td¡}| ||d¡}| ¡ }|  	|j
¡ | |¡ |  |j
¡ | ||d¡}| ||d¡}| |¡ | ¡  |  	|j
¡ |  |d¡ d S )NÚentryrÿ   iÒ  rè   ré   r   zˆ            entry:
                %"a" = add i32 1234, 1234
                %"b" = mul i32 %"a", %"a"
                ret void
        )r?   r   r~   r@   rº  rÉ   r:   rd   r   rQ  rü   Úremoverû   r  rM   )	r   rA   r   Zblkr  rè   Zretvoidré   r   r   r   r   Útest_instruction_removalà  s     


z(TestBuilderMisc.test_instruction_removalc                 C   sH   | j dd}t |¡}|j g ¡|_|jt t¡dd |  	|d¡ d S )Nr}   rÿ   r   zN            my_block:
                %"c" = alloca i32*, !dbg !0
            )
rB   r   r~   r7   rs   Zdebug_metadatarW  r<   r:   rM   ræ  r   r   r   Útest_metadata÷  s
    
zTestBuilderMisc.test_metadataN)rX   rY   rZ   r[   rþ   rç  rê  rë  rï  rö  rù  rú  rü  rÿ  r   r   r   r   r   rä  Ü  s   $0rä  c                   @   sl   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )Ú	TestTypesc                 C   s"   t |tjƒr|j}q t |tjƒ S r'   )Ú
isinstancer   r<   rj  Ú	LabelType)r   Útyr   r   r   Úhas_logical_equality  s    zTestTypes.has_logical_equalityc                 C   s´   t  ¡ }t  ¡ t  ¡ t  tttf¡t  ttf¡t jttfddt  ttf¡ttttt	t  
td¡t  
t	d¡t  
t	d¡t  ttf¡t  ttf¡| d¡| d¡g}|dd„ |D ƒ7 }|S )	z3
        A bunch of mutually unequal types
        Trb   rg   ri   ZMyType1ZMyType2c                 S   s(   g | ] }t |tjtjfƒst |¡‘qS r   )r  r   r×   r  r<   )r   Útpr   r   r   r      s    ÿz,TestTypes.assorted_types.<locals>.<listcomp>)r   ÚContextr  r×   r9   rO  re   r:   rv  r;   r´  rd  Úget_identified_type)r   Úcontextr¯   r   r   r   Úassorted_types	  s0          
 
 
 ÿözTestTypes.assorted_typesc                 C   s6   |   ¡ }|D ]$}|  |¡}|  |¡r|  ||¡ qd S r'   )r
  r4   r  r3   )r   r¯   r  Znewtyr   r   r   rŒ      s
    

zTestTypes.test_picklingc                 C   sÂ   |   ¡ }t ||¡D ]8\}}||k	r|  ||k||f¡ |  ||k||f¡ q|D ]j}t |¡}|  |¡r”|  ||k||f¡ |  ||k||f¡ qR|  ||k||f¡ |  ||k||f¡ qRd S r'   )r
  Ú	itertoolsÚproductrû   rQ  Úcopyr  )r   r¯   rè   ré   r  r¤  r   r   r   Útest_comparisons'  s    

zTestTypes.test_comparisonsc              
   C   sx  |   ttƒd¡ |   tt d¡ƒd¡ |   ttƒd¡ |   ttƒd¡ |   tt ¡ ƒd¡ |   tt td¡ƒd¡ |   tt ttf¡ƒd	¡ |   tt tttf¡ƒd
¡ |   ttjtdddƒd¡ |   ttjttfddƒd¡ |   ttjtttfddƒd¡ |   tt 	t
¡ƒd¡ |   tt 	t 	t
¡¡ƒd¡ |   tt td¡ƒd¡ |   tt t 	t¡d¡ƒd¡ |   tt 	t td¡¡ƒd¡ |   tt tf¡ƒd¡ |   tt ttf¡ƒd¡ |   tt t 	t¡t t
tf¡f¡ƒd¡ |   ttjtfddƒd¡ |   ttjttfddƒd¡ t ¡ }| d¡}|   t|ƒd¡ | d¡}|   t|ƒd¡ | d ¡}|   t|ƒd!¡ d"S )#z:
        Test the string representation of types.
        Úi1é   Zi29rÁ   ÚdoubleÚvoidr   zi1 ()z
i1 (float)zi1 (float, double)Trb   zi1 (...)zi1 (float, ...)zi1 (float, double, ...)zi32*zi32**rg   z[5 x i1]z	[5 x i1*]z	[5 x i1]*z{i1}z{i1, float}z{i1*, {i32, i8}})Úpackedz<{i1}>z<{i1, float}>ÚMyTypez	%"MyType"zMyType\z%"MyType\5c"zMyType"z%"MyType\22"N)r3   r!   rO  r   rË   rv  r;   r×   r9   r<   r:   r´  rd  re   r  r  )r   r	  ÚmytypeZmytype1Zmytype2r   r   r   Útest_str7  sb    ÿÿÿÿÿÿ
 ÿþÿÿ


zTestTypes.test_strc                 C   s4   t | j|  ¡ ƒD ]}|  t|ƒtt |¡ƒ¡ qd S r'   )Úfilterr  r
  r3   Úhashr  )r   Útypr   r   r   Ú	test_hashd  s    zTestTypes.test_hashc              	      s\  ‡ fdd„}‡ fdd„}t  t¡}tdƒD ]}|||tƒ q*||ƒ t  td¡}tdƒD ]}|||tƒ qX||ƒ t  tt  ttf¡f¡}||dtƒ ||dt  ttf¡ƒ ˆ  t	¡ | 
t  td	¡¡ W 5 Q R X ||ƒ t  ¡ }t  |d
¡}| tt  ttf¡¡ ||dtƒ ||dt  ttf¡ƒ ˆ  t	¡ | 
t  td	¡¡ W 5 Q R X ||ƒ d S )Nc                    s"   |   t t|¡¡}ˆ  ||¡ d S r'   )r^  r   rÉ   r:   r3   )r  rñ   rK   rO   r6   r   r   Úcheck_constanti  s    z*TestTypes.test_gep.<locals>.check_constantc              	      s0   t  td¡}ˆ  t¡ |  |¡ W 5 Q R X d S )Ng      ð?)r   rÉ   r;   r¦   rZ  r^  )r  Úindexr6   r   r   Úcheck_index_typem  s    z,TestTypes.test_gep.<locals>.check_index_typerg   rj   r   r   rS   r  )r   r<   r;   Úranger´  rO  rd  re   r¦   Ú
IndexErrorr^  rÉ   r:   r  ZIdentifiedStructTypeÚset_body)r   r  r  r  rñ   r	  r   r6   r   r`  h  s0    
zTestTypes.test_gepc                    st   t  d¡‰‡ ‡fdd„}|tdƒ |tdƒ |tdƒ |t td¡dƒ |t td¡dƒ |t tt	t	f¡d	ƒ d S )
Nú%e-m:e-i64:64-f80:128-n8:16:32:64-S128c                    s   ˆ   |  ˆ¡|¡ d S r'   )r3   Úget_abi_size©r  rK   ©r   Útdr   r   rŽ    s    z&TestTypes.test_abi_size.<locals>.checkr   ri   r   rg   é   r   ©
r(   Úcreate_target_datare   r:   rœ   r   r´  rd  r;   rv  ©r   rŽ  r   r$  r   Útest_abi_sizeŒ  s    



zTestTypes.test_abi_sizec                    st   t  d¡‰‡ ‡fdd„}|tdƒ |tdƒ |tdƒ |t td¡dƒ |t td¡dƒ |t tt	t	f¡dƒ d S )Nr!  c                    s   ˆ   |  ˆ¡|¡ d S r'   )r#   Zget_abi_alignmentr#  r$  r   r   rŽ  ›  s    z+TestTypes.test_abi_alignment.<locals>.check)r   rS   ri   )ri   )r   rg   r'  r)  r   r$  r   Útest_abi_alignment˜  s    



zTestTypes.test_abi_alignmentc                 C   s„   t  ¡ }| d¡}t j|d}|  |j¡ |  |¡ t|ƒ}| t  	d¡t  	d¡t  
¡ ¡ |  |j¡ |  |¡ |  |t|ƒ¡ d S )Nr  ©r	  r   r	   )r   r  r  r5   rQ  Z	is_opaquer+   r!   r   rË   rÈ   rû   ÚassertNotEqual)r   r	  r  r7   Zoldstrr   r   r   Útest_identified_struct¤  s    


z TestTypes.test_identified_structc                 C   sD   t  ¡ }| d¡}t  d¡g|_t d¡}|  |j||dd¡ d S )Nr  r   r!  r,  ri   )	r   r  r  rË   Úelementsr(   r(  r3   r"  )r   r	  r  r%  r   r   r   Ú$test_target_data_non_default_context°  s
    

z.TestTypes.test_target_data_non_default_contextc                 C   s&   t  t  d¡d¡}|  t|ƒd¡ d S )Nr   r   z	<8 x i32>)r   r¹  rË   r3   r!   )r   r»  r   r   r   Útest_vector·  s    zTestTypes.test_vectorN)rX   rY   rZ   r  r
  rŒ   r  r  r  r`  r*  r+  r.  r0  r1  r   r   r   r   r    s   -$r  c                 C   s   t  t| ¡S r'   ra  )rñ   r   r   r   Úc32¼  s    r2  c                   @   sü   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd1d2„ Zd3d4„ Zd5d6„ Zd7d8„ Zd9d:„ Zd;d<„ Z d=S )>ÚTestConstantc                 C   sÊ   t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  td	¡}|  t|ƒd
¡ t  tt j¡}|  t|ƒd¡ t  td ¡}|  t|ƒd¡ d S )Nr˜   zi32 42r   zi1 1r   zi1 0Túi1 trueFzi1 falsezi1 undef)r   rÉ   r:   r3   r!   rO  r7  ©r   r   r   r   r   Útest_integersÂ  s    zTestConstant.test_integersc                 C   s®   t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  td¡}|  t|ƒd¡ t  tt j¡}|  t|ƒd¡ t  td ¡}|  t|ƒd¡ d S )	Nç      ø?zfloat 0x3ff8000000000000g      ø¿zfloat 0xbff8000000000000zdouble 0x3ff8000000000000zdouble 0xbff8000000000000zdouble undefz
double 0.0)r   rÉ   rv  r3   r!   r;   r7  r5  r   r   r   Ú
test_realsÒ  s    zTestConstant.test_realsc              	   C   sÐ  t  t  td¡tdƒtdƒtdƒf¡}|  t|ƒd¡ t  t  td¡tdƒtt jƒf¡}|  t|ƒd¡ t j tdƒtdƒtt jƒf¡}|  t|ƒd¡ |  	t
¡"}t j tdƒt  td	¡f¡ W 5 Q R X |  t|jƒd
¡ t  t  td¡t j¡}|  t|ƒd¡ t  t  td¡d ¡}|  t|ƒd¡ t  t  td¡tdƒ¡}|  t|ƒd¡ t  t  td¡tdƒ¡}|  t|ƒd¡ t  t  td¡dt jdf¡}|  t|ƒd¡ |  	t¡ t  t  td¡d¡ W 5 Q R X d S )Nrj   rg   r  ri   z[3 x i32] [i32 5, i32 6, i32 4]rS   z[2 x i32] [i32 5, i32 undef]z#[3 x i32] [i32 5, i32 6, i32 undef]r7  z$all elements must have the same typez[2 x i32] undefz[2 x i32] zeroinitializerrµ   s   foobar_123€z[11 x i8] c"foobar_123\80"s    ÿz[4 x i8] c"\00\01\04\ff"z#[3 x i32] [i32 5, i32 undef, i32 6])rg   r  )r   rÉ   r´  r:   r2  r3   r!   r7  Zliteral_arrayr¦   rZ  rv  r$  re   Ú	bytearrayr‘   )r   r   rÄ  r   r   r   Útest_arraysá  s.    &" &ÿzTestConstant.test_arraysc                 C   s^   t  t  d¡d¡}dddddddd	g}t  ||¡}d
 d tdj|ƒ¡¡}|  t|ƒ|¡ d S )Nr   r   r   rS   ri   rj   r  é	   é   z<8 x i32> <{}>z, zi32 {})	r   r¹  rË   rÉ   r€   r   r   r3   r!   )r   r»  Úvalsr¼  Zvec_reprr   r   r   r1  þ  s    ÿzTestConstant.test_vectorc                 C   s$   t  t  d¡d ¡j}|  |d¡ d S )Nr   r   )r   rÉ   rË   Úconstantr3   )r   r>  r   r   r   Útest_non_nullable_int	  s    z"TestConstant.test_non_nullable_intc              	   C   s’  t  ttf¡}t  t|f¡}t  |t  t  ¡ d¡t  td¡f¡}|  t|ƒd¡ t j 	t  t  ¡ d¡t  td¡f¡}|  |j
|¡ |  t|ƒd¡ t j 	t  t  ¡ d¡t  tt j¡f¡}|  |j
|¡ |  t|ƒd¡ t  |t j¡}|  t|ƒd¡ t  |d ¡}|  t|ƒd¡ t  |d¡}|  t|ƒd¡ t  |d|f¡}|  t|ƒd	¡ t  |d
¡}|  t|ƒt|ƒ¡ |  t¡ t  |d¡ W 5 Q R X d S )Nr7  Tú/{float, i1} {float 0x3ff8000000000000, i1 true}z0{float, i1} {float 0x3ff8000000000000, i1 undef}z{float, i1} undefz{float, i1} zeroinitializer©r7  Tr˜   úL{i32, {float, i1}} {i32 42, {float, i1} {float 0x3ff8000000000000, i1 true}}©r˜   rA  )ri   rg   r  )r   rd  rv  rO  r:   rÉ   rÈ   r3   r!   Zliteral_structr¶   r7  r¦   r‘   )r   Úst1Úst2r   Zc1Úc2Úc3r   r   r   Útest_structs
	  sH    
ÿ
ÿ
ÿ
ÿÿ
ÿ
ÿzTestConstant.test_structsc                 C   s0   t  d¡}t  t  ||g¡t j¡}|  |¡ d S )Nr   )r   rË   rÉ   rd  r7  r4   )r   Úi8Ústr   r   r   Ú&test_undefined_literal_struct_pickling,	  s    
z3TestConstant.test_undefined_literal_struct_picklingc              	   C   s.  t dƒ}|  |tj¡ |  t|ƒd¡ tdƒ}|  |tj¡ |  t|ƒd¡ t td¡}|t	dƒt	dƒt	dƒgƒ}|  t|ƒd	¡ |dddgƒ}|  t|ƒd	¡ |d
ƒ}|  t|ƒd¡ |  
t¡ |ddddgƒ W 5 Q R X t ttf¡}t t|f¡}|dƒ}|  t|ƒd¡ |dƒ}|  t|ƒd¡ d
S )z@
        Instantiating a type should create a constant.
        r˜   zi8 42Tr4  rj   ri   rg   r  z[3 x i32] [i32 4, i32 5, i32 6]Nz[3 x i32] zeroinitializerr<  rA  r@  rC  rB  )re   rå  r   rÉ   r3   r!   rO  r´  r:   r2  r¦   r‘   rd  rv  )r   r   ÚatrD  rE  r   r   r   Útest_type_instantiaton1	  s*    z#TestConstant.test_type_instantiatonc                 C   s   t dƒ}|  t|ƒd¡ dS )z8
        Constants should have a useful repr().
        r˜   z!<ir.Constant type='i32' value=42>N)r:   r3   rä   r5  r   r   r   r  O	  s    zTestConstant.test_reprc                 C   sz   t  t  t  d¡d¡ttdƒƒ¡}|  ¡ }t  ||jd¡}d|_	||_
t t|ƒ¡}t t|ƒ¡}|  t|ƒt|ƒ¡ d S )Nr   é   Ú
myconstantT)r   rÉ   r´  rË   r9  r  r7   rÇ   r¶   rò   rÊ   r(   r)   r!   r3   )r   r   r  ÚgvÚparsedZreparsedr   r   r   Útest_encoding_problemV	  s    
ÿz"TestConstant.test_encoding_problemc              	   C   sÞ   |   ¡ }t ttf¡}t ||d¡}| dd„ dD ƒ¡}|  t|ƒd¡ |  |j	t 
t¡¡ t |d ¡}|  t¡ | t td¡g¡ W 5 Q R X t | ¡ d ¡}| t td¡g¡}|  t|ƒd¡ |  |j	t 
t¡¡ d S )NrO  c                 S   s   g | ]}t  t|¡‘qS r   ra  rb  r   r   r   r    i	  s     z)TestConstant.test_gep.<locals>.<listcomp>©r   r   zEgetelementptr ({float, i1}, {float, i1}* @"myconstant", i32 0, i32 1)r   z5getelementptr ({float, i1}, {float, i1}* null, i32 0))r7   r   rd  rv  rO  rÇ   r^  r3   r!   r¶   r<   rÉ   r¦   rZ  r:   r„   )r   r  r  rP  r   ÚconstZ	const_ptrrF  r   r   r   r`  e	  s"    
ÿ
ÿzTestConstant.test_gepc                 C   sŠ   |   ¡ }t ttf¡}d}tj||d|d}|  |j|¡ | dd„ dD ƒ¡}|  |j	j|¡ |  t
|ƒd¡ |  |j	tjt|d¡ d S )Nri   rO  rê   c                 S   s   g | ]}t  t|¡‘qS r   ra  rb  r   r   r   r    	  s     z=TestConstant.test_gep_addrspace_globalvar.<locals>.<listcomp>rS  zRgetelementptr ({float, i1}, {float, i1} addrspace(4)* @"myconstant", i32 0, i32 1))r7   r   rd  rv  rO  rÇ   r3   rë   r^  r¶   r!   r<   )r   r  r  rë   rP  r   r   r   r   Útest_gep_addrspace_globalvarx	  s    
ÿz)TestConstant.test_gep_addrspace_globalvarc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   ztrunc (i64 1 to i32))r   rÉ   rœ   rr  r:   r3   r!   r5  r   r   r   Ú
test_trunc†	  s    zTestConstant.test_truncc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   zzext (i32 1 to i64))r   rÉ   r:   rs  rœ   r3   r!   r5  r   r   r   Ú	test_zextŠ	  s    zTestConstant.test_zextc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr,   zsext (i32 -1 to i64))r   rÉ   r:   rt  rœ   r3   r!   r5  r   r   r   Ú	test_sextŽ	  s    zTestConstant.test_sextc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   z*fptrunc (float 0x3ff0000000000000 to half))r   rÉ   rv  ru  r  r3   r!   r5  r   r   r   Útest_fptrunc’	  s    zTestConstant.test_fptruncc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   z*fpext (float 0x3ff0000000000000 to double))r   rÉ   rv  rw  r;   r3   r!   r5  r   r   r   Ú
test_fpext–	  s    zTestConstant.test_fpextc                 C   s8   |   ¡ }t |td¡}| t ¡ ¡}|  t|ƒd¡ d S )NrO  z$bitcast (i32* @"myconstant" to i64*))	r7   r   rÇ   r:   re  rœ   r„   r3   r!   )r   r  rP  r   r   r   r   Útest_bitcastš	  s    zTestConstant.test_bitcastc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   z(fptoui (float 0x3ff0000000000000 to i32))r   rÉ   rv  rx  r:   r3   r!   r5  r   r   r   Útest_fptoui 	  s    zTestConstant.test_fptouic                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   zuitofp (i32 1 to float))r   rÉ   r:   rz  rv  r3   r!   r5  r   r   r   Útest_uitofp¤	  s    zTestConstant.test_uitofpc                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   z(fptosi (float 0x3ff0000000000000 to i32))r   rÉ   rv  ry  r:   r3   r!   r5  r   r   r   Útest_fptosi¨	  s    zTestConstant.test_fptosic                 C   s&   t  td¡ t¡}|  t|ƒd¡ d S )Nr   zsitofp (i32 1 to float))r   rÉ   r:   r{  rv  r3   r!   r5  r   r   r   Útest_sitofp¬	  s    zTestConstant.test_sitofpc                 C   sZ   t  t ¡ d ¡}t  td¡}| t¡}|  t|jt¡ |  t|jt¡ |  	t
|ƒd¡ d S )Nr   zptrtoint (i64* null to i32))r   rÉ   rœ   r„   r:   r|  r¦   rZ  rv  r3   r!   )r   rl  rè  r   r   r   r   Útest_ptrtoint_1°	  s    
zTestConstant.test_ptrtoint_1c                 C   sh   |   ¡ }t |td¡}| t¡}|  t|ƒd¡ |  t	d|jt 
¡ ¡ t td¡}|  t	d|jt¡ d S )NrO  z$ptrtoint (i32* @"myconstant" to i64)z2can only ptrtoint\(\) to integer type, not 'i64\*'r   z5can only call ptrtoint\(\) on pointer type, not 'i32')r7   r   rÇ   r:   r|  rœ   r3   r!   r   rZ  r„   rÉ   )r   r  rP  r   rF  r   r   r   Útest_ptrtoint_2¹	  s"    
üüzTestConstant.test_ptrtoint_2c                 C   s^   t  td¡}t  td¡}| t ¡ ¡}|  t|jt¡ |  t|jt ¡ ¡ |  	t
|ƒd¡ d S )Nr   g…ëQ¸	@zinttoptr (i32 1 to i64*))r   rÉ   r:   rv  r}  rœ   r„   r¦   rZ  r3   r!   )r   rè  Úpir   r   r   r   Útest_inttoptrÌ	  s    zTestConstant.test_inttoptrc                 C   s$   t  td¡}|  t| ¡ ƒd¡ d S )Nr   zsub (i32 0, i32 1))r   rÉ   r:   r3   r!   r2  ©r   rè  r   r   r   Útest_negÕ	  s    zTestConstant.test_negc                 C   s$   t  td¡}|  t| ¡ ƒd¡ d S )Nr   zxor (i32 1, i32 -1))r   rÉ   r:   r3   r!   r3  rd  r   r   r   Útest_notÙ	  s    zTestConstant.test_notc                 C   s$   t  td¡}|  t| ¡ ƒd¡ d S )Nr   zfneg (float 0x3ff0000000000000))r   rÉ   rv  r3   r!   r4  rd  r   r   r   Ú	test_fnegÝ	  s    zTestConstant.test_fnegc           
      C   s0  t  td¡}t  td¡}|jd|jd|jd|jd|jd|jd|j	d	|j
d
|jd|jd|jd|jd|jdi}| ¡ D ] \}}|  t||ƒƒ|d ¡ qrdddddddœ}| ¡ D ]0\}}t| ||¡ƒ}d| d }	|  ||	¡ q®dddddddœ}| ¡ D ]0\}}t| ||¡ƒ}d| d }	|  ||	¡ qúd S )Nr   rS   r!  r#  r"  rd   r   r  r  r  r  r  ÚorÚandr   z (i32 1, i32 2)ÚeqÚneÚugtÚugeÚultÚule©r9  r:  r=  r>  r;  r<  zicmp ZsgtZsgeZsltZsle)r   rÉ   r:   r!  r#  r"  rd   r   r  r  r  r  r  r  r  r   Úitemsr3   r!   rA  rB  )
r   rè  rû  Úoracler‹   ÚiropÚcopÚcondrO   rK   r   r   r   Útest_int_binopsá	  s\                    ü   ÿ   ÿzTestConstant.test_int_binopsc           
   
   C   s  t  td¡}t  td¡}|jd|jd|jd|jd|jdi}| ¡ D ](\}}t	||ƒƒ}|d }|  
||¡ qBd	d
dddddœ}| ¡ D ]0\}}	t	| ||¡ƒ}d|	 d }|  
||¡ q†dddddddœ}| ¡ D ]0\}}	t	| ||¡ƒ}d|	 d }|  
||¡ qÒd S )Nr   rS   r  r  r  r  r  z5 (float 0x3ff0000000000000, float 0x4000000000000000)Zoeqrè  ZogtZogerH  Zolerp  zfcmp ZueqZunerl  rm  rn  ro  )r   rÉ   rv  r  r  r  r  r  rq  r!   r3   rL  rM  )
r   rè  rû  rr  r‹   rs  rO   rK   rt  ru  r   r   r   Útest_flt_binopsý	  s8       ÿ ÿ ÿzTestConstant.test_flt_binopsN)!rX   rY   rZ   r6  r8  r:  r1  r?  rH  rK  rM  r  rR  r`  rU  rV  rW  rX  rY  rZ  r[  r\  r]  r^  r_  r`  ra  rc  re  rf  rg  rv  rw  r   r   r   r   r3  À  s<   "		r3  c                   @   s   e Zd Zdd„ ZdS )ÚTestTransformsc                 C   s¦   t  ¡ }t  |t  t  ¡ d¡d¡}t  |t  t  ¡ d¡d¡}t  ¡ }| | ¡ ¡ | |d¡}|  	|j
|¡ t  |||¡}|  ||¡ |  |j
|¡ |  	|j
|¡ d S )Nr   r£   r¤   )r   r5   r=   r9   r×   r~   rº  r@   rØ   r3   ZcalleeZreplace_all_callsr#   r-  )r   r$   r£   r¤   r   rØ   Úmodifiedr   r   r   Útest_call_transform
  s    z"TestTransforms.test_call_transformN)rX   rY   rZ   rz  r   r   r   r   rx  
  s   rx  c                   @   s   e Zd Zdd„ ZdS )ÚTestSingletonc                 C   sP   |   tjtj ¡ ¡ |   tjt tj¡¡ |   tjt tj¡¡ |  tj¡ d S r'   )r1   r   r7  ÚvaluesZ
_Undefinedr  Údeepcopyr4   r6   r   r   r   Útest_undefined,
  s    zTestSingleton.test_undefinedN)rX   rY   rZ   r~  r   r   r   r   r{  +
  s   r{  Ú__main__)'r[   r  r  r.   r   rF   Zunittestr   r   Zllvmliter   r   r(   rË   rO  re   rÊ  r:   rœ   ZHalfTyper  rÈ   rv  rã   r;   r
   r\   r–   rá   rö   r  rä  r  r2  r3  rx  r{  rX   Úmainr   r   r   r   Ú<module>   s^   




W 4  L-        .  ( ;  ^
