summaryrefslogtreecommitdiff
path: root/mlir/test/Target/LLVMIR/ifunc.mlir
blob: bba306c1e1ab387b2522fac1af723b15365d7c30 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// RUN: mlir-translate -mlir-to-llvmir %s --split-input-file | FileCheck %s

// CHECK: @foo = dso_local ifunc void (ptr, i32), ptr @resolve_foo
llvm.mlir.ifunc external @foo : !llvm.func<void (ptr, i32)>, !llvm.ptr @resolve_foo {dso_local}
llvm.func @call_foo(%arg0: !llvm.ptr {llvm.noundef}, %arg1: i32 {llvm.noundef}) attributes {dso_local} {
// CHECK: call void @foo
  llvm.call @foo(%arg0, %arg1) : (!llvm.ptr {llvm.noundef}, i32 {llvm.noundef}) -> ()
  llvm.return
}
llvm.func @foo_fptr() -> !llvm.ptr attributes {dso_local} {
  %1 = llvm.mlir.addressof @foo : !llvm.ptr
// CHECK: ret ptr @foo
  llvm.return %1 : !llvm.ptr
}
llvm.func internal @resolve_foo() -> !llvm.ptr attributes {dso_local} {
  %0 = llvm.mlir.addressof @foo_1 : !llvm.ptr
  llvm.return %0 : !llvm.ptr
}
llvm.func @foo_1(!llvm.ptr {llvm.noundef}, i32 {llvm.noundef})

// -----

llvm.mlir.alias external @resolver_alias : !llvm.func<ptr ()> {
  %0 = llvm.mlir.addressof @resolver : !llvm.ptr
  llvm.return %0 : !llvm.ptr
}
llvm.mlir.alias external @resolver_alias_alias : !llvm.func<ptr ()> {
  %0 = llvm.mlir.addressof @resolver_alias : !llvm.ptr
  llvm.return %0 : !llvm.ptr
}

// CHECK-DAG: @ifunc = ifunc float (i64), ptr @resolver_alias
// CHECK-DAG: @ifunc2 = ifunc float (i64), ptr @resolver_alias_alias
llvm.mlir.ifunc external @ifunc2 : !llvm.func<f32 (i64)>, !llvm.ptr @resolver_alias_alias
llvm.mlir.ifunc external @ifunc : !llvm.func<f32 (i64)>, !llvm.ptr @resolver_alias
llvm.func @resolver() -> !llvm.ptr {
  %0 = llvm.mlir.constant(333 : i64) : i64
  %1 = llvm.inttoptr %0 : i64 to !llvm.ptr
  llvm.return %1 : !llvm.ptr
}

// -----

// CHECK: @ifunc = linkonce_odr hidden ifunc
llvm.mlir.ifunc linkonce_odr hidden @ifunc : !llvm.func<f32 (i64)>, !llvm.ptr @resolver {dso_local}
llvm.func @resolver() -> !llvm.ptr {
  %0 = llvm.mlir.constant(333 : i64) : i64
  %1 = llvm.inttoptr %0 : i64 to !llvm.ptr
  llvm.return %1 : !llvm.ptr
}

// -----

// CHECK: @ifunc = private ifunc
llvm.mlir.ifunc private @ifunc : !llvm.func<f32 (i64)>, !llvm.ptr @resolver {dso_local}
llvm.func @resolver() -> !llvm.ptr {
  %0 = llvm.mlir.constant(333 : i64) : i64
  %1 = llvm.inttoptr %0 : i64 to !llvm.ptr
  llvm.return %1 : !llvm.ptr
}

// -----

// CHECK: @ifunc = weak ifunc
llvm.mlir.ifunc weak @ifunc : !llvm.func<f32 (i64)>, !llvm.ptr @resolver
llvm.func @resolver() -> !llvm.ptr {
  %0 = llvm.mlir.constant(333 : i64) : i64
  %1 = llvm.inttoptr %0 : i64 to !llvm.ptr
  llvm.return %1 : !llvm.ptr
}