80 lines · plain
1// RUN: mlir-opt -allow-unregistered-dialect -p 'builtin.module(buffer-results-to-out-params{hoist-dynamic-allocs})' %s -split-input-file | FileCheck %s2 3func.func private @single_alloc(%size : index) -> (memref<?xf32>) {4 %alloc = memref.alloc(%size) : memref<?xf32>5 return %alloc : memref<?xf32>6}7 8func.func @single_alloc_test(%size : index) {9 %alloc = call @single_alloc(%size) : (index) -> (memref<?xf32>)10 "test.sink"(%alloc) : (memref<?xf32>) -> ()11}12 13// CHECK-LABEL: func.func private @single_alloc(14// CHECK-SAME: %{{.*}}: index,15// CHECK-SAME: %{{.*}}: memref<?xf32>) {16 17// CHECK-LABEL: func.func @single_alloc_test(18// CHECK-SAME: %[[size:.*]]: index) {19// CHECK: %[[alloc:.*]] = memref.alloc(%[[size]]) : memref<?xf32>20// CHECK: call @single_alloc(%[[size]], %[[alloc]]) : (index, memref<?xf32>) -> ()21// CHECK: "test.sink"(%[[alloc]]) : (memref<?xf32>) -> ()22// CHECK: }23 24// -----25 26func.func private @mult_alloc(%size0 : index, %size1 : index) -> (memref<?x?xf32>, memref<?xf32>) {27 %alloc0 = memref.alloc(%size0, %size1) : memref<?x?xf32>28 %alloc1 = memref.alloc(%size1) : memref<?xf32>29 return %alloc0, %alloc1 : memref<?x?xf32>, memref<?xf32>30}31 32func.func @mult_alloc_test(%size0 : index, %size1: index) {33 %alloc0, %alloc1 = call @mult_alloc(%size0, %size1) : (index, index) -> (memref<?x?xf32>, memref<?xf32>)34 "test.sink"(%alloc0, %alloc1) : (memref<?x?xf32>, memref<?xf32>) -> ()35}36 37// CHECK-LABEL: func private @mult_alloc(38// CHECK-SAME: %{{.*}}: index, %{{.*}}: index,39// CHECK-SAME: %{{.*}}: memref<?x?xf32>, %{{.*}}: memref<?xf32>) {40 41// CHECK-LABEL: func @mult_alloc_test(42// CHECK-SAME: %[[size0:.*]]: index,43// CHECK-SAME: %[[size1:.*]]: index) {44// CHECK: %[[alloc0:.*]] = memref.alloc(%[[size0]], %[[size1]]) : memref<?x?xf32>45// CHECK: %[[alloc1:.*]] = memref.alloc(%[[size1]]) : memref<?xf32>46// CHECK: call @mult_alloc(%[[size0]], %[[size1]], %[[alloc0]], %[[alloc1]]) : (index, index, memref<?x?xf32>, memref<?xf32>) -> ()47// CHECK: "test.sink"(%[[alloc0]], %[[alloc1]]) : (memref<?x?xf32>, memref<?xf32>) -> ()48// CHECK: }49 50 51// -----52 53func.func private @complex_alloc(%size0 : index, %size1 : index) -> (memref<?x?xf32>, memref<4xf32>, memref<?xf32>) {54 %alloc0 = memref.alloc(%size0, %size1) : memref<?x?xf32>55 %alloc1 = memref.alloc() : memref<4xf32>56 %alloc2 = memref.alloc(%size1) : memref<?xf32>57 return %alloc0, %alloc1, %alloc2 : memref<?x?xf32>, memref<4xf32>, memref<?xf32>58}59 60func.func @complex_alloc_test(%size0 : index, %size1: index) {61 %alloc0, %alloc1, %alloc2 = call @complex_alloc(%size0, %size1) : (index, index) -> (memref<?x?xf32>, memref<4xf32>, memref<?xf32>)62 "test.sink"(%alloc0, %alloc1, %alloc2) : (memref<?x?xf32>, memref<4xf32>, memref<?xf32>) -> ()63}64 65// CHECK-LABEL: func private @complex_alloc(66// CHECK-SAME: %{{.*}}: index, %{{.*}}: index,67// CHECK-SAME: %{{.*}}: memref<?x?xf32>,68// CHECK-SAME: %{{.*}}: memref<4xf32>,69// CHECK-SAME: %{{.*}}: memref<?xf32>) {70 71// CHECK-LABEL: func @complex_alloc_test(72// CHECK-SAME: %[[size0:.*]]: index,73// CHECK-SAME: %[[size1:.*]]: index) {74// CHECK: %[[alloc0:.*]] = memref.alloc(%[[size0]], %[[size1]]) : memref<?x?xf32>75// CHECK: %[[alloc1:.*]] = memref.alloc() : memref<4xf32>76// CHECK: %[[alloc2:.*]] = memref.alloc(%[[size1]]) : memref<?xf32>77// CHECK: call @complex_alloc(%[[size0]], %[[size1]], %[[alloc0]], %[[alloc1]], %[[alloc2]]) : (index, index, memref<?x?xf32>, memref<4xf32>, memref<?xf32>) -> ()78// CHECK: "test.sink"(%[[alloc0]], %[[alloc1]], %[[alloc2]]) : (memref<?x?xf32>, memref<4xf32>, memref<?xf32>) -> ()79// CHECK: }80