xref: /llvm-project/mlir/test/Conversion/FuncToLLVM/convert-funcs.mlir (revision eb6c4197d5263ed2e086925b2b2f032a19442d2b)
1// RUN: mlir-opt -convert-func-to-llvm -convert-cf-to-llvm -reconcile-unrealized-casts -split-input-file -verify-diagnostics %s | FileCheck %s
2
3//CHECK: llvm.func @second_order_arg(!llvm.ptr)
4func.func private @second_order_arg(%arg0 : () -> ())
5
6//CHECK: llvm.func @second_order_result() -> !llvm.ptr
7func.func private @second_order_result() -> (() -> ())
8
9//CHECK: llvm.func @second_order_multi_result() -> !llvm.struct<(ptr, ptr, ptr)>
10func.func private @second_order_multi_result() -> (() -> (i32), () -> (i64), () -> (f32))
11
12// Check that memrefs are converted to argument packs if appear as function arguments.
13// CHECK: llvm.func @memref_call_conv(!llvm.ptr, !llvm.ptr, i64, i64, i64)
14func.func private @memref_call_conv(%arg0: memref<?xf32>)
15
16// Same in nested functions.
17// CHECK: llvm.func @memref_call_conv_nested(!llvm.ptr)
18func.func private @memref_call_conv_nested(%arg0: (memref<?xf32>) -> ())
19
20//CHECK-LABEL: llvm.func @pass_through(%arg0: !llvm.ptr) -> !llvm.ptr {
21func.func @pass_through(%arg0: () -> ()) -> (() -> ()) {
22// CHECK-NEXT:  llvm.br ^bb1(%arg0 : !llvm.ptr)
23  cf.br ^bb1(%arg0 : () -> ())
24
25//CHECK-NEXT: ^bb1(%0: !llvm.ptr):
26^bb1(%bbarg: () -> ()):
27// CHECK-NEXT:  llvm.return %0 : !llvm.ptr
28  return %bbarg : () -> ()
29}
30
31// CHECK-LABEL: llvm.func extern_weak @llvmlinkage(i32)
32func.func private @llvmlinkage(i32) attributes { "llvm.linkage" = #llvm.linkage<extern_weak> }
33
34// CHECK-LABEL: llvm.func @llvmreadnone(i32)
35// CHECK-SAME: memory_effects = #llvm.memory_effects<other = none, argMem = none, inaccessibleMem = none>
36func.func private @llvmreadnone(i32) attributes { llvm.readnone }
37
38// CHECK-LABEL: llvm.func @body(i32)
39func.func private @body(i32)
40
41// CHECK-LABEL: llvm.func @indirect_const_call
42// CHECK-SAME: (%[[ARG0:.*]]: i32) {
43func.func @indirect_const_call(%arg0: i32) {
44// CHECK-NEXT: %[[ADDR:.*]] = llvm.mlir.addressof @body : !llvm.ptr
45  %0 = constant @body : (i32) -> ()
46// CHECK-NEXT:  llvm.call %[[ADDR]](%[[ARG0:.*]]) : !llvm.ptr, (i32) -> ()
47  call_indirect %0(%arg0) : (i32) -> ()
48// CHECK-NEXT:  llvm.return
49  return
50}
51
52// CHECK-LABEL: llvm.func @indirect_call(%arg0: !llvm.ptr, %arg1: f32) -> i32 {
53func.func @indirect_call(%arg0: (f32) -> i32, %arg1: f32) -> i32 {
54// CHECK-NEXT:  %0 = llvm.call %arg0(%arg1) : !llvm.ptr, (f32) -> i32
55  %0 = call_indirect %arg0(%arg1) : (f32) -> i32
56// CHECK-NEXT:  llvm.return %0 : i32
57  return %0 : i32
58}
59
60func.func @variadic_func(%arg0: i32) attributes { "func.varargs" = true } {
61  return
62}
63
64// CHECK-LABEL: llvm.func @target_cpu()
65// CHECK-SAME: target_cpu = "gfx90a"
66func.func private @target_cpu() attributes { "target_cpu" = "gfx90a" }
67
68// CHECK-LABEL: llvm.func @target_features()
69// CHECK-SAME: target_features = #llvm.target_features<["+sme", "+sve"]>
70func.func private @target_features() attributes {
71  "target_features" = #llvm.target_features<["+sme", "+sve"]>
72}
73
74// -----
75
76// CHECK-LABEL: llvm.func @private_callee
77// CHECK-SAME: sym_visibility = "private"
78func.func private @private_callee(%arg1: f32) -> i32 {
79  %0 = arith.constant 0 : i32
80  return %0 : i32
81}
82
83// CHECK-LABEL: llvm.func @caller_private_callee
84func.func @caller_private_callee(%arg1: f32) -> i32 {
85  %0 = call @private_callee(%arg1) : (f32) -> i32
86  return %0 : i32
87}
88
89// -----
90
91func.func private @badllvmlinkage(i32) attributes { "llvm.linkage" = 3 : i64 } // expected-error {{Contains llvm.linkage attribute not of type LLVM::LinkageAttr}}
92
93// -----
94
95// expected-error@+1{{C interface for variadic functions is not supported yet.}}
96func.func @variadic_func(%arg0: i32) attributes { "func.varargs" = true, "llvm.emit_c_interface" } {
97  return
98}
99