o
    ٴi?                    @   s   d dl Z d dlZd dl mZmZmZ d dlm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Zd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZ e
 d
krged eejdkdej Zeejdkdej Z dZ!zd dl"Z"dZ!W n	 e#y   Y nw ee!dZ$dd Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,e
- dkre,.ddZ,dZ/d Z0d!Z1d"Z2d#Z3d$Z4d%Z5d&Z6d'Z7d(Z8d)Z9d*Z:d+Z;d,Z<g d-Z=g d.Z>g d/Z?d0Z@d1ZAd2ZBd3ZCd4ZDd5ZEd6ZFd7ZGd8ZHG d9d: d:eZIG d;d< d<eIZJG d=d> d>eIZKG d?d@ d@eIZLG dAdB dBeIZMG dCdD dDeNZOG dEdF dFeOZPG dGdH dHeIePZQeRdIG dJdK dKeIZSG dLdM dMeIZTG dNdO dOeIZUG dPdQ dQeIZVG dRdS dSeIZWG dTdU dUeIZXG dVdW dWeIZYG dXdY dYeIZZG dZd[ d[eIZ[G d\d] d]eQZ\G d^d_ d_eIZ]ee
 ^d`daG dbdc dceIZ_G ddde deeIZ`G dfdg dgeIZaG dhdi dieIZbG djdk dkeNZcG dldm dmeIecZdG dndo doeIecZeG dpdq dqeIecZfee	jghdrdse$G dtdu dueZiejdvkrek  dS dS )w    N)	CFUNCTYPEc_intc_int32)find_library)contextmanager)mkstemp)ir)binding)ffi)TestCasearmv7lzlibgcc_s.so.1condazconda package test only, have wheelzwheel package test only, have FTztest needs py-lief packagec               
   C   sl   t t j} z'z	t t jd W n t jy#   Y W t t j|  dS w W t t j|  dS t t j|  w )Nde_DETF)locale	setlocaleLC_ALLError)cur r   N/home/kuhnn/.local/lib/python3.10/site-packages/llvmlite/tests/test_binding.pyno_de_locale,   s   "r   a  
    ; ModuleID = '<string>'
    source_filename = "asm_sum.c"
    target triple = "{triple}"
    %struct.glob_type = type {{ i64, [2 x i64]}}
    %struct.glob_type_vec = type {{ i64, <2 x i64>}}

    @glob = global i32 0
    @glob_b = global i8 0
    @glob_f = global float 1.5
    @glob_struct = global %struct.glob_type {{i64 0, [2 x i64] [i64 0, i64 0]}}

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = add i32 %.1, %.2
      %.4 = add i32 0, %.3
      ret i32 %.4
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = add i32 %.1, %.2
      ret i32 %.3
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i64 @sum(i64 %.1, i64 %.2) {{
      %.3 = add i64 %.1, %.2
      %.4 = add i64 5, %.3
      %.5 = add i64 -5, %.4
      ret i64 %.5
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i32 @sum(i32 %.1, i32 %.2) {{
        %.3 = add i32 %.1, %.2
        ret i32 0
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"
    @mul_glob = global i32 0

    define i32 @mul(i32 %.1, i32 %.2) {{
      %.3 = mul i32 %.1, %.2
      ret i32 %.3
    }}
    a  
    ; ModuleID = '<string>'
    target triple = "{triple}"
    @mul_glob = global i32 0

    declare i32 @sum(i32, i32)
    define i32 @square_sum(i32 %.1, i32 %.2) {{
      %.3 = call i32 @sum(i32 %.1, i32 %.2)
      %.4 = mul i32 %.3, %.3
      ret i32 %.4
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    declare i8* @Py_GetVersion()

    define void @getversion(i32 %.1, i32 %.2) {{
      %1 = call i8* @Py_GetVersion()
      ret void
    }}
    PyPyPy_GetVersionPyPy_GetVersionz
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define i32 @sum(i32 %.1, i32 %.2) {{
      %.3 = fadd i32 %.1, %.2
      ret i32 %.3
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @sum() {{
      %.bug = add i32 1, %.bug
      ret void
    }}
    zh
    ; ModuleID = '<string>'
    target triple = "{triple}"

    declare i32 @sum(i32 %.1, i32 %.2)
    zg
    ; ModuleID = '<string>'
    target triple = "{triple}"

    declare i32 @vararg(i32 %.1, ...)
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo() {{
      %const = fadd fp128 0xLF3CB1CCF26FBC178452FB4EC7F91DEAD, 0xL00000000000000000000000000000001
      ret void
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo() {{
      %const = fadd double 0.0, 3.14
      ret void
    }}
    z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo() {{
      call void asm sideeffect "nop", ""()
      ret void
    }}
    a  
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @inlineme() {{
        ret void
    }}

    define i32 @caller(i32 %.1, i32 %.2) {{
    entry:
      %stack = alloca i32
      store i32 %.1, i32* %stack
      br label %main
    main:
      %loaded = load i32, i32* %stack
      %.3 = add i32 %loaded, %.2
      %.4 = add i32 0, %.3
      call void @inlineme()
      ret i32 %.4
    }}
a  
; ModuleID = 'test.c'
source_filename = "test.c"
target triple = "{triple}"

; Function Attrs: noinline nounwind optnone ssp uwtable
define void @inlineme() noinline !dbg !15 {{
  ret void, !dbg !18
}}

; Function Attrs: noinline nounwind optnone ssp uwtable
define i32 @foo(i32 %0, i32 %1) !dbg !19 {{
  %3 = alloca i32, align 4
  %4 = alloca i32, align 4
  store i32 %0, i32* %3, align 4
  call void @llvm.dbg.declare(metadata i32* %3, metadata !23, metadata !DIExpression()), !dbg !24
  store i32 %1, i32* %4, align 4
  call void @llvm.dbg.declare(metadata i32* %4, metadata !25, metadata !DIExpression()), !dbg !26
  call void @inlineme(), !dbg !27
  %5 = load i32, i32* %3, align 4, !dbg !28
  %6 = load i32, i32* %4, align 4, !dbg !29
  %7 = add nsw i32 %5, %6, !dbg !30
  ret i32 %7, !dbg !31
}}

; Function Attrs: nofree nosync nounwind readnone speculatable willreturn
declare void @llvm.dbg.declare(metadata, metadata, metadata) #1

attributes #1 = {{ nofree nosync nounwind readnone speculatable willreturn }}

!llvm.module.flags = !{{!1, !2, !3, !4, !5, !6, !7, !8, !9, !10}}
!llvm.dbg.cu = !{{!11}}
!llvm.ident = !{{!14}}

!0 = !{{i32 2, !"SDK Version", [2 x i32] [i32 12, i32 3]}}
!1 = !{{i32 7, !"Dwarf Version", i32 4}}
!2 = !{{i32 2, !"Debug Info Version", i32 3}}
!3 = !{{i32 1, !"wchar_size", i32 4}}
!4 = !{{i32 1, !"branch-target-enforcement", i32 0}}
!5 = !{{i32 1, !"sign-return-address", i32 0}}
!6 = !{{i32 1, !"sign-return-address-all", i32 0}}
!7 = !{{i32 1, !"sign-return-address-with-bkey", i32 0}}
!8 = !{{i32 7, !"PIC Level", i32 2}}
!9 = !{{i32 7, !"uwtable", i32 1}}
!10 = !{{i32 7, !"frame-pointer", i32 1}}
!11 = distinct !DICompileUnit(language: DW_LANG_C99, file: !12, producer: "Apple clang version 13.1.6 (clang-1316.0.21.2.3)", isOptimized: false, runtimeVersion: 0, emissionKind: FullDebug, enums: !13, splitDebugInlining: false, nameTableKind: None, sysroot: "/Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/MacOSX.sdk", sdk: "MacOSX.sdk")
!12 = !DIFile(filename: "test.c", directory: "/")
!13 = !{{}}
!14 = !{{!"Apple clang version 13.1.6 (clang-1316.0.21.2.3)"}}
!15 = distinct !DISubprogram(name: "inlineme", scope: !12, file: !12, line: 1, type: !16, scopeLine: 1, flags: DIFlagPrototyped, spFlags: DISPFlagDefinition, unit: !11, retainedNodes: !13)
!16 = !DISubroutineType(types: !17)
!17 = !{{null}}
!18 = !DILocation(line: 1, column: 22, scope: !15)
!19 = distinct !DISubprogram(name: "foo", scope: !12, file: !12, line: 3, type: !20, scopeLine: 3, flags: DIFlagPrototyped, spFlags: DISPFlagDefinition, unit: !11, retainedNodes: !13)
!20 = !DISubroutineType(types: !21)
!21 = !{{!22, !22, !22}}
!22 = !DIBasicType(name: "int", size: 32, encoding: DW_ATE_signed)
!23 = !DILocalVariable(name: "a", arg: 1, scope: !19, file: !12, line: 3, type: !22)
!24 = !DILocation(line: 3, column: 13, scope: !19)
!25 = !DILocalVariable(name: "b", arg: 2, scope: !19, file: !12, line: 3, type: !22)
!26 = !DILocation(line: 3, column: 20, scope: !19)
!27 = !DILocation(line: 4, column: 5, scope: !19)
!28 = !DILocation(line: 5, column: 12, scope: !19)
!29 = !DILocation(line: 5, column: 16, scope: !19)
!30 = !DILocation(line: 5, column: 14, scope: !19)
!31 = !DILocation(line: 5, column: 5, scope: !19)
aN  
; ModuleID = "<string>"
target triple = "{triple}"

define double @licm(i32 %0) {{
  %2 = alloca i32, align 4
  %3 = alloca double, align 8
  %4 = alloca i32, align 4
  %5 = alloca double, align 8
  store i32 %0, i32* %2, align 4
  store double 0.000000e+00, double* %3, align 8
  store i32 0, i32* %4, align 4
  br label %6

6:                                                ; preds = %14, %1
  %7 = load i32, i32* %4, align 4
  %8 = load i32, i32* %2, align 4
  %9 = icmp slt i32 %7, %8
  br i1 %9, label %10, label %17

10:                                               ; preds = %6
  store double 7.000000e+00, double* %5, align 8
  %11 = load double, double* %5, align 8
  %12 = load double, double* %3, align 8
  %13 = fadd double %12, %11
  store double %13, double* %3, align 8
  br label %14

14:                                               ; preds = %10
  %15 = load i32, i32* %4, align 4
  %16 = add nsw i32 %15, 1
  store i32 %16, i32* %4, align 4
  br label %6

17:                                               ; preds = %6
  %18 = load double, double* %3, align 8
  ret double %18
}}
a  
    ; ModuleID = "<string>"
    target triple = "{triple}"

    @A = global i32 undef

    define void @ctor_A()
    {{
      store i32 10, i32* @A
      ret void
    }}

    define void @dtor_A()
    {{
      store i32 20, i32* @A
      ret void
    }}

    define i32 @foo()
    {{
      %.2 = load i32, i32* @A
      %.3 = add i32 %.2, 2
      ret i32 %.3
    }}

    @llvm.global_ctors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @ctor_A, i8* null}}]
    @llvm.global_dtors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @dtor_A, i8* null}}]
    a  
    ; ModuleID = "<string>"
    target triple = "{triple}"

    @A = external global i32

    define void @ctor_A()
    {{
      store i32 10, i32* @A
      ret void
    }}

    define void @dtor_A()
    {{
      store i32 20, i32* @A
      ret void
    }}

    define i32 @foo()
    {{
      %.2 = load i32, i32* @A
      %.3 = add i32 %.2, 2
      ret i32 %.3
    }}

    @llvm.global_ctors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @ctor_A, i8* null}}]
    @llvm.global_dtors = appending global [1 x {{i32, void ()*, i8*}}] [{{i32, void ()*, i8*}} {{i32 0, void ()* @dtor_A, i8* null}}]
    z; ModuleID = ""
target triple = "unknown-unknown-unknown"
target datalayout = ""

define i32 @"foo"()
{
"<>!*''#":
  ret i32 12345
}
z
    ; ModuleID = '<string>'
    target triple = "{triple}"

    define void @foo(i64* %.1) {{
      ret void
    }}

    define void @bar() {{
      call void @foo(i64* null)
      ret void
    }}
)addi	sp, sp, -16zsw	a1, 8(sp)zsw	a2, 12(sp)fld	fa5, 8(sp)zfmv.w.x	fa4, a0zfcvt.d.s	fa4, fa4fadd.d	fa5, fa4, fa5fsd	fa5, 8(sp)lw	a0, 8(sp)lw	a1, 12(sp)addi	sp, sp, 16ret)r   zsw	a0, 8(sp)zsw	a1, 12(sp)r   zfcvt.d.s	fa4, fa0r   r   r   r    r!   r"   )zfcvt.d.s	fa5, fa0zfadd.d	fa0, fa5, fa1r"   zc
declare void @a_readonly_func(i8 *) readonly

