23
23
// CHECK: }
24
24
// CHECK: %[[RESULT:.*]] = bufferization.to_tensor %[[RESULT_MEMREF]] : memref<4xf32>
25
25
// CHECK: return %[[RESULT]] : tensor<4xf32>
26
- func @basic (%arg0: tensor <4 xf32 >) -> tensor <4 xf32 > {
26
+ func.func @basic (%arg0: tensor <4 xf32 >) -> tensor <4 xf32 > {
27
27
%0 = linalg.generic {
28
28
indexing_maps = [#map0 , #map0 ],
29
29
iterator_types = [" parallel" ]
@@ -51,7 +51,7 @@ func @basic(%arg0: tensor<4xf32>) -> tensor<4xf32> {
51
51
// CHECK: linalg.generic
52
52
// CHECK-SAME: ins(%[[MEMREF]] : memref<?xf32>)
53
53
// CHECK-SAME: outs(%[[OUT_BUF]] : memref<?xf32>) {
54
- func @init_tensor (%in : tensor <?xf32 >, %size: index ) -> tensor <?xf32 > {
54
+ func.func @init_tensor (%in : tensor <?xf32 >, %size: index ) -> tensor <?xf32 > {
55
55
%init = linalg.init_tensor [%size ] : tensor <?xf32 >
56
56
%0 = linalg.generic {
57
57
indexing_maps = [#map0 , #map0 ],
@@ -77,7 +77,7 @@ func @init_tensor(%in : tensor<?xf32>, %size: index) -> tensor<?xf32> {
77
77
// CHECK-SAME: ins(%{{.*}} : memref<4xf32>)
78
78
// CHECK-SAME: outs(%[[RESULT0]], %[[RESULT1]] : memref<4xf32>, memref<4xf32>)
79
79
// CHECK-NEXT: ^bb0(%{{.*}}: f32, %{{.*}}: f32, %{{.*}}: f32):
80
- func @multiple_results (%arg0: tensor <4 xf32 >) -> (tensor <4 xf32 >, tensor <4 xf32 >) {
80
+ func.func @multiple_results (%arg0: tensor <4 xf32 >) -> (tensor <4 xf32 >, tensor <4 xf32 >) {
81
81
%0 , %1 = linalg.generic {
82
82
indexing_maps = [#map0 , #map0 , #map0 ],
83
83
iterator_types = [" parallel" ]
@@ -109,7 +109,7 @@ func @multiple_results(%arg0: tensor<4xf32>) -> (tensor<4xf32>, tensor<4xf32>) {
109
109
// CHECK: linalg.generic
110
110
// CHECK-SAME: ins(%[[MEMREF_ARG]] : memref<?x?xf32>)
111
111
// CHECK-SAME: outs(%[[RESULT0]], %[[RESULT1]] : memref<?x?xf32>, memref<?x?xf32>)
112
- func @dynamic_results (%arg0: tensor <?x?xf32 >)
112
+ func.func @dynamic_results (%arg0: tensor <?x?xf32 >)
113
113
-> (tensor <?x?xf32 >, tensor <?x?xf32 >) {
114
114
%0 , %1 = linalg.generic {
115
115
indexing_maps = [#map_2d , #map_2d , #map_2d ],
@@ -147,7 +147,7 @@ func @dynamic_results(%arg0: tensor<?x?xf32>)
147
147
// CHECK: linalg.generic
148
148
// CHECK-SAME: ins(%[[ARG0_MEMREF]] : memref<2x3x4xvector<3x4xi4>>)
149
149
// CHECK-SAME: outs(%[[INIT_BUFFER]] : memref<3x2xf32>) {
150
- func @generic_with_init_tensor (%arg0: tensor <2 x3 x4 xvector <3 x4 xi4 >>,
150
+ func.func @generic_with_init_tensor (%arg0: tensor <2 x3 x4 xvector <3 x4 xi4 >>,
151
151
%arg1: tensor <3 x2 xf32 >) -> (tensor <3 x2 xf32 >) {
152
152
153
153
%0 = linalg.generic #trait
@@ -164,7 +164,7 @@ func @generic_with_init_tensor(%arg0: tensor<2x3x4xvector<3x4xi4>>,
164
164
165
165
// CHECK-LABEL: func @bufferize_fill(
166
166
// CHECK-SAME: %[[IN:.*]]: tensor<?xf32>
167
- func @bufferize_fill (%arg0: tensor <?xf32 >) -> tensor <?xf32 > {
167
+ func.func @bufferize_fill (%arg0: tensor <?xf32 >) -> tensor <?xf32 > {
168
168
%c0 = arith.constant 0.0 : f32
169
169
// CHECK: %[[ALLOC:.*]] = memref.alloc
170
170
// CHECK: linalg.fill ins(%cst : f32) outs(%[[ALLOC]] : memref<?xf32>)
@@ -177,7 +177,7 @@ func @bufferize_fill(%arg0: tensor<?xf32>) -> tensor<?xf32> {
177
177
// -----
178
178
179
179
// CHECK-LABEL: func @bufferize_dot
180
- func @bufferize_dot (%in: tensor <4 xf32 >, %out: tensor <f32 >) -> tensor <f32 > {
180
+ func.func @bufferize_dot (%in: tensor <4 xf32 >, %out: tensor <f32 >) -> tensor <f32 > {
181
181
%dot = linalg.dot ins (%in , %in : tensor <4 xf32 >, tensor <4 xf32 >)
182
182
outs (%out : tensor <f32 >) -> tensor <f32 >
183
183
return %dot : tensor <f32 >
0 commit comments