1// RUN: mlir-opt -convert-func-to-llvm -reconcile-unrealized-casts %s | FileCheck %s 2// RUN: mlir-opt -convert-func-to-llvm='use-bare-ptr-memref-call-conv=1' -split-input-file %s | FileCheck %s --check-prefix=BAREPTR 3 4// These tests were separated from func-memref.mlir because applying 5// -reconcile-unrealized-casts resulted in `llvm.extractvalue` ops getting 6// folded away. 7 8// CHECK-LABEL: func @check_static_return 9// CHECK-COUNT-2: !llvm.ptr<f32> 10// CHECK-COUNT-5: i64 11// CHECK-SAME: -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 12// BAREPTR-LABEL: func @check_static_return 13// BAREPTR-SAME: (%[[arg:.*]]: !llvm.ptr<f32>) -> !llvm.ptr<f32> { 14func.func @check_static_return(%static : memref<32x18xf32>) -> memref<32x18xf32> { 15// CHECK: llvm.return %{{.*}} : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 16 17// BAREPTR: %[[udf:.*]] = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 18// BAREPTR-NEXT: %[[base0:.*]] = llvm.insertvalue %[[arg]], %[[udf]][0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 19// BAREPTR-NEXT: %[[aligned:.*]] = llvm.insertvalue %[[arg]], %[[base0]][1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 20// BAREPTR-NEXT: %[[val0:.*]] = llvm.mlir.constant(0 : index) : i64 21// BAREPTR-NEXT: %[[ins0:.*]] = llvm.insertvalue %[[val0]], %[[aligned]][2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 22// BAREPTR-NEXT: %[[val1:.*]] = llvm.mlir.constant(32 : index) : i64 23// BAREPTR-NEXT: %[[ins1:.*]] = llvm.insertvalue %[[val1]], %[[ins0]][3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 24// BAREPTR-NEXT: %[[val2:.*]] = llvm.mlir.constant(18 : index) : i64 25// BAREPTR-NEXT: %[[ins2:.*]] = llvm.insertvalue %[[val2]], %[[ins1]][4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 26// BAREPTR-NEXT: %[[val3:.*]] = llvm.mlir.constant(18 : index) : i64 27// BAREPTR-NEXT: %[[ins3:.*]] = llvm.insertvalue %[[val3]], %[[ins2]][3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 28// BAREPTR-NEXT: %[[val4:.*]] = llvm.mlir.constant(1 : index) : i64 29// BAREPTR-NEXT: %[[ins4:.*]] = llvm.insertvalue %[[val4]], %[[ins3]][4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 30// BAREPTR-NEXT: %[[base1:.*]] = llvm.extractvalue %[[ins4]][1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 31// BAREPTR-NEXT: llvm.return %[[base1]] : !llvm.ptr<f32> 32 return %static : memref<32x18xf32> 33} 34 35// ----- 36 37// CHECK-LABEL: func @check_static_return_with_offset 38// CHECK-COUNT-2: !llvm.ptr<f32> 39// CHECK-COUNT-5: i64 40// CHECK-SAME: -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 41// BAREPTR-LABEL: func @check_static_return_with_offset 42// BAREPTR-SAME: (%[[arg:.*]]: !llvm.ptr<f32>) -> !llvm.ptr<f32> { 43func.func @check_static_return_with_offset(%static : memref<32x18xf32, offset:7, strides:[22,1]>) -> memref<32x18xf32, offset:7, strides:[22,1]> { 44// CHECK: llvm.return %{{.*}} : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 45 46// BAREPTR: %[[udf:.*]] = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 47// BAREPTR-NEXT: %[[base0:.*]] = llvm.insertvalue %[[arg]], %[[udf]][0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 48// BAREPTR-NEXT: %[[aligned:.*]] = llvm.insertvalue %[[arg]], %[[base0]][1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 49// BAREPTR-NEXT: %[[val0:.*]] = llvm.mlir.constant(7 : index) : i64 50// BAREPTR-NEXT: %[[ins0:.*]] = llvm.insertvalue %[[val0]], %[[aligned]][2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 51// BAREPTR-NEXT: %[[val1:.*]] = llvm.mlir.constant(32 : index) : i64 52// BAREPTR-NEXT: %[[ins1:.*]] = llvm.insertvalue %[[val1]], %[[ins0]][3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 53// BAREPTR-NEXT: %[[val2:.*]] = llvm.mlir.constant(22 : index) : i64 54// BAREPTR-NEXT: %[[ins2:.*]] = llvm.insertvalue %[[val2]], %[[ins1]][4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 55// BAREPTR-NEXT: %[[val3:.*]] = llvm.mlir.constant(18 : index) : i64 56// BAREPTR-NEXT: %[[ins3:.*]] = llvm.insertvalue %[[val3]], %[[ins2]][3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 57// BAREPTR-NEXT: %[[val4:.*]] = llvm.mlir.constant(1 : index) : i64 58// BAREPTR-NEXT: %[[ins4:.*]] = llvm.insertvalue %[[val4]], %[[ins3]][4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 59// BAREPTR-NEXT: %[[base1:.*]] = llvm.extractvalue %[[ins4]][1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> 60// BAREPTR-NEXT: llvm.return %[[base1]] : !llvm.ptr<f32> 61 return %static : memref<32x18xf32, offset:7, strides:[22,1]> 62} 63 64// ----- 65 66// BAREPTR: llvm.func @foo(!llvm.ptr<i8>) -> !llvm.ptr<i8> 67func.func private @foo(memref<10xi8>) -> memref<20xi8> 68 69// BAREPTR-LABEL: func @check_memref_func_call 70// BAREPTR-SAME: %[[in:.*]]: !llvm.ptr<i8>) -> !llvm.ptr<i8> 71func.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<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 74 // BAREPTR-NEXT: %[[call:.*]] = llvm.call @foo(%[[barePtr]]) : (!llvm.ptr<i8>) -> !llvm.ptr<i8> 75 // BAREPTR-NEXT: %[[desc0:.*]] = llvm.mlir.undef : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 76 // BAREPTR-NEXT: %[[desc1:.*]] = llvm.insertvalue %[[call]], %[[desc0]][0] : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 77 // BAREPTR-NEXT: %[[desc2:.*]] = llvm.insertvalue %[[call]], %[[desc1]][1] : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 78 // BAREPTR-NEXT: %[[c0:.*]] = llvm.mlir.constant(0 : index) : i64 79 // BAREPTR-NEXT: %[[desc4:.*]] = llvm.insertvalue %[[c0]], %[[desc2]][2] : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 80 // BAREPTR-NEXT: %[[c20:.*]] = llvm.mlir.constant(20 : index) : i64 81 // BAREPTR-NEXT: %[[desc6:.*]] = llvm.insertvalue %[[c20]], %[[desc4]][3, 0] : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 82 // BAREPTR-NEXT: %[[c1:.*]] = llvm.mlir.constant(1 : index) : i64 83 // BAREPTR-NEXT: %[[outDesc:.*]] = llvm.insertvalue %[[c1]], %[[desc6]][4, 0] : !llvm.struct<(ptr<i8>, ptr<i8>, 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]][1] : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 86 // BAREPTR-NEXT: llvm.return %[[res]] : !llvm.ptr<i8> 87 return %res : memref<20xi8> 88} 89 90// ----- 91 92// BAREPTR-LABEL: func @check_return( 93// BAREPTR-SAME: %{{.*}}: memref<?xi8>) -> memref<?xi8> 94func.func @check_return(%in : memref<?xi8>) -> memref<?xi8> { 95 // BAREPTR: llvm.return {{.*}} : !llvm.struct<(ptr<i8>, ptr<i8>, i64, array<1 x i64>, array<1 x i64>)> 96 return %in : memref<?xi8> 97} 98