130 lines · plain
1// RUN: mlir-opt -convert-func-to-llvm -reconcile-unrealized-casts %s | FileCheck %s2 3// RUN: mlir-opt -convert-func-to-llvm='use-bare-ptr-memref-call-conv=1' -reconcile-unrealized-casts %s | FileCheck %s --check-prefix=BAREPTR4 5// RUN: mlir-opt -transform-interpreter -reconcile-unrealized-casts %s | FileCheck %s --check-prefix=BAREPTR6 7// These tests were separated from func-memref.mlir because applying8// -reconcile-unrealized-casts resulted in `llvm.extractvalue` ops getting9// folded away.10 11// CHECK-LABEL: func @check_static_return12// CHECK-COUNT-2: !llvm.ptr13// CHECK-COUNT-5: i6414// CHECK-SAME: -> !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>15// BAREPTR-LABEL: func @check_static_return16// BAREPTR-SAME: (%[[arg:.*]]: !llvm.ptr) -> !llvm.ptr {17func.func @check_static_return(%static : memref<32x18xf32>) -> memref<32x18xf32> {18// CHECK: llvm.return %{{.*}} : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>19 20// BAREPTR: %[[udf:.*]] = llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>21// BAREPTR-NEXT: %[[base0:.*]] = llvm.insertvalue %[[arg]], %[[udf]][0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>22// BAREPTR-NEXT: %[[aligned:.*]] = llvm.insertvalue %[[arg]], %[[base0]][1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>23// BAREPTR-NEXT: %[[val0:.*]] = llvm.mlir.constant(0 : index) : i6424// BAREPTR-NEXT: %[[ins0:.*]] = llvm.insertvalue %[[val0]], %[[aligned]][2] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>25// BAREPTR-NEXT: %[[val1:.*]] = llvm.mlir.constant(32 : index) : i6426// BAREPTR-NEXT: %[[ins1:.*]] = llvm.insertvalue %[[val1]], %[[ins0]][3, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>27// BAREPTR-NEXT: %[[val2:.*]] = llvm.mlir.constant(18 : index) : i6428// BAREPTR-NEXT: %[[ins2:.*]] = llvm.insertvalue %[[val2]], %[[ins1]][4, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>29// BAREPTR-NEXT: %[[val3:.*]] = llvm.mlir.constant(18 : index) : i6430// BAREPTR-NEXT: %[[ins3:.*]] = llvm.insertvalue %[[val3]], %[[ins2]][3, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>31// BAREPTR-NEXT: %[[val4:.*]] = llvm.mlir.constant(1 : index) : i6432// BAREPTR-NEXT: %[[ins4:.*]] = llvm.insertvalue %[[val4]], %[[ins3]][4, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>33// BAREPTR-NEXT: %[[base1:.*]] = llvm.extractvalue %[[ins4]][0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>34// BAREPTR-NEXT: llvm.return %[[base1]] : !llvm.ptr35 return %static : memref<32x18xf32>36}37 38// CHECK-LABEL: func @check_static_return_with_offset39// CHECK-COUNT-2: !llvm.ptr40// CHECK-COUNT-5: i6441// CHECK-SAME: -> !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>42// BAREPTR-LABEL: func @check_static_return_with_offset43// BAREPTR-SAME: (%[[arg:.*]]: !llvm.ptr) -> !llvm.ptr {44func.func @check_static_return_with_offset(%static : memref<32x18xf32, strided<[22,1], offset: 7>>) -> memref<32x18xf32, strided<[22,1], offset: 7>> {45// CHECK: llvm.return %{{.*}} : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>46 47// BAREPTR: %[[udf:.*]] = llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>48// BAREPTR-NEXT: %[[base0:.*]] = llvm.insertvalue %[[arg]], %[[udf]][0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>49// BAREPTR-NEXT: %[[aligned:.*]] = llvm.insertvalue %[[arg]], %[[base0]][1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>50// BAREPTR-NEXT: %[[val0:.*]] = llvm.mlir.constant(7 : index) : i6451// BAREPTR-NEXT: %[[ins0:.*]] = llvm.insertvalue %[[val0]], %[[aligned]][2] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>52// BAREPTR-NEXT: %[[val1:.*]] = llvm.mlir.constant(32 : index) : i6453// BAREPTR-NEXT: %[[ins1:.*]] = llvm.insertvalue %[[val1]], %[[ins0]][3, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>54// BAREPTR-NEXT: %[[val2:.*]] = llvm.mlir.constant(22 : index) : i6455// BAREPTR-NEXT: %[[ins2:.*]] = llvm.insertvalue %[[val2]], %[[ins1]][4, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>56// BAREPTR-NEXT: %[[val3:.*]] = llvm.mlir.constant(18 : index) : i6457// BAREPTR-NEXT: %[[ins3:.*]] = llvm.insertvalue %[[val3]], %[[ins2]][3, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>58// BAREPTR-NEXT: %[[val4:.*]] = llvm.mlir.constant(1 : index) : i6459// BAREPTR-NEXT: %[[ins4:.*]] = llvm.insertvalue %[[val4]], %[[ins3]][4, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>60// BAREPTR-NEXT: %[[base1:.*]] = llvm.extractvalue %[[ins4]][0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>61// BAREPTR-NEXT: llvm.return %[[base1]] : !llvm.ptr62 return %static : memref<32x18xf32, strided<[22,1], offset: 7>>63}64 65 66// BAREPTR: llvm.func @foo(!llvm.ptr) -> !llvm.ptr67func.func private @foo(memref<10xi8>) -> memref<20xi8>68 69// BAREPTR-LABEL: func @check_memref_func_call70// BAREPTR-SAME: %[[in:.*]]: !llvm.ptr) -> !llvm.ptr71func.func @check_memref_func_call(%in : memref<10xi8>) -> memref<20xi8> {72 // BAREPTR: %[[inDesc:.*]] = llvm.insertvalue %{{.*}}, %{{.*}}[4, 0]73 // BAREPTR-NEXT: %[[barePtr:.*]] = llvm.extractvalue %[[inDesc]][1] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>74 // BAREPTR-NEXT: %[[call:.*]] = llvm.call @foo(%[[barePtr]]) : (!llvm.ptr) -> !llvm.ptr75 // BAREPTR-NEXT: %[[desc0:.*]] = llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>76 // BAREPTR-NEXT: %[[desc1:.*]] = llvm.insertvalue %[[call]], %[[desc0]][0] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>77 // BAREPTR-NEXT: %[[desc2:.*]] = llvm.insertvalue %[[call]], %[[desc1]][1] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>78 // BAREPTR-NEXT: %[[c0:.*]] = llvm.mlir.constant(0 : index) : i6479 // BAREPTR-NEXT: %[[desc4:.*]] = llvm.insertvalue %[[c0]], %[[desc2]][2] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>80 // BAREPTR-NEXT: %[[c20:.*]] = llvm.mlir.constant(20 : index) : i6481 // BAREPTR-NEXT: %[[desc6:.*]] = llvm.insertvalue %[[c20]], %[[desc4]][3, 0] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>82 // BAREPTR-NEXT: %[[c1:.*]] = llvm.mlir.constant(1 : index) : i6483 // BAREPTR-NEXT: %[[outDesc:.*]] = llvm.insertvalue %[[c1]], %[[desc6]][4, 0] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>84 %res = call @foo(%in) : (memref<10xi8>) -> (memref<20xi8>)85 // BAREPTR-NEXT: %[[res:.*]] = llvm.extractvalue %[[outDesc]][0] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>86 // BAREPTR-NEXT: llvm.return %[[res]] : !llvm.ptr87 return %res : memref<20xi8>88}89 90// BAREPTR-LABEL: func @check_return(91// BAREPTR-SAME: %{{.*}}: memref<?xi8>) -> memref<?xi8>92func.func @check_return(%in : memref<?xi8>) -> memref<?xi8> {93 // BAREPTR: llvm.return {{.*}} : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>94 return %in : memref<?xi8>95}96 97// BAREPTR-LABEL: func @unconvertible_multiresult98// BAREPTR-SAME: %{{.*}}: memref<?xf32>, %{{.*}}: memref<?xf32>) -> (memref<?xf32>, memref<?xf32>)99func.func @unconvertible_multiresult(%arg0: memref<?xf32> , %arg1: memref<?xf32>) -> (memref<?xf32>, memref<?xf32>) {100 return %arg0, %arg1 : memref<?xf32>, memref<?xf32>101}102 103// BAREPTR-LABEL: func @unranked_memref(104// BAREPTR-SAME: %{{.*}}: memref<*xi32>)105func.func @unranked_memref(%arg0:memref<*xi32>) {106 // BAREPTR: call @printMemrefI32(%arg{{.*}}) : (memref<*xi32>) -> ()107 // BAREPTR-NEXT: llvm.return108 call @printMemrefI32(%arg0) : (memref<*xi32>) -> ()109 return110}111func.func private @printMemrefI32(memref<*xi32>)112 113module attributes {transform.with_named_sequence} {114 transform.named_sequence @__transform_main(115 %toplevel_module: !transform.any_op {transform.readonly}) {116 %func = transform.structured.match ops{["func.func"]} in %toplevel_module117 : (!transform.any_op) -> !transform.any_op118 transform.apply_conversion_patterns to %func {119 transform.apply_conversion_patterns.func.func_to_llvm120 } with type_converter {121 transform.apply_conversion_patterns.memref.memref_to_llvm_type_converter122 {use_bare_ptr_call_conv = true, use_opaque_pointers = true}123 } {124 legal_dialects = ["llvm"],125 partial_conversion126 } : !transform.any_op127 transform.yield128 }129}130