declare i8* @a_arg0_return_func(i8* returned, i32*)
a*  
define double @foo(i32 %i, double %j) optnone noinline {
    %I = alloca i32		; <i32*> [#uses=4]
    %J = alloca double		; <double*> [#uses=2]
    store i32 %i, i32* %I
    store double %j, double* %J
    %t1 = load i32, i32* %I		; <i32> [#uses=1]
    %t2 = add i32 %t1, 1		; <i32> [#uses=1]
    store i32 %t2, i32* %I
    %t3 = load i32, i32* %I		; <i32> [#uses=1]
    %t4 = sitofp i32 %t3 to double		; <double> [#uses=1]
    %t5 = load double, double* %J		; <double> [#uses=1]
    %t6 = fmul double %t4, %t5		; <double> [#uses=1]
    ret double %t6
}
z#
declare void @test_declare(i32* )
@  7f454c4602010100000000000000000001003e000100000000000000000000000000000000000000e0000000000000000000000040000000000040000500010048c1e2204809c24889d048c1c03d4831d048b90120000480001070480fafc8000000000000000000000000000000000000000000000000002f0000000400f1ff00000000000000000000000000000000070000001200020000000000000000001f00000000000000002e74657874005f5f617279626f002e6e6f74652e474e552d737461636b002e737472746162002e73796d746162003c737472696e673e00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001f0000000300000000000000000000000000000000000000a800000000000000380000000000000000000000000000000100000000000000000000000000000001000000010000000600000000000000000000000000000040000000000000001f000000000000000000000000000000100000000000000000000000000000000f00000001000000000000000000000000000000000000005f00000000000000000000000000000000000000000000000100000000000000000000000000000027000000020000000000000000000000000000000000000060000000000000004800000000000000010000000200000008000000000000001800000000000000>48c1e2204809c24889d048c1c03d4831d048b90120000480001070480fafc8ay  
; ModuleID = '<lambda>'
source_filename = "<string>"
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-pc-windows-msvc"

declare float @llvm.exp2.f32(float %casted)

define float @foo(i16 %arg) {
entry:
  %casted = sitofp i16 %arg to float
  %ret = call float @llvm.exp2.f32(float %casted)
  ret float %ret
}
a  
; ModuleID = '<string>'
target triple = "{triple}"

define void @foo(i32 %N) {{
  ; unnamed block for testing
  %cmp4 = icmp sgt i32 %N, 0
  br i1 %cmp4, label %for.body, label %for.cond.cleanup

for.cond.cleanup:
  ret void

for.body:
  %i.05 = phi i32 [ %inc, %for.body ], [ 0, %0 ]
  %inc = add nuw nsw i32 %i.05, 1
  %exitcond.not = icmp eq i32 %inc, %N
  br i1 %exitcond.not, label %for.cond.cleanup, label %for.body
}}
aw  
; Source C++
;-----------------------------------------
; class MyClass;
;
; class MyClassDefined{
;     MyClass *member;
;     MyClass *m2;
;     MyClass *m3;
; };
;
; void foo(MyClass *c, MyClassDefined){ }
;-----------------------------------------
; LLVM-IR by: clang -arch arm64 -S -emit-llvm file.cpp
; ModuleID = 'file.cpp'
source_filename = "class.cpp"
target datalayout = "e-m:o-i64:64-i128:128-n32:64-S128"
target triple = "arm64-apple-macosx13.3.0"

%class.MyClass = type opaque
%class.MyClassDefined = type { %class.MyClass*, %class.MyClass*, %class.MyClass* }

; Function Attrs: noinline nounwind optnone ssp uwtable(sync)
define void @_Z3fooP7MyClass14MyClassDefined(%class.MyClass* noundef %0, %class.MyClassDefined* noundef %1) {
  %3 = alloca %class.MyClass*, align 8
  store %class.MyClass* %0, %class.MyClass** %3, align 8
  ret void
}

!llvm.module.flags = !{!0, !1, !2, !3, !4, !5, !6, !7, !8}
!llvm.ident = !{!9}

!0 = !{i32 2, !"SDK Version", [2 x i32] [i32 13, i32 3]}
!1 = !{i32 1, !"wchar_size", i32 4}
!2 = !{i32 8, !"branch-target-enforcement", i32 0}
!3 = !{i32 8, !"sign-return-address", i32 0}
!4 = !{i32 8, !"sign-return-address-all", i32 0}
!5 = !{i32 8, !"sign-return-address-with-bkey", i32 0}
!6 = !{i32 7, !"PIC Level", i32 2}
!7 = !{i32 7, !"uwtable", i32 1}
!8 = !{i32 7, !"frame-pointer", i32 1}
!9 = !{!"Apple clang version 14.0.3 (clang-1403.0.22.14.1)"}

a  ; Source C++
;-----------------------------------------

; struct Vector2D{
;     float x, y;
; };
;
; void foo(Vector2D vec, Vector2D *out) {
;     *out = vec;
; }
;-----------------------------------------
; LLVM-IR by: clang -arch x86_64 -S -emit-llvm file.cpp
; ModuleID = 'file.cpp'
source_filename = "class.cpp"
target datalayout = "e-m:o-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-apple-macosx13.3.0"

%struct.Vector2D = type { float, float }

; Function Attrs: noinline nounwind optnone ssp uwtable
define void @_Z3foo8Vector2DPS_(<2 x float> %0, %struct.Vector2D* noundef %1) #0 {
  %3 = alloca %struct.Vector2D, align 4
  %4 = alloca %struct.Vector2D*, align 8
  %5 = bitcast %struct.Vector2D* %3 to <2 x float>*
  store <2 x float> %0, <2 x float>* %5, align 4
  store %struct.Vector2D* %1, %struct.Vector2D** %4, align 8
  %6 = load %struct.Vector2D*, %struct.Vector2D** %4, align 8
  %7 = bitcast %struct.Vector2D* %6 to i8*
  %8 = bitcast %struct.Vector2D* %3 to i8*
  call void @llvm.memcpy.p0i8.p0i8.i64(i8* align 4 %7, i8* align 4 %8, i64 8, i1 false)
  ret void
}

; Function Attrs: argmemonly nofree nounwind willreturn
declare void @llvm.memcpy.p0i8.p0i8.i64(i8* noalias nocapture writeonly, i8* noalias nocapture readonly, i64, i1 immarg) #1

attributes #0 = { noinline nounwind optnone ssp uwtable "darwin-stkchk-strong-link" "frame-pointer"="all" "min-legal-vector-width"="64" "no-trapping-math"="true" "probe-stack"="___chkstk_darwin" "stack-protector-buffer-size"="8" "target-cpu"="penryn" "target-features"="+cx16,+cx8,+fxsr,+mmx,+sahf,+sse,+sse2,+sse3,+sse4.1,+ssse3,+x87" "tune-cpu"="generic" }
attributes #1 = { argmemonly nofree nounwind willreturn }

!llvm.module.flags = !{!0, !1, !2, !3, !4}
!llvm.ident = !{!5}

!0 = !{i32 2, !"SDK Version", [2 x i32] [i32 13, i32 3]}
!1 = !{i32 1, !"wchar_size", i32 4}
!2 = !{i32 7, !"PIC Level", i32 2}
!3 = !{i32 7, !"uwtable", i32 2}
!4 = !{i32 7, !"frame-pointer", i32 2}
!5 = !{!"Apple clang version 14.0.3 (clang-1403.0.22.14.1)"}

c                   @   s<   e Zd Zdd Zdd ZedfddZdd	d
Zdd ZdS )BaseTestc                 C   s:   t   t   t  tjd d  | _g tjd d < d S N)llvminitialize_native_targetinitialize_native_asmprintergccollectgarbageold_garbageselfr   r   r   setUp  s
   zBaseTest.setUpc                 C   s   t   | t jg  | `d S r&   )r*   r+   assertEqualr,   r-   r.   r   r   r   tearDown  s   zBaseTest.tearDownNc                 C   s    |j t d}t||}|S )Ntriple)formatr'   get_default_tripleparse_assembly)r/   asmcontextmodr   r   r   module  s   zBaseTest.moduleglobc                 C   s   |d u r|   }||S r&   )r;   get_global_variable)r/   namer:   r   r   r   r<     s   
zBaseTest.globc                C   s   t j }|j|dS )Njit)r'   Targetfrom_default_triplecreate_target_machine)r/   r@   targetr   r   r   target_machine  s   
zBaseTest.target_machine)r<   N)	__name__
__module____qualname__r0   r2   asm_sumr;   r<   rE   r   r   r   r   r%     s    	
r%   c                   @   s@   e Zd ZdZeejddee	j
dddd ZdS )	TestDependencieszB
    Test DLL dependencies are within a certain expected set.
    linuxzLinux-specific testLLVMLITE_DIST_TESTDistribution-specific testc                 C   s  t jj}tj }d|d< tjdd|gtj|d}|	 \}}| 
d|j td}t }|  D ]/}| }	|	rc|	d dkrc|	d	 }
||
}t|	d
ksS|s[| d|f  ||d	 q4d|vsld|vrt| d|f  tg d}t dkr|d g }|D ]}
|
ds|
|vr||
 qt|d	kr| d|d |f  d S t|d	kr| d||f  d S 	 d S )NCLANGobjdumpz-p)stdoutenvr   z%^([+-_a-zA-Z0-9]+)\.so(?:\.\d+){0,3}$NEEDED      zinvalid NEEDED line: %rlibc
libpthreadz#failed parsing dependencies? got %r)librtlibdlrW   libzlibmlibgcc_srV   zld-linuxld64libzstdz	libstdc++r   libtinfoz	ld-linux-zunexpected dependency %r in %rz unexpected dependencies %r in %r)r
   lib_nameosenvironcopy
subprocessPopenPIPEcommunicater1   
returncoderecompilesetdecode
splitlinessplitmatchlenfailaddgroupplatformpython_implementation
startswithappend)r/   lib_pathrR   pout_lib_patdepslinepartsdepmallowedfailsr   r   r   
test_linux  sD   




zTestDependencies.test_linuxN)rF   rG   rH   __doc__unittest
skipUnlesssysru   rw   rb   rc   getr   r   r   r   r   rJ     s    rJ   c                       s\   e Zd ZdZdZ fddZdd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Z  ZS )TestRISCVABIz`
    Test calling convention of floating point arguments of RISC-V
    using different ABI.
    zriscv32-unknown-linuxc                    s   t    t  t  d S r&   )superr0   r'   initialize_all_targetsinitialize_all_asmprintersr.   	__class__r   r   r0     s   
zTestRISCVABI.setUpc              
   C   sX   z
t j| j W d S  ty+ } zdt|v r| d n|W Y d }~d S d }~ww )Nz/No available targets are compatible with triplez(RISCV target unsupported by linked LLVM.)r'   rA   from_tripler4   RuntimeErrorstrskipTest)r/   er   r   r   check_riscv_target  s   zTestRISCVABI.check_riscv_targetc                 K   s   t j| j}|jdi |S )Nr   )r'   rA   r   r4   rC   )r/   kwarglltargetr   r   r   riscv_target_machine  s   z!TestRISCVABI.riscv_target_machinec                 C   s   t  }t  }t |||f}t  }t j||dd}| }t |}|j\}}	|	||}
|
|
|	}|| tt|}|  |S )Nfpaddr>   )r   
DoubleType	FloatTypeFunctionTypeModuleFunctionappend_basic_block	IRBuilderargsfpextfaddr"   r'   r7   r   verify)r/   f64f32fntyr;   funcblockbuilderabarg0resultllmodr   r   r   fpadd_ll_module	  s   


