1// RUN: mlir-opt %s -one-shot-bufferize="allow-unknown-ops" -split-input-file | FileCheck %s
2
3// Run fuzzer with different seeds.
4// RUN: mlir-opt %s -one-shot-bufferize="test-analysis-only analysis-fuzzer-seed=23" -split-input-file -o /dev/null
5// RUN: mlir-opt %s -one-shot-bufferize="test-analysis-only analysis-fuzzer-seed=59" -split-input-file -o /dev/null
6// RUN: mlir-opt %s -one-shot-bufferize="test-analysis-only analysis-fuzzer-seed=91" -split-input-file -o /dev/null
7
8// CHECK-LABEL: func @use_tensor_func_arg(
9//  CHECK-SAME:     %[[A:.*]]: tensor<?xf32>
10func @use_tensor_func_arg(%A : tensor<?xf32>) -> (vector<4xf32>) {
11  %c0 = arith.constant 0 : index
12  %f0 = arith.constant 0.0 : f32
13
14  // CHECK: %[[A_memref:.*]] = bufferization.to_memref %[[A]]
15  // CHECK: %[[res:.*]] = vector.transfer_read %[[A_memref]]
16  %0 = vector.transfer_read %A[%c0], %f0 : tensor<?xf32>, vector<4xf32>
17
18  // CHECK: return %[[res]]
19  return %0 : vector<4xf32>
20}
21
22// -----
23
24// CHECK-LABEL: func @return_tensor(
25//  CHECK-SAME:     %[[A:.*]]: tensor<?xf32>
26func @return_tensor(%A : tensor<?xf32>, %v : vector<4xf32>) -> (tensor<?xf32>) {
27  %c0 = arith.constant 0 : index
28
29  // CHECK: %[[A_memref:.*]] = bufferization.to_memref %[[A]]
30  // CHECK: %[[dim:.*]] = tensor.dim %[[A]]
31  // CHECK: %[[alloc:.*]] = memref.alloc(%[[dim]])
32  // CHECK: %[[casted:.*]] = memref.cast %[[alloc]]
33  // CHECK: memref.copy %[[A_memref]], %[[alloc]]
34  // CHECK: vector.transfer_write %{{.*}}, %[[alloc]]
35  // CHECK: %[[res_tensor:.*]] = bufferization.to_tensor %[[casted]]
36  %0 = vector.transfer_write %v, %A[%c0] : vector<4xf32>, tensor<?xf32>
37
38  // CHECK: memref.dealloc %[[alloc]]
39  // CHECK: return %[[res_tensor]]
40  return %0 : tensor<?xf32>
41}
42
43// -----
44
45// CHECK-LABEL: func @func_without_tensor_args
46func @func_without_tensor_args(%v : vector<10xf32>) -> () {
47  // CHECK: %[[alloc:.*]] = memref.alloc()
48  %0 = linalg.init_tensor[10] : tensor<10xf32>
49
50  %c0 = arith.constant 0 : index
51  // CHECK: vector.transfer_write %{{.*}}, %[[alloc]]
52  %1 = vector.transfer_write %v, %0[%c0] : vector<10xf32>, tensor<10xf32>
53
54  %cst = arith.constant 0.0 : f32
55  // CHECK: vector.transfer_read %[[alloc]]
56  %r = vector.transfer_read %1[%c0], %cst : tensor<10xf32>, vector<11xf32>
57
58  vector.print %r : vector<11xf32>
59  return
60}
61
62// -----
63
64// CHECK-LABEL: func private @private_func
65func private @private_func(tensor<?xf32>) -> ()
66
67// CHECK-LABEL: func @empty_func()
68func @empty_func() -> () {
69  return
70}
71
72// -----
73
74// CHECK-LABEL: func @read_after_write_conflict(
75func @read_after_write_conflict(%cst : f32, %idx : index, %idx2 : index)
76    -> (f32, f32) {
77  // CHECK-DAG: %[[alloc:.*]] = memref.alloc
78  // CHECK-DAG: %[[dummy:.*]] = "test.dummy_op"
79  // CHECK-DAG: %[[dummy_m:.*]] = bufferization.to_memref %[[dummy]]
80  %t = "test.dummy_op"() : () -> (tensor<10xf32>)
81
82  // CHECK: memref.copy %[[dummy_m]], %[[alloc]]
83  // CHECK: memref.store %{{.*}}, %[[alloc]]
84  %write = tensor.insert %cst into %t[%idx2] : tensor<10xf32>
85
86  // CHECK: %[[read:.*]] = "test.some_use"(%[[dummy]])
87  %read = "test.some_use"(%t) : (tensor<10xf32>) -> (f32)
88  // CHECK: %[[read2:.*]] = memref.load %[[alloc]]
89  %read2 = tensor.extract %write[%idx] : tensor<10xf32>
90
91  // CHECK: memref.dealloc %[[alloc]]
92  // CHECK: return %[[read]], %[[read2]]
93  return %read, %read2 : f32, f32
94}
95
96// -----
97
98// CHECK-LABEL: func @copy_deallocated(
99func @copy_deallocated() -> tensor<10xf32> {
100  // CHECK: %[[alloc:.*]] = memref.alloc()
101  %0 = linalg.init_tensor[10] : tensor<10xf32>
102  // CHECK: %[[alloc_tensor:.*]] = bufferization.to_tensor %[[alloc]]
103  // CHECK: memref.dealloc %[[alloc]]
104  // CHECK: return %[[alloc_tensor]]
105  return %0 : tensor<10xf32>
106}
107
108
109