41 lines · plain
1// RUN: mlir-opt -p 'builtin.module(buffer-results-to-out-params{modify-public-functions})' %s | FileCheck %s2 3// Test if `public` functions' return values are transformed into out parameters4// when `buffer-results-to-out-params` is invoked with `modifyPublicFunctions`.5 6// CHECK-LABEL: func.func @basic(7// CHECK-SAME: %[[ARG0:.*]]: memref<f32>) {8// CHECK: %[[VAL_0:.*]] = "test.source"() : () -> memref<f32>9// CHECK: memref.copy %[[VAL_0]], %[[ARG0]] : memref<f32> to memref<f32>10// CHECK: return11// CHECK: }12func.func @basic() -> (memref<f32>) {13 %0 = "test.source"() : () -> (memref<f32>)14 return %0 : memref<f32>15}16 17// CHECK-LABEL: func.func @presence_of_existing_arguments(18// CHECK-SAME: %[[ARG0:.*]]: memref<1xf32>,19// CHECK-SAME: %[[ARG1:.*]]: memref<2xf32>) {20// CHECK: %[[VAL_0:.*]] = "test.source"() : () -> memref<2xf32>21// CHECK: memref.copy %[[VAL_0]], %[[ARG1]] : memref<2xf32> to memref<2xf32>22// CHECK: return23// CHECK: }24func.func @presence_of_existing_arguments(%arg0: memref<1xf32>) -> (memref<2xf32>) {25 %0 = "test.source"() : () -> (memref<2xf32>)26 return %0 : memref<2xf32>27}28 29// CHECK-LABEL: func.func @multiple_results(30// CHECK-SAME: %[[ARG0:.*]]: memref<1xf32>,31// CHECK-SAME: %[[ARG1:.*]]: memref<2xf32>) {32// CHECK: %[[VAL_0:.*]]:2 = "test.source"() : () -> (memref<1xf32>, memref<2xf32>)33// CHECK: memref.copy %[[VAL_0]]#0, %[[ARG0]] : memref<1xf32> to memref<1xf32>34// CHECK: memref.copy %[[VAL_0]]#1, %[[ARG1]] : memref<2xf32> to memref<2xf32>35// CHECK: return36// CHECK: }37func.func @multiple_results() -> (memref<1xf32>, memref<2xf32>) {38 %0, %1 = "test.source"() : () -> (memref<1xf32>, memref<2xf32>)39 return %0, %1 : memref<1xf32>, memref<2xf32>40}41