zTestRISCVABI.fpadd_ll_modulec                 C   sD   g }|  D ]}| }|ds|ds|dks|| q|S )N.r    )rn   striprw   rx   )r/   r8   asm_listr   s_liner   r   r   break_up_asm  s   
zTestRISCVABI.break_up_asmc                 C   :   |    |  }| jddd}| | ||t d S )N+f,+dilp32featuresabiname)r   r   r   r1   r   emit_assemblyriscv_asm_ilp32r/   r   rD   r   r   r   test_rv32d_ilp32#     zTestRISCVABI.test_rv32d_ilp32c                 C   r   )Nr   ilp32fr   )r   r   r   r1   r   r   riscv_asm_ilp32fr   r   r   r   test_rv32d_ilp32f*  r   zTestRISCVABI.test_rv32d_ilp32fc                 C   r   )Nr   ilp32dr   )r   r   r   r1   r   r   riscv_asm_ilp32dr   r   r   r   test_rv32d_ilp32d1  r   zTestRISCVABI.test_rv32d_ilp32d)rF   rG   rH   r   r4   r0   r   r   r   r   r   r   r   __classcell__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ee ddd  Zd!d" Zd#S )$TestMiscz7
    Test miscellaneous functions in llvm.binding.
    c                 C   s   |  t d S r&   )r;   rI   r.   r   r   r   test_parse_assembly>     zTestMisc.test_parse_assemblyc                 C   sZ   |  t}| t W d    n1 sw   Y  t|j}| d| | d| d S )Nzparsing errorzinvalid operand type)assertRaisesr   r;   asm_parse_errorr   	exceptionassertIn)r/   cmsr   r   r   test_parse_assembly_errorA  s   
z"TestMisc.test_parse_assembly_errorc                 C   sj   t  }t t dg }t ||d}t |jdd}|t t dd t	|}| 
|t d S )N    fooz<>!*''#r   i90  )r   r   r   IntTyper   r   r   r"   Constantr   r1   asm_nonalphanum_blocklabel)r/   r:   ftfnbdr8   r   r   r   test_nonalphanum_block_nameH  s   z$TestMisc.test_nonalphanum_block_namec                 C   s$   t j }t j }||ksJ d S r&   )r'   r9   get_global_context)r/   	gcontext1	gcontext2r   r   r   test_global_contextQ  s   

zTestMisc.test_global_contextc                 C   sH   t dd t dd t d}| |d t d}| |d  d S )N__xyzzy  i.  __foobar)r'   
add_symboladdress_of_symbolr1   assertIs)r/   addrr   r   r   test_dylib_symbolsV  s   

zTestMisc.test_dylib_symbolsc                 C   "   t  }| |t | | d S r&   )r'   r6   assertIsInstancer   
assertTrue)r/   r4   r   r   r   test_get_default_triple^     z TestMisc.test_get_default_triplec                 C   sb   dd }t  }t  }| |t | | ||dd }||dd }| || d S )Nc                 S   s   | dkrdS | S )Npowerpc64leppc64ler   )archr   r   r   normalize_ppce  s   z7TestMisc.test_get_process_triple.<locals>.normalize_ppc-r   )r'   get_process_tripler6   r   r   r   ro   r1   )r/   r   r4   defaultdefault_archtriple_archr   r   r   test_get_process_triplec  s   
z TestMisc.test_get_process_triplec                 C   s   t  }| |t | |t j | D ]\}}| |t | | | |t q| |	 t d}d
|}| t|d | t|d | t|d t|dkre| |	 d d S | t||	  d S )Nz[+\-][a-zA-Z0-9\._-]+z^({0}|{0}(,{0})*)?$r   z+aaz+a,-bbr   )r'   get_host_cpu_featuresr   dict
FeatureMapitemsr   r   boolflattenr5   assertIsNotNonerj   rp   rq   r1   )r/   r   kvre_termregexr   r   r   test_get_host_cpu_featuresu  s    

z#TestMisc.test_get_host_cpu_featuresc                 C   r   r&   )r'   get_host_cpu_namer   r   r   )r/   cpur   r   r   test_get_host_cpu_name  r   zTestMisc.test_get_host_cpu_namec                 C      d}t tjd|g d S )Na  if 1:
            from llvmlite import binding as llvm

            llvm.initialize_native_target()
            llvm.initialize_native_asmprinter()
            llvm.initialize_all_targets()
            llvm.initialize_all_asmprinters()
            llvm.shutdown()
            -cre   
check_callr   
executabler/   coder   r   r   test_initfini  s   	zTestMisc.test_initfinic                 C   s@   d}| j t|d t  W d    d S 1 sw   Y  d S )Nz-llvmlite.binding.initialize\(\) is deprecated)expected_regex)assertRaisesRegexr   r'   
initialize)r/   r  r   r   r   test_deprecated_init  s   
"zTestMisc.test_deprecated_initc                 C   r  )Nzif 1:
            from llvmlite import binding as llvm

            llvm.set_option("progname", "-debug-pass=Disabled")
            r	  r
  r  r   r   r   test_set_option  s   zTestMisc.test_set_optionc                 C   s0   t j\}}}d}| || | |td d S )N)   	   )r'   llvm_version_infor   range)r/   majorminorpatchvalidr   r   r   test_version  s   zTestMisc.test_versionc                 C   s   t   d S r&   )r'   check_jit_executionr.   r   r   r   test_check_jit_execution  s   z!TestMisc.test_check_jit_executionzLocale not availablec                 C   s8   |  t}t|}ttjd t|}| || d S )Nr   )r;   asm_double_localer   r   r   r   r1   )r/   r   expectgotr   r   r   test_print_double_locale  s
   
z!TestMisc.test_print_double_localec                 C   s$   d}d}t j|d|g}t| d S )Nzfrom llvmlite import bindingz-Werrorr	  )r   r  re   r  )r/   r  flagscmdargsr   r   r   test_no_accidental_warnings  s   z$TestMisc.test_no_accidental_warningsN)rF   rG   rH   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r   skipIfr   r#  r&  r   r   r   r   r   9  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d  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+S ),TestModuleRefc                 C   s*   |   }t| }| |d| d S )Nz; ModuleID =)r;   r   r   r   rw   r/   r:   r   r   r   r   test_str  s   zTestModuleRef.test_strc                 C   sX   |   }t| |  | tj t| W d    n1 s!w   Y  |  d S r&   )r;   r   closer   ctypesArgumentErrorr/   r:   r   r   r   
test_close  s   
zTestModuleRef.test_closec              	   C   s   |   }t| | t| W d    n1 sw   Y  | tj t| W d    n1 s3w   Y  | t# | W d    n1 sKw   Y  W d    d S W d    d S 1 scw   Y  d S r&   )r;   r   r   r,  r-  r   r.  r   r   r   	test_with  s   

"zTestModuleRef.test_withc                 C   s4   |   }d|_| |jd d|_| |jd d S )Nr   bar)r;   r>   r1   r.  r   r   r   	test_name  s
   zTestModuleRef.test_namec                 C   s   |   }| |jd d S )Nz	asm_sum.c)r;   r1   source_filer.  r   r   r   test_source_file  s   zTestModuleRef.test_source_filec                 C   s2   |   }|j}| |t ||_| ||j d S r&   )r;   data_layoutr   r   r1   r)  r   r   r   test_data_layout  s
   zTestModuleRef.test_data_layoutc                 C   s6   |   }|j}| |t  d|_| |jd d S )Nr   )r;   r4   r1   r'   r6   r)  r   r   r   test_triple  s
   zTestModuleRef.test_triplec                 C   sn   |   }| | d  |  t}| t}|  W d    n1 s%w   Y  t|j}| d| d S )Nz%.bug = add i32 1, %.bug)	r;   r   r   asm_verification_failr   r   r   r   r   )r/   r:   r   r   r   r   r   test_verify  s   


zTestModuleRef.test_verifyc                 C   r   |   }|d}| |tj | |jd | t |d W d    n1 s,w   Y  ~t	|j  d S )Nsumr   )
r;   get_functionr   r'   ValueRefr1   r>   r   	NameErrorr   r/   r:   r   r   r   r   test_get_function     
zTestModuleRef.test_get_functionc                 C   sp   |   }|d}| |jd | tdt| | t	 |d W d    d S 1 s1w   Y  d S )Nstruct.glob_type8%struct\.glob_type(\.[\d]+)? = type { i64, \[2 x i64\] }zstruct.doesnt_exist)
r;   get_struct_typer1   r>   r   rj   rp   r   r   r>  )r/   r:   st_tyr   r   r   test_get_struct_type  s   
"z"TestModuleRef.test_get_struct_typec                 C   r:  )Nr<   r1  )
r;   r=   r   r'   r=  r1   r>   r   r>  r   )r/   r:   gvr   r   r   test_get_global_variable   rA  z&TestModuleRef.test_get_global_variablec                 C   sN   |   }|j}~t|dd d}| t|d | dd |D g d d S )Nc                 S   s   | j S r&   r   )valuer   r   r   <lambda>1  s    z5TestModuleRef.test_global_variables.<locals>.<lambda>)key   c                 S   s   g | ]}|j qS r   r   ).0gr   r   r   
<listcomp>3      z7TestModuleRef.test_global_variables.<locals>.<listcomp>)r<   glob_bglob_fglob_struct)r;   global_variablessortedr1   rq   )r/   r:   itglobsr   r   r   test_global_variables-  s   z#TestModuleRef.test_global_variablesc                 C   s>   |   }|j}~t|}| t|d | |d jd d S NrT   r   r;  )r;   	functionslistr1   rq   r>   )r/   r:   rV  funcsr   r   r   test_functions6  s   zTestModuleRef.test_functionsc                 C   s^   |   }|j}~t|}| t|d | td|d j | tdt	|d  d S )NrT   struct\.glob_type(\.[\d]+)?r   rC  )
r;   struct_typesr[  r1   rq   r   rj   rp   r>   r   )r/   r:   rV  structsr   r   r   test_structs>  s   


zTestModuleRef.test_structsc                 C   s   |   }|  t}|| | tdd |jD ddg |d |  | t	j
 |d W d    d S 1 s>w   Y  d S )Nc                 s       | ]}|j V  qd S r&   r   rM  fr   r   r   	<genexpr>O      z-TestModuleRef.test_link_in.<locals>.<genexpr>mulr;  )r;   asm_mullink_inr1   rU  rZ  r<  r+  r   r,  r-  )r/   destsrcr   r   r   test_link_inJ  s   


