113 lines · plain
1// RUN: mlir-opt -buffer-results-to-out-params -split-input-file -verify-diagnostics %s | FileCheck %s2 3// CHECK-LABEL: func private @basic(4// CHECK-SAME: %[[ARG:.*]]: memref<f32>) {5// CHECK: %[[RESULT:.*]] = "test.source"() : () -> memref<f32>6// CHECK: memref.copy %[[RESULT]], %[[ARG]] : memref<f32> to memref<f32>7// CHECK: return8// CHECK: }9func.func private @basic() -> (memref<f32>) {10 %0 = "test.source"() : () -> (memref<f32>)11 return %0 : memref<f32>12}13 14// CHECK-LABEL: func private @presence_of_existing_arguments(15// CHECK-SAME: %[[ARG0:.*]]: memref<1xf32>,16// CHECK-SAME: %[[ARG1:.*]]: memref<2xf32>) {17// CHECK: %[[RESULT:.*]] = "test.source"() : () -> memref<2xf32>18// CHECK: memref.copy %[[RESULT]], %[[ARG1]] : memref<2xf32> to memref<2xf32>19// CHECK: return20// CHECK: }21func.func private @presence_of_existing_arguments(%arg0: memref<1xf32>) -> (memref<2xf32>) {22 %0 = "test.source"() : () -> (memref<2xf32>)23 return %0 : memref<2xf32>24}25 26// CHECK-LABEL: func private @multiple_results(27// CHECK-SAME: %[[ARG0:.*]]: memref<1xf32>,28// CHECK-SAME: %[[ARG1:.*]]: memref<2xf32>) {29// CHECK: %[[RESULTS:.*]]:2 = "test.source"() : () -> (memref<1xf32>, memref<2xf32>)30// CHECK: memref.copy %[[RESULTS]]#0, %[[ARG0]] : memref<1xf32> to memref<1xf32>31// CHECK: memref.copy %[[RESULTS]]#1, %[[ARG1]] : memref<2xf32> to memref<2xf32>32// CHECK: return33// CHECK: }34func.func private @multiple_results() -> (memref<1xf32>, memref<2xf32>) {35 %0, %1 = "test.source"() : () -> (memref<1xf32>, memref<2xf32>)36 return %0, %1 : memref<1xf32>, memref<2xf32>37}38 39// CHECK-LABEL: func private @non_memref_types(40// CHECK-SAME: %[[OUTPARAM:.*]]: memref<f32>) -> (i1, i32) {41// CHECK: %[[RESULT1:.*]]:3 = "test.source"() : () -> (i1, memref<f32>, i32)42// CHECK: memref.copy %[[RESULT1]]#1, %[[OUTPARAM]] : memref<f32> to memref<f32>43// CHECK: return %[[RESULT1]]#0, %[[RESULT1]]#2 : i1, i3244// CHECK: }45func.func private @non_memref_types() -> (i1, memref<f32>, i32) {46 %0, %1, %2 = "test.source"() : () -> (i1, memref<f32>, i32)47 return %0, %1, %2 : i1, memref<f32>, i3248}49 50// CHECK: func private @external_function() -> memref<f32>51func.func private @external_function() -> (memref<f32>)52// CHECK: func private @result_attrs() -> (memref<f32> {test.some_attr})53func.func private @result_attrs() -> (memref<f32> {test.some_attr})54// CHECK: func private @mixed_result_attrs() -> (memref<1xf32>, memref<2xf32> {test.some_attr}, memref<3xf32>)55func.func private @mixed_result_attrs() -> (memref<1xf32>, memref<2xf32> {test.some_attr}, memref<3xf32>)56 57// -----58 59// CHECK-LABEL: func private @callee() -> memref<1xf32>60func.func private @callee() -> memref<1xf32>61 62// CHECK-LABEL: func @call_basic() {63// CHECK: %[[OUTPARAM:.*]] = call @callee() : () -> memref<1xf32> 64// CHECK: "test.sink"(%[[OUTPARAM]]) : (memref<1xf32>) -> ()65// CHECK: return66// CHECK: }67func.func @call_basic() {68 %0 = call @callee() : () -> memref<1xf32>69 "test.sink"(%0) : (memref<1xf32>) -> ()70 return71}72 73// -----74 75// CHECK-LABEL: func private @callee() -> (memref<1xf32>, memref<2xf32>)76func.func private @callee() -> (memref<1xf32>, memref<2xf32>)77 78// CHECK-LABEL: func @call_multiple_result() {79// CHECK: %[[RESULTS:.*]]:2 = call @callee() : () -> (memref<1xf32>, memref<2xf32>)80// CHECK: "test.sink"(%[[RESULTS]]#0, %[[RESULTS]]#1) : (memref<1xf32>, memref<2xf32>) -> ()81// CHECK: }82func.func @call_multiple_result() {83 %0, %1 = call @callee() : () -> (memref<1xf32>, memref<2xf32>)84 "test.sink"(%0, %1) : (memref<1xf32>, memref<2xf32>) -> ()85}86 87// -----88 89// CHECK-LABEL: func private @callee() -> (i1, memref<1xf32>, i32)90func.func private @callee() -> (i1, memref<1xf32>, i32)91 92// CHECK-LABEL: func @call_non_memref_result() {93// CHECK: %[[RESULTS:.*]]:3 = call @callee() : () -> (i1, memref<1xf32>, i32)94// CHECK: "test.sink"(%[[RESULTS]]#0, %[[RESULTS]]#1, %[[RESULTS]]#2) : (i1, memref<1xf32>, i32) -> ()95// CHECK: }96func.func @call_non_memref_result() {97 %0, %1, %2 = call @callee() : () -> (i1, memref<1xf32>, i32)98 "test.sink"(%0, %1, %2) : (i1, memref<1xf32>, i32) -> ()99}100 101// -----102 103func.func private @callee(%size: index) -> (memref<?xf32>) {104 %alloc = memref.alloc(%size) : memref<?xf32>105 return %alloc : memref<?xf32>106}107 108func.func @call_non_memref_result(%size: index) {109 // expected-error @+1 {{cannot create out param for dynamically shaped result}}110 %0 = call @callee(%size) : (index) -> (memref<?xf32>)111 "test.sink"(%0) : (memref<?xf32>) -> ()112}113