"zTestModuleRef.test_link_inc                 C   st   |   }|  t}|j|dd | tdd |jD ddg |  | tdd |jD dg |d d S )NTpreservec                 s   rb  r&   r   rc  r   r   r   re  Z  rf  z6TestModuleRef.test_link_in_preserve.<locals>.<genexpr>rg  r;  c                 s   rb  r&   r   rc  r   r   r   re  \  rf  )r;   rh  ri  r1   rU  rZ  r+  r<  )r/   rj  src2r   r   r   test_link_in_preserveU  s   
z#TestModuleRef.test_link_in_preservec                 C   s\   |   }|  t}| t}|| W d    n1 sw   Y  | dt|j d S )Nzsymbol multiply defined)r;   asm_sum2r   r   ri  r   r   r   )r/   rj  rk  r   r   r   r   test_link_in_error_  s   
z TestModuleRef.test_link_in_errorc                 C   s6   |   }| }d}d}| ||p|| d S )Ns   s   BC)r;   
as_bitcoder   rw   )r/   r:   bcbitcode_wrapper_magicbitcode_magicr   r   r   test_as_bitcodeh  s   zTestModuleRef.test_as_bitcodec                 C   s\   |  t}td W d    n1 sw   Y  | dt|j | dt|j d S )N    zLLVM bitcode parsing errorz(file too small to contain bitcode header)r   r   r'   parse_bitcoder   r   r   )r/   r   r   r   r   test_parse_bitcode_errorr  s   
z&TestModuleRef.test_parse_bitcode_errorc                 C   sT   t  }| j|d }t  }t ||}| | | |d |d d S )Nr9   r;  r<   )r'   create_contextr;   rs  ry  r1   r<  r=   )r/   context1rt  context2r:   r   r   r   test_bitcode_roundtripz  s   
z$TestModuleRef.test_bitcode_roundtripc                 C   s4   |   }| }| || | | |  d S r&   )r;   cloneassertIsNotr1   rs  )r/   r   clonedr   r   r   test_cloning  s   zTestModuleRef.test_cloningN)rF   rG   rH   r*  r/  r0  r2  r4  r6  r7  r9  r@  rF  rH  rX  r]  ra  rl  rp  rr  rw  rz  r  r  r   r   r   r   r(    s,    	
	
r(  c                   @   s   e Zd Z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d Zdd Zdd Zdd Zdd Zdd ZdS )!JITTestMixinz*
    Mixin for ExecutionEngine tests.
    r;  c                 C   s,   |   ||}| | tttt|S r&   )finalize_objectget_function_addressr   r   r   )r/   ee	func_namecfptrr   r   r   get_sum  s   

zJITTestMixin.get_sumc                 C   sX   |   }| |}| |}|dd}| d| W d    d S 1 s%w   Y  d S NrU   )r;   r@   r  r1   )r/   r:   r  cfuncresr   r   r   test_run_code  s   

"zJITTestMixin.test_run_codec                 C   sX   |  |  }|  |  | tj |  W d    d S 1 s%w   Y  d S r&   )r@   r;   r+  r   r,  r-  r  r/   r  r   r   r   r/    s   
"zJITTestMixin.test_closec              	   C   s   |  |  }| W d    n1 sw   Y  | t | W d    n1 s,w   Y  W d    n1 s;w   Y  | tj |  W d    d S 1 sVw   Y  d S r&   )r@   r;   r   r   r,  r-  r  r  r   r   r   r0    s   
"zJITTestMixin.test_withc                 C   s&   |   }| |}|  |  d S r&   r;   r@   r+  r/   r:   r  r   r   r   test_module_lifetime     
z!JITTestMixin.test_module_lifetimec                 C   s&   |   }| |}|  |  d S r&   r  r  r   r   r   test_module_lifetime2  r  z"JITTestMixin.test_module_lifetime2c                 C   sz   |  |  }| t}|| | t || W d    n1 s&w   Y  | |j |  | 	|j d S r&   )
r@   r;   rh  
add_moduler   KeyErrorassertFalseclosedr+  r   r/   r  r:   r   r   r   test_add_module  s   

zJITTestMixin.test_add_modulec                 C   s6   |  |  }| t}|| |  |  d S r&   r@   r;   rh  r  r+  r  r   r   r   test_add_module_lifetime  
   

z%JITTestMixin.test_add_module_lifetimec                 C   s6   |  |  }| t}|| |  |  d S r&   r  r  r   r   r   test_add_module_lifetime2  r  z&JITTestMixin.test_add_module_lifetime2c                 C   s   |  |  }| t}|| || | t || W d    n1 s+w   Y  | |j |	  | |j d S r&   )
r@   r;   rh  r  remove_moduler   r  r  r  r+  r  r   r   r   test_remove_module  s   


zJITTestMixin.test_remove_modulec                 C   s>   |   }| |}|j}| |j| t| ~~t| d S r&   )r;   r@   target_datar   r   )r/   r:   r  tdr   r   r   test_target_data  s   
zJITTestMixin.test_target_datac           	      C   s   |   }| |}|j}|d}|d}|d}tjdk r"dnd}|||fD ]}| ||j| q)| ||j	d | |
|j	d | ||j	d | |
|j	d | ||j	d	 | |
|j	d
 d S )Nr<   rQ  rS          rL     rT   )rT   rU   rL     )rL  r  )r;   r@   r  r=   r   maxsizer1   get_abi_sizetypeglobal_value_typeget_abi_alignmentr   )	r/   r:   r  r  gv_i32gv_i8	gv_structpointer_sizerN  r   r   r   test_target_data_abi_enquiries  s   



z+JITTestMixin.test_target_data_abi_enquiriesc                    s   g   fdd}|   }| |}|| | t d | |}|dd | t d |  d d | |  d d t g  d d < |  t	}|
| | |d}| t d |  d d | |  d d t d S )Nc                         | |f d S r&   rx   r:   bufnotifiesr   r   notify      z5JITTestMixin.test_object_cache_notify.<locals>.notifyr   rU   r  rT   rg  )r;   r@   set_object_cacher1   rq   r  r   r   bytesrh  r  )r/   r  r:   r  r  mod2r   r  r   test_object_cache_notify  s$   





z%JITTestMixin.test_object_cache_notifyc                    sJ  g g  fdd} fdd}|   }| |}||| | td | t d | |}| td | t d |  d | d d  fdd}g d d < g  d d < |  t}| |}||| | td | t d | |}| |d	d
d | td | t d d S )Nc                    r  r&   r  r  r  r   r   r    r  z8JITTestMixin.test_object_cache_getbuffer.<locals>.notifyc                    s     |  d S r&   r  r:   )
getbuffersr   r   	getbuffer  r   z;JITTestMixin.test_object_cache_getbuffer.<locals>.getbufferr   rT   c                    s     |  S r&   r  r  )r  
sum_bufferr   r   getbuffer_successful2  s   
zFJITTestMixin.test_object_cache_getbuffer.<locals>.getbuffer_successfulrU   r  r  )r;   r@   r  r1   rq   r  r   rh  )r/   r  r  r:   r  r  r  r   )r  r  r  r   test_object_cache_getbuffer  s4   




z(JITTestMixin.test_object_cache_getbufferN)r;  )rF   rG   rH   r   r  r  r/  r0  r  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S )JITWithTMTestMixinc                 C   sf   | j dd}|  }| ||}||}| d| |d ||}| d| | || dS )z%Test TargetMachineRef.emit_assembly()Tr?   r;  N)rE   r;   r@   r   r   set_asm_verbosityassertNotEqual)r/   rE   r:   r  raw_asmraw_asm_verboser   r   r   test_emit_assemblyG  s   


z%JITWithTMTestMixin.test_emit_assemblyc                 C   s^   | j dd}|  }| ||}||}| |t tjdr-| 	d|dd  dS dS )z#Test TargetMachineRef.emit_object()Tr?   rK   s   ELFN
   )
rE   r;   r@   emit_objectr   r  r   ru   rw   r   )r/   rE   r:   r  code_objectr   r   r   test_emit_objectS  s   
z#JITWithTMTestMixin.test_emit_objectN)rF   rG   rH   r  r  r   r   r   r   r  E  s    r  c                   @   s   e Zd ZdZdddZdS )	TestMCJitz@
    Test JIT engines created with create_mcjit_compiler().
    Nc                 C   s    |d u r
| j dd}t||S )NTr?   )rE   r'   create_mcjit_compiler)r/   r:   rE   r   r   r   r@   d  s   zTestMCJit.jitr&   )rF   rG   rH   r   r@   r   r   r   r   r  _  s    r  zOrcJIT support is experimentalc                   @   s   e Zd Zedddeeeedf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S )&TestOrcLLJITr;  NFc                 C   sv   t j|d|d}t  }|r|  ||jt  d|||}	|	| }
| 	|
 | 
||	j ||	||
fS )NF)use_jit_linksuppress_errorsr3   )r'   create_lljit_compilerJITLibraryBuilderadd_current_processadd_irr5   r6   export_symbollinkr   r1   r>   )r/   r8   r  rE   add_process	func_typer  lljitr   rtr  r   r   r   r@   t  s    
zTestOrcLLJIT.jitc                 C   s:   t  }t  ddd|d}| |d d d S )Nr   r   r   )r'   r  r  import_symbolr  r  r1   r/   r  r  r   r   r   test_define_symbol  s
   zTestOrcLLJIT.test_define_symbolc                 C   s   t  }| td |dd W d    n1 sw   Y  t  ddd|d}| 	|d d | td |dd W d    d S 1 sOw   Y  d S )NzNo such libraryr   r   r   r   r   zSymbols not found.*__foobar)
r'   r  r  r   lookupr  r  r  r  r  r  r   r   r   "test_lookup_undefined_symbol_fails  s   "z/TestOrcLLJIT.test_lookup_undefined_symbol_failsc                 C   s^   t jdkr$| td tjdd W d    d S 1 sw   Y  d S | tjdd d S )Nwin32zJITLink .* WindowsT)r  )r   ru   r  r   r'   r  r   r.   r   r   r   test_jit_link  s   
"zTestOrcLLJIT.test_jit_linkc                 C   sN   |   \}}}| |dd}| d| W d    d S 1 s w   Y  d S r  )r@   r1   )r/   r  r  r  r  r   r   r   r    s
   
"zTestOrcLLJIT.test_run_codec                 C   sZ   |   \}}}|  |  | t |dd W d    d S 1 s&w   Y  d S Nr   r   )r@   r+  r   AssertionErrorr  r/   r  r  r  r   r   r   r/    s   "zTestOrcLLJIT.test_closec              	   C   s   |   \}}}| W d    n1 sw   Y  | t | W d    n1 s,w   Y  W d    n1 s;w   Y  | t |dd W d    d S 1 sWw   Y  d S r  )r@   r   r   r  r  r  r   r   r   r0    s   "zTestOrcLLJIT.test_withc                 C   s   |   \}}}t tjt dd|d}t	t
t
t
|d dd}| d| | |ddd d | |ddd d | t |dd W d    n1 s\w   Y  | t |dd W d    d S 1 sxw   Y  d S )Nr3   rg  rU   r  ir;  r   )r@   r'   r  r  rh  r5   r6   r  r  r   r   r1   r  r  r   r   r/   r  rt_sum	cfunc_sumrt_mulr  r   r   r   test_add_ir_module  s    "zTestOrcLLJIT.test_add_ir_modulec                 C   sZ   |   \}}}~t  | t |dd W d    n1 s"w   Y  |  d S Nr;  )r@   r*   r+   r   r   r  r+  )r/   r  r  r|   r   r   r   r    s   zTestOrcLLJIT.test_remove_modulec                 C   sf   |   \}}}t tjt ddd	|d}t
ttt|d dd}| d| d S )Nr3   
square_sumr;  rU   r  r  )r@   r'   r  r  asm_square_sumr5   r6   r  add_jit_libraryr  r   r   r1   r  r   r   r   test_lib_depends  s   zTestOrcLLJIT.test_lib_dependsc                 C   s8   |   \}}}|j}| |j| t| ~t| d S r&   )r@   r  r   r   )r/   r  r  r|   r  r   r   r   r    s   zTestOrcLLJIT.test_target_datac                 C   s   t d}t }t }|tjt ddt	
|d|d}|d }| | | tt| d ~| |jd d S )Nr   r3   Ar      r  )r   r'   r  r  r  asm_ext_ctorsr5   r6   r  r,  	addressofr  r  r   r1   r   r   r  rI  )r/   shared_valuer  r   r  r   r   r   r   test_global_ctors_dtors  s   
z$TestOrcLLJIT.test_global_ctors_dtorsc                 C   sF   d}|  t| | jtddd W d    d S 1 sw   Y  d S )Nz*Failed to materialize symbols:.*getversion
getversionT)r  )r  r   r@   asm_getversion)r/   msgr   r   r   (test_lookup_current_process_symbol_fails  s   "z5TestOrcLLJIT.test_lookup_current_process_symbol_failsc                 C   s   |  tdd d d S )Nr  T)r@   r  r.   r   r   r   "test_lookup_current_process_symbol  s   z/TestOrcLLJIT.test_lookup_current_process_symbolc                    sh   t  tjt  dfdd  fddtt D }|D ]}|  q"|D ]}|	  q+d S )Nr3   c                    s    fdd}|S )Nc               
      sD   g } t dD ]}| t dd  d|  qd S )N2   r;  sum_r|   )r  rx   r'   r  r  r  r  )trackingc)ir  llvm_irr   r   do_work  s   
zDTestOrcLLJIT.test_thread_safe.<locals>.compile_many.<locals>.do_workr   )r  r  )r  r  )r  r   compile_many  s   z3TestOrcLLJIT.test_thread_safe.<locals>.compile_manyc                    s   g | ]
}t j |d qS ))rD   )	threadingThread)rM  r  )r  r   r   rO    s    z1TestOrcLLJIT.test_thread_safe.<locals>.<listcomp>)
r'   r  rI   r5   r6   r  rb   	cpu_countstartjoin)r/   thsthr   )r  r  r  r   test_thread_safe  s   



zTestOrcLLJIT.test_thread_safec                 C   sj   | j dd}|  }t|}t ||d|d}t	t
t
t
|d }| |ddd d S )NFr?   r;  rU         )rE   r;   r'   r  r  add_object_imgr  r  r  r   r   r1   )r/   rE   r:   r  r  r;  r   r   r   test_add_object_file  s   
z!TestOrcLLJIT.test_add_object_filec           
   	   C   s   | j dd}|  }||}t \}}zIt|d}|| W d    n1 s+w   Y  t|}t	 
|d|d}tttt|d }	| |	ddd W t| d S t| w )NFr?   wbr;  rU   r  r  )rE   r;   r  r   rb   fdopenwriter'   r  r  add_object_filer  r  r   r   r1   unlink)
r/   rE   r:   obj_bin	temp_desc	temp_pathrd  r  r  r;  r   r   r   $test_add_object_file_from_filesystem  s"   


z1TestOrcLLJIT.test_add_object_file_from_filesystem)rF   rG   rH   rI   r   r   r@   r  r  r  r  r/  r0  r  r  r  r  r  r  r  r  r  r  r   r   r   r   r  q  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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d7S )8TestValueRefc                 C   s&   |   }|d}| t|d d S )Nr<   z@glob = global i32 0)r;   r=   r1   r   r/   r:   r<   r   r   r   r*  4  s   
zTestValueRef.test_strc                 C   s8   |   }|d}| |jd d|_| |jd d S )Nr<   foobar)r;   r=   r1   r>   r  r   r   r   r2  9  s
   
zTestValueRef.test_namec                 C   p   |   }|d}|j}| |jtj ||_| |j| dD ]}||_| |jtj | |jj| q d S )Nr<   )internalexternal)r;   r=   linkager   r'   Linkager1   r>   )r/   r:   r<   r   r   r   r   test_linkage@     
zTestValueRef.test_linkagec                 C   r  )Nr<   )hidden	protectedr   )r;   r=   
visibilityr   r'   
Visibilityr1   r>   )r/   r:   r<   r&  r   r   r   test_visibilityL  r#  zTestValueRef.test_visibilityc                 C   r  )Nr<   )	dllimport	dllexportr   )r;   r=   storage_classr   r'   StorageClassr1   r>   )r/   r:   r<   r+  r   r   r   test_storage_classX  r#  zTestValueRef.test_storage_classc                 C   sf   |   }|d}|d | t}|d W d    n1 s#w   Y  | t|jd d S )Nr;  	nocapturezextzno such attribute 'zext')r;   r<  add_function_attributer   
ValueErrorr1   r   r   )r/   r:   r   raisesr   r   r   test_add_function_attributed  s   

z(TestValueRef.test_add_function_attributec                 C   s$   |   }|d}| |j | d S Nr<   )r;   r=   r   r  r   r   r   test_modulel  s   
zTestValueRef.test_modulec                 C   s*   |   }|d}|j}| |tj d S r4  )r;   r=   r  r   r'   TypeRefr/   r:   r<   tpr   r   r   	test_typeq  s   
zTestValueRef.test_typec                 C   sJ   |   }|d}|j}| |jd |d}| td|jj d S )Nr<   r   rS  r^  )	r;   r=   r  r1   r>   r   rj   rp   r  )r/   r:   r<   r8  str   r   r   test_type_namew  s   



zTestValueRef.test_type_namec                 C   s,   |   }|d}|j}| t|d d S )Nr<   i32)r;   r=   r  r1   r   r7  r   r   r   test_type_printing_variable  s   
z(TestValueRef.test_type_printing_variablec                 C   s(   |   }|d}| t|jd d S )Nr;  zi32 (i32, i32))r;   r<  r1   r   r  r?  r   r   r   test_type_printing_function  s   
z(TestValueRef.test_type_printing_functionc                 C   sT   |   }|d}| |jj | tdt|j | tdt|j	 d S )NrS  ptrrC  )
r;   r=   r   r  
is_pointerr   rj   rp   r   r  )r/   r:   r:  r   r   r   test_type_printing_struct  s   

z&TestValueRef.test_type_printing_structc                 C   s   |   }|  |  d S r&   )r<   r+  )r/   r<   r   r   r   r/    s   zTestValueRef.test_closec                 C   s:   |   d}|  td}| |j | |j d S r  )r;   r<  asm_sum_declarer  is_declarationr   )r/   defineddeclaredr   r   r   test_is_declaration  s   z TestValueRef.test_is_declarationc                 C   s>   |  t}t|j}| t|d |D ]}| |j qd S NrL  )r;   rI   r[  rT  r1   rq   r   	is_global)r/   r:   gvarsr  r   r   r   test_module_global_variables  s   

z)TestValueRef.test_module_global_variablesc                 C   s   |   }t|j}| t|d |d }| |j | |jd | t	 |j
 W d    n1 s5w   Y  | t	 |j W d    n1 sMw   Y  | t	 |j W d    d S 1 sfw   Y  d S rY  )r;   r[  rZ  r1   rq   r   is_functionr>   r   r1  instructionsoperandsopcode)r/   r:   r\  r   r   r   r   test_module_functions  s   
"z"TestValueRef.test_module_functionsc                 C   s  |   }|d}| |j t|j}| t|d | |d j | |d j | |d j	d | t
|d jd | |d j	d | t
|d jd | t |d j W d    n1 siw   Y  | t |d j W d    d S 1 sw   Y  d S )Nr;  rU   r   rT   .1r<  .2)r;   r<  r   rK  r[  	argumentsr1   rq   is_argumentr>   r   r  r   r1  blocks)r/   r:   r   r   r   r   r   test_function_arguments  s"   

"z$TestValueRef.test_function_argumentsc                 C   s@   |   d}t|j}| t|d |d }| |j d S )Nr;  rT   r   )r;   r<  r[  rT  r1   rq   r   is_block)r/   r   rT  r   r   r   r   test_function_blocks  s
   
z!TestValueRef.test_function_blocksc                 C   s   |   d}tt|jd j}| t|d | |d j | |d j | |d j | |d j	d | |d j	d | |d j	d d S )Nr;  r   r  rT   rU   rs   r"   )
r;   r<  r[  rT  rL  r1   rq   r   is_instructionrN  )r/   r   instsr   r   r   test_block_instructions  s   z$TestValueRef.test_block_instructionsc                 C   s   |   d}tt|jd jd }| |jd t|j}| t|d | 	|d j
 | 	|d j
 | |d jd | t|d jd | |d jd | t|d jd d S )	Nr;  r   rs   rU   rT   rP  r<  rQ  )r;   r<  r[  rT  rL  r1   rN  rM  rq   r   
is_operandr>   r   r  )r/   r   rs   rM  r   r   r   test_instruction_operands  s   
z&TestValueRef.test_instruction_operandsc                 C   s   t jd }|dkrdndg}| t}|jD ]:}t|j}|jdkr)| || q|jdkrP| |g  t|j	}| t|d jdg | t|d jg  qd S )	Nr      s   memory(read)s   readonlya_readonly_funca_arg0_return_funcs   returnedrT   )
r'   r  r;   asm_attributesrZ  r[  
attributesr>   r1   rR  )r/   verreadonly_attrsr:   r   attrsr   r   r   r   test_function_attributes  s   






z%TestValueRef.test_function_attributesc                 C   s   |   }| |djtjj |d}| |jtjj t	|j
d }| |jtjj t	|jd }| |jtjj | t	|jd jtjj | t	|jd jtjj |  td}t	t	|j
d jd }| t	|jd jtjj d S )Nr<   r;  r   rT   r   )r;   r1   r=   
value_kindr'   	ValueKindglobal_variabler<  functionr[  rT  basic_blockrL  instructionrM  constant_intasm_inlineasm
inline_asm)r/   r:   r   r   inst	iasm_func	iasm_instr   r   r   test_value_kind  s*   
zTestValueRef.test_value_kindc                 C   s   |   }| |dj d}|jD ]-}| |j |jD ]!}| |j |jD ]}| |j |jD ]	}|jr<|d7 }q3q(qq| 	|d d S )Nr<   r   rT   )
r;   r   r=   is_constantrZ  rT  r  rL  rM  r1   )r/   r:   constant_operandsr   r   ro  opr   r   r   test_is_constant  s"   



zTestValueRef.test_is_constantc           	      C   sT  |   }|d}tt|jd j}| |d jd t|d j}| |d j	 | 
|d j	 | |d  d | t |d   W d    n1 sSw   Y  |  t}|d}tt|jd j}t|d jd }t|d jd }| | d | |jddd	 |jd
d}tj|ddddd}| |d	 d S )Nr;  r   rT   rs   rU   r  T)
signed_intr  Fr  little)signed)r;   r<  r[  rT  rL  r1   rN  rM  r   rs  r  get_constant_valuer   r1  asm_sum3int
from_bytesto_bytes)	r/   r:   r   rY  rM  posint64negint64as_u64as_i64r   r   r   test_constant_int  s.   


zTestValueRef.test_constant_intc                 C   s(  |  t}|d}tt|jd j}| t|d | |d jd t|d j	}| 
|d j | |d  d | 
|d j | |d  d |  t}|d}tt|jd jd }t|j	}| t |d   W d    n1 sw   Y  | |d jdd	d d S )
Nr   r   rU   r   g        rT   gQ	@T)round_fp)r;   r   r<  r[  rT  rL  r1   rq   rN  rM  r   rs  assertAlmostEqualrz  asm_double_inaccurater   r1  )r/   r:   r   rY  rM  ro  r   r   r   test_constant_fp2  s$   




zTestValueRef.test_constant_fpc                 C   sZ   |  t}|d}tt|jd jd }t|jd }| |j | 	|
 d d S )Nr1  r   zptr null)r;   asm_null_constantr<  r[  rT  rL  rM  r   rs  r1   rz  )r/   r:   r   ro  argr   r   r   test_constant_as_stringF  s   

z$TestValueRef.test_constant_as_stringc                 C   s   |  t}|d}t|j}t|d j}| |d j | |d j	d t|d j
}| t|d | |d j | |d j | |d |d  | |d |d  | |d j	d | t |d j
 W d    d S 1 syw   Y  d S )Nr   r   phirU   rT   )r;   asm_phi_blocksr<  r[  rT  rL  r   rX  r1   rN  incoming_blocksrq   rV  r  r   r1  )r/   r:   r   rT  rL  r  r   r   r   test_incoming_phi_blocksN  s    


"z%TestValueRef.test_incoming_phi_blocksN)rF   rG   rH   r*  r2  r"  r(  r-  r3  r5  r9  r;  r=  r>  rA  r/  rF  rJ  rO  rU  rW  rZ  r\  re  rr  rv  r  r  r  r  r   r   r   r   r  2  s8    		r  c                   @   d   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S )TestTypeRefc                 C   s   |   }|d}| t|jd |d}| t|d t|j}| t|d | t|d d | t|d d	 d S )
Nr<   r<  rB  z+%struct.glob_type = type { i64, [2 x i64] }rU   r   i64rT   z	[2 x i64])	r;   r=   r1   r   r  rD  r[  elementsrq   )r/   r:   r<   glob_struct_typer  r   r   r   r*  f  s   



zTestTypeRef.test_strc                 C   s0  |   }|d}| |jjtjj | |jj	 |d}| |jjtjj | |jj	 |j
}| |jtjj | |j |j\}}| |jtjj | |jtjj | |j |d}|j\}}	| |	jtjj | |	j |dj}
| |
jtjj |dj
}| |jtjj d S )Nr<   rS  struct.glob_type_vecr;  )r;   r=   r1   r  	type_kindr'   TypeKindpointerr   r@  r  struct	is_structr  integerarrayis_arrayrD  vector	is_vectorr<  ri  )r/   r:   r<   rS  stypestype_astype_bglob_vec_struct_typer|   vector_typefuncptrfunctyper   r   r   test_type_kinds  s,   




zTestTypeRef.test_type_kindc                 C   s`   |   }|d}|j\}}| |jd | t |j W d    d S 1 s)w   Y  d S )NrB  rU   )r;   rD  r  r1   element_countr   r1  )r/   r:   r  r|   
array_typer   r   r   test_element_count  s   

"zTestTypeRef.test_element_countc                 C   sl   |   }|d}|d}|j\}}|j\}}| |jd | |jd | |jd | |jd d S )NrB  r  @      r   )r;   rD  r  r1   
type_width)r/   r:   r  r  integer_typer  r|   r  r   r   r   test_type_width  s   



zTestTypeRef.test_type_widthc                 C   s   |  t}|d}|j}| |j |  t}|d}|j}| |j | |jj	 | 
t}|jj W d    n1 sAw   Y  | dt|j d S )Nvarargr;  zType ptr is not a function)r;   asm_vararg_declarer<  r  r   is_function_varargrB  r  r  r@  r   r1  r   r   r   )r/   r:   r   decltyper2  r   r   r   test_vararg_function  s   




z TestTypeRef.test_vararg_functionc                 C   s   |   }t|j\}| |jjtjj | 	|jj
 |j}| |jtjj | |j
 |tj}| |tj | |jtdtdf | |jtd d S Nr   )r;   r[  rZ  r1   r  r  r'   r  r  r  rK  r  ri  r   as_irr   global_contextr   r   r   r   return_type)r/   r:   r   r   tyirr   r   r   test_function_typeref_as_ir  s   z'TestTypeRef.test_function_typeref_as_irc                 C   s`   t t  d}t  }t ||d}| t|}|d}|j}| 	|j
|t jj
 d S )Nr   r   )r   r   VoidTyper   r   r;   r   r<  r  r1   r  r  r  )r/   r   irmodr   r:   gvtyr   r   r   test_void_typeref_as_ir  s   
z#TestTypeRef.test_void_typeref_as_irc              
      s  ddl m} t  ddh}i }dd }||d< dd	 }||d
< dd }||d< dd }||d<  fdd}||d< dd }	|	|d< dd }
|
|d< dd }||d< | dd | D t| t|B  | D ]_\}}|j|v ruqk||j  D ]N}| 	|d| : tj
 d }tj||d!d" t|}t|}|d!}|j}|t }| || | || W d    n1 sw   Y  q{qkd S )#Nr   )_TypeKindToIRTyperi  voidc                   s       t  V  d S r&   )r   HalfTyper   r   r   r   
maker_half     z9TestTypeRef.test_global_typeref_as_ir.<locals>.maker_halfhalfc                   s   r  r&   )r   r   r   r   r   r   maker_float  r  z:TestTypeRef.test_global_typeref_as_ir.<locals>.maker_floatfloatc                   s   r  r&   )r   r   r   r   r   r   maker_double  r  z;TestTypeRef.test_global_typeref_as_ir.<locals>.maker_doubledoublec                   s   s    t dV  d S r  )r   r   r   r   r   r   maker_integer  s   z<TestTypeRef.test_global_typeref_as_ir.<locals>.maker_integerr  c                  3   sL    t t dV   d V   d} | t d |  V  d S )Nr  myclassmyclass2)r   PointerTyper   get_identified_type
as_pointerset_body)r  ctxr   r   maker_pointer  s   
z<TestTypeRef.test_global_typeref_as_ir.<locals>.maker_pointerr  c                   s   s    t t ddV  d S )Nr  {   )r   	ArrayTyper   r   r   r   r   maker_array  s   z:TestTypeRef.test_global_typeref_as_ir.<locals>.maker_arrayr  c                   s   s    t t  dV  d S )NrU   )r   
VectorTyper   r   r   r   r   maker_vector  s   z;TestTypeRef.test_global_typeref_as_ir.<locals>.maker_vectorr  c                   s   s>    t t  t dgV  t jt  t dgddV  d S )Nr  Tpacked)r   LiteralStructTyper   r   r   r   r   r   maker_struct  s
   z;TestTypeRef.test_global_typeref_as_ir.<locals>.maker_structr  c                 S   s   h | ]}|j qS r   r   )rM  xr   r   r   	<setcomp>  rP  z8TestTypeRef.test_global_typeref_as_ir.<locals>.<setcomp>z -> r{  rG  r   )llvmlite.binding.typerefr  r   Contextr1   keysrl   r   r>   subTestr   GlobalVariabler   r'   r7   r=   r  r  r   )r/   r  skippedmakersr  r  r  r  r  r  r  r  r  irtypetyr  r8   r:   rG  r  r"  r   r  r   test_global_typeref_as_ir  sV   	


z%TestTypeRef.test_global_typeref_as_irc                 C   s  t |}||}|j}t }|j|d}tj|d}tj|||j	d}	tj||dd}
t
|
 }||	|
j |  t t|}| j||	j	jdd |j|dd | jtt|jtt|jd	 d
d | j||	j	jdd | j||	j	j||
j	jdd d S )N)ir_ctxr{  r   wrapperzdeclfn must not have a body)r  Trm  rT   z$the only new function is the wrapperzdeclfn must have a body nowz.declfn and wrapfn must have the same llvm Type)r'   r7   r<  r  r   r  r  r   r   r>   r   r   callr   ret_voidr   r   rC  ri  r1   rq   r[  rZ  r  )r/   r8   target_symbolr:   cppfn	cppfntype
my_contextr  wrapper_moddeclfnwrapfnr   new_modr   r   r   !_check_typeref_as_ir_for_wrappers"  s6   


z-TestTypeRef._check_typeref_as_ir_for_wrappersc                 C      |  td dS )ztExercise extracting C++ defined class types.
        Contains both opaque and non-opaque class definitions.
        _Z3fooP7MyClass14MyClassDefinedN)r  asm_cpp_classr.   r   r   r   ,test_typeref_as_ir_for_wrappers_of_cpp_classE  s   z8TestTypeRef.test_typeref_as_ir_for_wrappers_of_cpp_classc                 C   r  )zExercise extracting C++ struct types that are passed as vectors.

        IA64 ABI on x86_64 will put struct with two floats as
        a vector of two floats.
        _Z3foo8Vector2DPS_N)r  asm_cpp_vectorr.   r   r   r   4test_typeref_as_ir_for_wrappers_of_cpp_vector_structN  s   z@TestTypeRef.test_typeref_as_ir_for_wrappers_of_cpp_vector_structN)rF   rG   rH   r*  r  r  r  r  r  r  r  r  r  r  r   r   r   r   r  d  s    O#	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S )
TestTargetc                 C   sv   t jj}| t}|d W d    n1 sw   Y  | dt|j t  }||}| 	|j
| |  d S )Nr  z(No available targets are compatible with)r'   rA   r   r   r   r   r   r   r6   r1   r4   r+  )r/   rd  r   r4   rD   r   r   r   test_from_triple\  s   
zTestTarget.test_from_triplec                 C   s@   t jt  }|ddddd t  }||dddd d S )Nr   rT   r   )r'   rA   r   r6   rC   r  )r/   rD   r  r   r   r   test_create_target_machineg  s   z%TestTarget.test_create_target_machinec                 C   <   t jt  }t j }| |jt | |j|j d S r&   )	r'   rA   r   r6   rB   r   r>   r   r1   r/   tur   r   r   r2  o     
zTestTarget.test_namec                 C   r   r&   )	r'   rA   r   r6   rB   r   descriptionr   r1   r  r   r   r   test_descriptionu  r  zTestTarget.test_descriptionc                 C   s8   t jt  }t|}| |j| | |j| d S r&   )r'   rA   r   r6   r   r   r>   r  )r/   rD   r   r   r   r   r*  {  s   zTestTarget.test_strc                 C   s4  dt jjdddddddfd	t jjdddddddfd
t jjdddddddfdt jjdddddddfdt jjdddddddfdt jjdddddddfdt jjdddddddfdt jjdddddddfdt jjd dd!d"dd#dfd$t jjd%ddd&dd'dfg
}|D ]}|\}}t |}| || qd S )(Nzx86_64-scei-ps4x86_64r   sceips4unknownELF)ArchSubArchVendorOSEnvObjectFormatzx86_64-sie-ps4zpowerpc-dunno-notsurepowerpczpowerpcspe-unknown-freebsdspefreebsdzarmv6hl-none-linux-gnueabiarmv6hlrK   gnueabizi686-unknown-linux-gnui386gnuzi686-apple-macosxapplemacosxMachOzi686-dunno-win32windowsmsvcCOFFzs390x-ibm-zoss390xibmzosGOFFzwasm64-wasiwasm64wasiWasm)r'   targetsTripleget_triple_partsr1   )r/   casescase
triple_str
triple_objr  r   r   r   test_get_parts_from_triple  s   









+
z%TestTarget.test_get_parts_from_tripleN)	rF   rG   rH   r  r  r2  r  r*  r.  r   r   r   r   r  Z  s    r  c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestTargetDatac                 C   s
   t dS )Nz%e-m:e-i64:64-f80:128-n8:16:32:64-S128)r'   create_target_datar.   r   r   r   r    s   
zTestTargetData.target_datac                 C   s(   |   }|  }| ||jd d S )Nr  )r  r<   r1   r  r  r/   r  r<   r   r   r   test_get_abi_size  s   z TestTargetData.test_get_abi_sizec                 C   sF   |   }|  }| ||jd | d}| ||jd d S )NrL  rS  r  )r  r<   r1   r  r  r1  r   r   r   test_get_pointee_abi_size  s
   
z(TestTargetData.test_get_pointee_abi_sizec                 C   s|   |   }| d}| t ||jd W d    n1 s w   Y  |j}| ||dd | ||dd d S )NrS  r   rT   r  )r  r<   r   r1  get_element_offsetr  r  r1   )r/   r  r<   struct_typer   r   r   test_get_struct_element_offset  s   
z-TestTargetData.test_get_struct_element_offsetN)rF   rG   rH   r  r2  r3  r6  r   r   r   r   r/    s
    	r/  c                   @      e Zd Zdd ZdS )TestTargetMachinec                 C   sN   | j dd}|j}|  }|d}tjdk rdnd}| ||j| d S )NFr?   r<   r  rL  r  )	rE   r  r;   r=   r   r  r1   r  r  )r/   tmr  r:   r  r  r   r   r   test_target_data_from_tm  s   
z*TestTargetMachine.test_target_data_from_tmN)rF   rG   rH   r:  r   r   r   r   r8        r8  c                   @   s0   e Zd Zdd Zee dv ddd ZdS )	TestDylibc                 C   s:   |  t td W d    d S 1 sw   Y  d S )Nzzzzasdkf;jasd;l)r   r   r'   load_library_permanentlyr.   r   r   r   test_bad_library  s   "zTestDylib.test_bad_library)Linuxztest only works on Linuxc                 C   s   t d}t| d S )Nr   )r   r'   r=  )r/   r[   r   r   r   	test_libm  s   zTestDylib.test_libmN)	rF   rG   rH   r>  r   r   ru   systemr@  r   r   r   r   r<    s    r<  c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestAnalysisc                 C   sj   t  }t t dt dt dg}t ||d}t | }|j\}}|||}|	| |S )Nr   r   )
r   r   r   r   r   r   r   r   rs   r"   )r/   r   r   r   r   r  yzr   r   r   build_ir_module  s   "

zTestAnalysis.build_ir_modulec                 C   sJ   |   }|d}t|}tj|dd}d}| || | || d S )Nr   F	show_instz%.5 = add i32 %.1, %.2)rE  
get_globalr'   get_function_cfgr   assertNotIn)r/   r:   r   dot_showing_instdot_without_instro  r   r   r   test_get_function_cfg_on_ir  s   

z(TestAnalysis.test_get_function_cfg_on_irc                 C   s   |   d}tj|dd}tj|dd}d}| || | || d}| || | || d}| || | || d S )Nr;  TrF  FdigraphzCFG for 'sum' functionz%.3 = add i32 %.1, %.2)r;   r<  r'   rI  r   rJ  )r/   rD  rK  rL  prefixfnamero  r   r   r   test_function_cfg_on_llvm_value	  s   z,TestAnalysis.test_function_cfg_on_llvm_valueN)rF   rG   rH   rE  rM  rQ  r   r   r   r   rB    s    
	rB  c                   @   s    e Zd Zedd Zdd ZdS )TestTypeParsingc                 c   s&    t  }|V  t|}t| d S r&   )r   r   r   r'   r7   )r/   r:   r8   r   r   r   check_parsing	  s
   zTestTypeParsing.check_parsingc                 C   s   |    }ttdg}t||d}t|dg|_W d    n1 s'w   Y  |   "}tjtdgdd}t||d}t|dg|_W d    n1 sUw   Y  |   (}tjjtddtddgdd}t||jd}||_W d    d S 1 sw   Y  d S )Nr   r   rT   Tr  rU   )	rS  r   r  r   r  r   initializerliteral_structr  )r/   r:   typrG  constr   r   r   test_literal_struct	  s*   


"z#TestTypeParsing.test_literal_structN)rF   rG   rH   r   rS  rX  r   r   r   r   rR  	  s    
	rR  c                   @   s(   e Zd Zee dkddd ZdS )TestGlobalConstructorsDarwinz&__cxa_atexit is broken on OSX in MCJITc                 C   s   |  t}| |}|  |  |d}ttj}t	||}| 
|jjd |d}ttj|}| 
| d |  | 
|jjd d S )Nr  r  r   r  r  )r;   asm_global_ctorsr@   r  run_static_constructorsget_global_value_addressr,  POINTERr   castr1   contentsrI  r  r   run_static_destructors)r/   r:   r  ptr_addrptr_tr?  foo_addrr   r   r   r   r  8	  s   



z.TestGlobalConstructors.test_global_ctors_dtorsN)rF   rG   rH   r   r'  ru   rA  r  r   r   r   r   rY  7	  s
    rY  c                   @   sn   e Z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d Zdd Zdd Zdd ZdS )TestGlobalVariablesTc                 C   sZ   t  }t d}t ||d}||_t|}|r | d| n| d| | | d S )Nr   r   undef)	r   r   r   r  r   r   r   rJ  r;   )r/   r   	has_undefr:   rV  rG  r8   r   r   r   check_global_variable_linkageT	  s   
z1TestGlobalVariables.check_global_variable_linkagec                 C      |  d d S )Nr  rh  r.   r   r   r   test_internal_linkageh	  r   z)TestGlobalVariables.test_internal_linkagec                 C   ri  )Ncommonrj  r.   r   r   r   test_common_linkagek	  r   z'TestGlobalVariables.test_common_linkagec                 C      | j ddd d S )Nr  Frg  rj  r.   r   r   r   test_external_linkagen	  r  z)TestGlobalVariables.test_external_linkagec                 C   ri  )Navailable_externallyrj  r.   r   r   r   !test_available_externally_linkageq	  r   z5TestGlobalVariables.test_available_externally_linkagec                 C   ri  )Nprivaterj  r.   r   r   r   test_private_linkaget	  r   z(TestGlobalVariables.test_private_linkagec                 C   ri  )Nlinkoncerj  r.   r   r   r   test_linkonce_linkagew	  r   z)TestGlobalVariables.test_linkonce_linkagec                 C   ri  )Nweakrj  r.   r   r   r   test_weak_linkagez	  r   z%TestGlobalVariables.test_weak_linkagec                 C   ri  )N	appendingrj  r.   r   r   r   test_appending_linkage}	  r   z*TestGlobalVariables.test_appending_linkagec                 C   rn  )Nextern_weakFro  rj  r.   r   r   r   test_extern_weak_linkage	  r  z,TestGlobalVariables.test_extern_weak_linkagec                 C   ri  )Nlinkonce_odrrj  r.   r   r   r   test_linkonce_odr_linkage	  r   z-TestGlobalVariables.test_linkonce_odr_linkagec                 C   ri  )Nweak_odrrj  r.   r   r   r   test_weak_odr_linkage	  r   z)TestGlobalVariables.test_weak_odr_linkageN)T)rF   rG   rH   rh  rk  rm  rp  rr  rt  rv  rx  rz  r|  r~  r  r   r   r   r   re  S	  s    
re  x86zonly on x86c                   @   r7  )TestInlineAsmc                 C   s:   t   | jtd}| jdd}||}| d| d S )N)r8   Fr?   nop)r'   initialize_native_asmparserr;   rm  rE   r   r   )r/   r   r9  r8   r   r   r   test_inlineasm	  s
   
zTestInlineAsm.test_inlineasmN)rF   rG   rH   r  r   r   r   r   r  	  r;  r  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )TestObjectFilea,  
        ;ModuleID = <string>
        target triple = "{triple}"

        declare i32 @sum(i32 %.1, i32 %.2)

        define i32 @sum_twice(i32 %.1, i32 %.2) {{
            %.3 = call i32 @sum(i32 %.1, i32 %.2)
            %.4 = call i32 @sum(i32 %.3, i32 %.3)
            ret i32 %.4
        }}
    c                 C   s   | j dd}|  }||}tj|}d}d}| D ]"}| r?t|	 dkr?|
 d ur?||
 k r?d}|
 } nq| | d S )NFr?   r  r   T)rE   r;   r  r'   ObjectFileRef	from_datasectionsis_textrq   dataaddressr   )r/   rE   r:   r  objhas_text_and_datalast_addressr   r   r   r   test_object_file	  s"   
zTestObjectFile.test_object_filec                 C   st   | j dd}|  }||}tj|}t| | j|}|| t	t
t
t
|d}| |ddd d S )NFr?   	sum_twicerU   r  r  )rE   r;   r  r'   r  r  r  mod_asmr  r   r   r  r1   )r/   rE   r:   r  r  r@   r  r   r   r   r  	  s   


z#TestObjectFile.test_add_object_filec           	   	   C   s   | j dd}|  }||}t \}}z1zt|d}|| |  W |  n|  w t	
| | j|}|| W t| nt| w tttt|d}| |ddd d S )NFr?   r  r  rU   r  r  )rE   r;   r  r   rb   r  r  flushr+  r'   r  r  r  r  r   r   r  r1   )	r/   rE   r:   r  r  r  rd  r@   r  r   r   r   r  	  s&   




z3TestObjectFile.test_add_object_file_from_filesystemc                 C   sX   t t}tj|}| D ]}| r)| t	|
 d | |
  t qd S )N   )r  fromhexissue_632_elfr'   r  r  r  r  r1   rq   r  hexissue_632_text)r/   elfr  r   r   r   r   test_get_section_content	  s   
z'TestObjectFile.test_get_section_contentN)rF   rG   rH   r  r  r  r  r  r   r   r   r   r  	  s    r  c                   @   r7  )TestLLVMLockCallbacksc                    s   g   fdd} fdd}t j|| |   t   | d  | d  t j|| | t t j|| W d    d S 1 sIw   Y  d S )Nc                           d d S )Nacqr  r   eventsr   r   r  	  r   z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.acqc                      r  )Nrelr  r   r  r   r   r  	  r   z6TestLLVMLockCallbacks.test_lock_callbacks.<locals>.relr  r  )	r'   r
   register_lock_callbackr  create_new_module_pass_managerr   unregister_lock_callbackr   r1  )r/   r  r  r   r  r   test_lock_callbacks	  s   
"z)TestLLVMLockCallbacks.test_lock_callbacksN)rF   rG   rH   r  r   r   r   r   r  	  s    r  c                   @   r  )TestPipelineTuningOptionsc                 C      t  S r&   )r'   create_pipeline_tuning_optionsr.   r   r   r   pto
     zTestPipelineTuningOptions.ptoc                 C      |   }|  d S r&   )r  r+  r/   r  r   r   r   r/  
     z$TestPipelineTuningOptions.test_closec                 C   <   |   }| |jt tdD ]}||_| |j| qd S rG  )r  r   speed_levelr|  r  r1   r/   r  r  r   r   r   test_speed_level
     z*TestPipelineTuningOptions.test_speed_levelc                 C   r  Nr  )r  r   
size_levelr|  r  r1   r  r   r   r   test_size_level
  r  z)TestPipelineTuningOptions.test_size_levelc                 C   s*   |   }| |jt dD ]}||_qd S )N)   P   i^  )r  r   inlining_thresholdr|  r  r   r   r   test_inlining_threshold 
  s
   z1TestPipelineTuningOptions.test_inlining_thresholdc                 C   8   |   }| |jt dD ]}||_| |j| qd S N)TF)r  r   loop_interleavingr   r1   r/   r  r   r   r   r   test_loop_interleaving&
     z0TestPipelineTuningOptions.test_loop_interleavingc                 C   r  r  )r  r   loop_vectorizationr   r1   r  r   r   r   test_loop_vectorization-
  r  z1TestPipelineTuningOptions.test_loop_vectorizationc                 C   r  r  )r  r   slp_vectorizationr   r1   r  r   r   r   test_slp_vectorization4
  r  z0TestPipelineTuningOptions.test_slp_vectorizationc                 C   r  r  )r  r   loop_unrollingr   r1   r  r   r   r   test_loop_unrolling;
  r  z-TestPipelineTuningOptions.test_loop_unrollingc                 C   sn   |   }| t d|_W d    n1 sw   Y  | t d|_W d    d S 1 s0w   Y  d S )NrL  r  )r  r   r1  r  r  r   r   r   test_speed_level_constraintsB
  s   "z6TestPipelineTuningOptions.test_speed_level_constraintsc                 C   s   |   }| t d|_W d    n1 sw   Y  | t d|_W d    n1 s/w   Y  | t d|_d|_W d    d S 1 sKw   Y  d S )Nr  r  rU   )r  r   r1  r  r  r  r   r   r   test_size_level_constraintsI
  s   "z5TestPipelineTuningOptions.test_size_level_constraintsN)rF   rG   rH   r  r/  r  r  r  r  r  r  r  r  r  r   r   r   r   r  	
  s    r  c                   @   s   e Zd ZdddZdS )NewPassManagerMixinr   c                 C   s(   | j dd}t||}t||}|S )NFr?   )rE   r'   r  create_pass_builder)r/   r  r  r9  r  pbr   r   r   r  V
  s   zNewPassManagerMixin.pbN)r   r   )rF   rG   rH   r  r   r   r   r   r  T
  s    r  c                   @   L   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S )TestPassBuilderc                 C   r  r&   )r  r+  )r/   r  r   r   r   r/  _
  r  zTestPassBuilder.test_closec                 C   sN   | j dd}tdd}d|_d|_d|_d|_d|_t||}|	  d S )NFr?   r  r   rU   T)
rE   r'   r  r  r  r  r  r  r  r+  )r/   r9  r  r  r   r   r   test_ptoc
  s   zTestPassBuilder.test_ptoc                 C   s,   |   }| }||  | |  d S r&   )r  getModulePassManagerrunr;   r+  )r/   r  mpmr   r   r   test_get_module_pass_managern
  s   z,TestPassBuilder.test_get_module_pass_managerc                 C   s2   |   }| }||  d| |  d S r  )r  getFunctionPassManagerr  r;   r<  r+  )r/   r  fpmr   r   r   test_get_function_pass_managert
  s   z.TestPassBuilder.test_get_function_pass_managerc                    s\    fdd}|d}|d}  |t   |t  |dd  |dd dS )z:Test pass timing reports for O3 and O0 optimization levelsc                    sF      } j| dd}|  | }||| | }|  |S Nr   r  r  )r;   r  start_pass_timingr  r  finish_pass_timingr+  )r  r:   r  r  reportr.   r   r   run_with_timing|
  s   z9TestPassBuilder.test_time_passes.<locals>.run_with_timingr  r   zPass execution timing reportrT   N)r   r   r1   count)r/   r  	report_O3	report_O0r   r.   r   test_time_passesz
  s   
z TestPassBuilder.test_time_passesc                 C   sJ   |   }|  }| }||| |  | }|  | | d S r&   )r;   r  r  r  r  r  r+  r  )r/   r:   r  r  r  r   r   r   test_empty_report
  s   z!TestPassBuilder.test_empty_reportc                 C   st   |   }|  }|  | }||| |  | td |  W d    n1 s/w   Y  |  d S )Nzonly be done once)	r;   r  r  r  r  r  r  r   r+  r/   r:   r  r  r   r   r   test_multiple_timers_error
  s   
z*TestPassBuilder.test_multiple_timers_errorc                 C   sd   |   }|  }| }||| | td |  W d    n1 s'w   Y  |  d S )Nznot enabled)r;   r  r  r  r  r   r  r+  r  r   r   r   test_empty_report_error
  s   
z'TestPassBuilder.test_empty_report_errorN)rF   rG   rH   r/  r  r  r  r  r  r  r  r   r   r   r   r  ]
  s    
r  c                   @   r  )TestNewModulePassManagerc                 C   r  r&   )r'   r  r.   r   r   r   pm
  r  zTestNewModulePassManager.pmc                 C   sB   |   }t|}| j|dd}| }||| t|}||fS r  )r;   r   r  r  r  )r/   levelr:   orig_asmr  r  optimized_asmr   r   r   run_o_n
  s   z TestNewModulePassManager.run_o_nc                 C   r  r&   r  r+  r/   r  r   r   r   r/  
  r  z#TestNewModulePassManager.test_closec                 C   *   |  d\}}| d| | d| d S Nr  %.4r  r   rJ  r/   r  r  r   r   r   test_run_o3
     z$TestNewModulePassManager.test_run_o3c                 C   *   |  d\}}| d| | d| d S Nr   r  r  r   r  r   r   r   test_run_o0
  r  z$TestNewModulePassManager.test_run_o0c                 C   sZ   |   }|  }|  | t}t|}||| t|}| d| | d| d S )N%.3)	r  r  add_instruction_combine_passr;   asm_sum4r   r  r   rJ  )r/   r  r  r:   r  r  r   r   r   test_instcombine
  s   
z)TestNewModulePassManager.test_instcombinec                 C   s   | j ddd}ttdd}t|}| }||| t|}| d| | 	d| tt}| }t|}||| t|}| d| | d| d S )Nr  r   r  optnone r   alloca)
r  r   asm_alloca_optnonereplacer'   r7   r  r  r   rJ  )r/   r  r  r:   r  r  orig_asm_optnoneoptimized_asm_optnoner   r   r   test_optnone
  s   

z%TestNewModulePassManager.test_optnonec                 C   s  |   }|  |  |  |  |  |  |  |  |	  |
  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |   |  |!  |"  |#  |$  |%  |&  |'  |(  |)  |*  |+  |,  |-  |.  |/  |0  |1  |2  |3  |4  |5  |6  |7  |8  |9  |:  |;  |<  d S r&   )=r  add_argument_promotion_pass'add_post_order_function_attributes_passadd_verifieradd_constant_merge_passadd_dead_arg_elimination_passadd_dot_call_graph_printer_passadd_always_inliner_passadd_rpo_function_attrs_pass#add_global_dead_code_eliminate_passadd_global_opt_passadd_ipsccp_passadd_internalize_passadd_loop_extract_passadd_merge_functions_passadd_partial_inliner_passadd_strip_symbols_passadd_strip_dead_debug_info_passadd_strip_dead_prototype_passadd_strip_debug_declare_pass add_strip_non_debug_symbols_passadd_aa_eval_passadd_simplify_cfg_passadd_loop_unroll_passr  add_jump_threading_passadd_cfg_printer_passadd_cfg_only_printer_passadd_dom_printer_passadd_dom_only_printer_passadd_post_dom_printer_passadd_post_dom_only_printer_passadd_dom_viewer_passadd_post_dom_viewer_passadd_post_dom_only_viewer_passadd_lint_passadd_aggressive_dce_passadd_break_critical_edges_passadd_dead_store_elimination_passadd_dead_code_elimination_passadd_aggressive_instcombine_passadd_lcssa_passadd_new_gvn_passadd_loop_simplify_passadd_loop_unroll_and_jam_passadd_sccp_passadd_lower_atomic_passadd_lower_invoke_passadd_lower_switch_passadd_mem_copy_opt_pass"add_unify_function_exit_nodes_passadd_reassociate_passadd_register_to_memory_passadd_sroa_passadd_sinking_passadd_tail_call_elimination_passadd_instruction_namer_passadd_loop_deletion_passadd_loop_strength_reduce_passadd_loop_rotate_passadd_refprune_passr  r   r   r   test_add_passes
  s|   z(TestNewModulePassManager.test_add_passesN)rF   rG   rH   r  r  r/  r  r  r  r  r;  r   r   r   r   r  
  s    	r  c                   @   sT   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S )TestNewFunctionPassManagerc                 C   r  r&   )r'    create_new_function_pass_managerr.   r   r   r   r  (  r  zTestNewFunctionPassManager.pmc                 C   r  r&   r  r/   r  r   r   r   r/  +  r  z%TestNewFunctionPassManager.test_closec                 C   sL   |   }|d}t|}| j|dd}| }||| t|}||fS )Nr;  r   r  )r;   r<  r   r  r  r  )r/   r  r:   funr  r  r  r  r   r   r   r  /  s   
z"TestNewFunctionPassManager.run_o_nc                 C   r  r  r  r  r   r   r   r  9  r  z&TestNewFunctionPassManager.test_run_o3c                 C   r  r  r  r  r   r   r   r  >  r  z&TestNewFunctionPassManager.test_run_o0c                 C   s   | j ddd}ttdd}t|d}| }||| t|}| 	d| | 
d| tt}t|d}| }||| t|}| 	d| | 	d| d S )Nr  r   r  r  r   r   r  )r  r   r  r  r'   r7   r<  r  r  r   rJ  )r/   r  r  r?  r  r  r  r  r   r   r   r  C  s   z'TestNewFunctionPassManager.test_optnonec                 C   s`   |   }|  }| td}|  t|}||| t|}| d| | 	d| d S )Nr;  r  )
r  r  r;   r  r<  r  r   r  r   rJ  )r/   r  r  r?  r  r  r   r   r   r  V  s   z+TestNewFunctionPassManager.test_instcombinec                 C   s4   |  d}| }ttjD ]}||| qd S r  )r  r  r'   r7   asm_declarationrZ  r  )r/   r  r  r?  r   r   r   test_declarationsb  s
   
z,TestNewFunctionPassManager.test_declarationsc                 C   sT  |   }|  |  |  |  |  |  |  |  |	  |
  |  |  |	  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |  |   |!  |"  |#  |$  |%  |&  |'  |(  d S 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/  r0  r1  r2  r3  r4  r5  r6  r7  r8  r9  r:  r>  r   r   r   r;  h  sT   z*TestNewFunctionPassManager.test_add_passesN)rF   rG   rH   r  r/  r  r  r  r  r  rA  r;  r   r   r   r   r<  '  s    
r<  rL   rM   c                   @   s   e Zd Zeg deg dddeg dieg deg ddd	Zeg d
eg dddeg dieg deg ddd	Zdd Zedd Ze	dd Z
dS )	TestBuild)	pthreadrD  dlr   gcc_sr   r  stdc++ld-linux-x86-64)rC  rD  rD  r   rE  r   r  rF  )r  aarch64amd64)advapi32kernel32ntdllmsvcp140vcruntime140vcruntime140_1api-ms-win-crt-convert-l1-1-0!api-ms-win-crt-environment-l1-1-0api-ms-win-crt-heap-l1-1-0api-ms-win-crt-locale-l1-1-0api-ms-win-crt-math-l1-1-0api-ms-win-crt-runtime-l1-1-0api-ms-win-crt-stdio-l1-1-0api-ms-win-crt-string-l1-1-0api-ms-win-crt-time-l1-1-0api-ms-win-crt-utility-l1-1-0shell32ole32)llvmliterA  rD  corefoundationc++)r\  rA  rD  r^  )r  arm64)rK   r  darwin)rC  rD  zstdrD  r   rE  r   rG  )rC  rD  ra  rD  r   rE  r   zld-linux-aarch64)rD  ra  rJ  rK  rL  rM  rN  rO  rP  rQ  rR  rS  rT  rU  rV  rW  rX  rY  rZ  r[  )r\  rA  rD  ra  r]  r^  )r\  rA  rD  ra  r^  c           	      C   s   t   }t   }|dkr| j| | }n|dkr$| j| | }ntd| t|d }dd |D }z	| || W d S  t	yt } z'dt
| dt
| d	t|| d
t|| dt|| d}t	||d }~ww )Nr   r   zUnexpected package type: canonicalised_linked_librariesc                 S   s   h | ]}| d rd n|qS )rM  )rw   )rM  r`   r   r   r   r  0  s    z*TestBuild.check_linkage.<locals>.<setcomp>z:Unexpected linkage encountered for libllvmlite:
Expected: z
     Got: z

Difference: z
Only in Expected: z
Only in Got: 
)ru   machinelowerrA  wheel_expectedconda_expectedr1  rl   r1   r  rU  symmetric_difference
difference)	r/   infopackage_typerd  os_nameexpectedr"  r   r  r   r   r   check_linkage#  s8   



zTestBuild.check_linkagec                 C   s:   t j }| |d d | |d d | |d d S )Nllvm_linkage_typestaticllvm_assertions_stateonr   )r'   configget_sysinfor1   rn  r/   rj  r   r   r   test_wheel_build@  s   
zTestBuild.test_wheel_buildc                 C   s^   t j }| |d d | |d d | |d t  dkr-| |d d d S d S )Nro  rp  rq  rr  r   rK   libstdcxx_linkage_type)r'   rs  rt  r1   rn  ru   rA  re  ru  r   r   r   test_conda_buildG  s   
zTestBuild.test_conda_buildN)rF   rG   rH   rl   rf  rg  rn  is_wheel_packagerv  is_conda_packagerx  r   r   r   r   rB    s,    
!
	


4
	

9
rB  __main__)lr,  r  r   r   r   ctypes.utilr   r*   r   rb   ru   rj   re   r   r   
contextlibr   tempfiler   r\  r   r	   r'   llvmlite.bindingr
   llvmlite.testsr   rd  r=  r   package_formatrz  ry  
_HAVE_LIEFliefImportError
needs_liefr   rI   rq  r{  r  rh  r  r  rv   r  r   r8  rB  r  r  r   rm  asm_inlineasm2asm_inlineasm3licm_asmr[  r  r   r  r   r   r   r`  r  r@  r  r  asm_tli_exp2r  r  r  r%   rJ   r   r   r(  objectr  r  r  skipr  r  r  r  r/  r8  r<  rB  rR  rY  re  rw   r  r  r  r  r  r  r  r<  rc   r   rB  rF   mainr   r   r   r   <module>   s    





D(-4!2I  D : A  4 w[&$7	XK	Pzn 
;