Skip to content

Commit f52b638

Browse files
ftynseftynse
committed
[mlir] update types in remaining Linalg TransformOps test
All ops now support explicit type specification, update types to use `!transform.any_op` instead of `!pdl.operation` for consistency. Depends On D144515 Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D150592
1 parent 2fe4d90 commit f52b638

17 files changed

+319
-319
lines changed

‎mlir/test/Dialect/Linalg/convert-conv2d-to-img2col.mlir

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@ func.func @conv_non_static(%arg0: tensor<?x?x?x?xf32>, %arg1: tensor<3x3x4x16xf3
1616
}
1717

1818
transform.sequence failures(propagate) {
19-
^bb1(%arg1: !pdl.operation):
20-
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!pdl.operation) -> !pdl.operation
19+
^bb1(%arg1: !transform.any_op):
20+
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!transform.any_op) -> !transform.any_op
2121
// expected-error@below {{failed to apply}}
22-
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
22+
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
2323
}
2424

2525
// -----
@@ -88,11 +88,11 @@ func.func @conv_16433136(%arg0: tensor<1x16x16x4xf32>, %arg1: tensor<3x3x4x16xf3
8888
}
8989

9090
transform.sequence failures(propagate) {
91-
^bb1(%arg1: !pdl.operation):
92-
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!pdl.operation) -> !pdl.operation
93-
%img2col_tensor_producer, %transformed = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
94-
transform.print %img2col_tensor_producer {name = "tensor_producer"}: !pdl.operation
95-
transform.print %transformed {name = "transformed"}: !pdl.operation
91+
^bb1(%arg1: !transform.any_op):
92+
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!transform.any_op) -> !transform.any_op
93+
%img2col_tensor_producer, %transformed = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
94+
transform.print %img2col_tensor_producer {name = "tensor_producer"}: !transform.any_op
95+
transform.print %transformed {name = "transformed"}: !transform.any_op
9696
}
9797

9898
// -----
@@ -167,9 +167,9 @@ func.func @depthwise_conv_hwc_114x16x3(%input: tensor<1x114x114x16xf32>, %filter
167167
}
168168

169169
transform.sequence failures(propagate) {
170-
^bb1(%arg1: !pdl.operation):
171-
%0 = transform.structured.match ops{["linalg.depthwise_conv_2d_nhwc_hwc"]} in %arg1 : (!pdl.operation) -> !pdl.operation
172-
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
170+
^bb1(%arg1: !transform.any_op):
171+
%0 = transform.structured.match ops{["linalg.depthwise_conv_2d_nhwc_hwc"]} in %arg1 : (!transform.any_op) -> !transform.any_op
172+
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
173173
}
174174

175175
// -----
@@ -209,9 +209,9 @@ func.func @batch_nhwc_conv(%arg0: tensor<8x16x16x4xf32>, %arg1: tensor<3x3x4x16x
209209
}
210210

211211
transform.sequence failures(propagate) {
212-
^bb1(%arg1: !pdl.operation):
213-
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!pdl.operation) -> !pdl.operation
214-
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
212+
^bb1(%arg1: !transform.any_op):
213+
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!transform.any_op) -> !transform.any_op
214+
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
215215
}
216216

217217
// -----
@@ -272,9 +272,9 @@ func.func @batch_nchw_conv(%arg0: tensor<8x4x16x16xf32>, %arg1: tensor<16x4x3x3x
272272
}
273273

274274
transform.sequence failures(propagate) {
275-
^bb1(%arg1: !pdl.operation):
276-
%0 = transform.structured.match ops{["linalg.conv_2d_nchw_fchw"]} in %arg1 : (!pdl.operation) -> !pdl.operation
277-
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
275+
^bb1(%arg1: !transform.any_op):
276+
%0 = transform.structured.match ops{["linalg.conv_2d_nchw_fchw"]} in %arg1 : (!transform.any_op) -> !transform.any_op
277+
%1:2 = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
278278
}
279279

280280
// -----
@@ -308,9 +308,9 @@ func.func @conv_integer_extend(%arg0: tensor<1x16x16x4xi8>, %arg1: tensor<3x3x4x
308308
}
309309

310310
transform.sequence failures(propagate) {
311-
^bb1(%arg1: !pdl.operation):
312-
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!pdl.operation) -> !pdl.operation
313-
%img2col_tensor_producer, %transformed = transform.structured.convert_conv2d_to_img2col %0 : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
314-
transform.print %img2col_tensor_producer {name = "tensor_producer"}: !pdl.operation
315-
transform.print %transformed {name = "transformed"}: !pdl.operation
311+
^bb1(%arg1: !transform.any_op):
312+
%0 = transform.structured.match ops{["linalg.conv_2d_nhwc_hwcf"]} in %arg1 : (!transform.any_op) -> !transform.any_op
313+
%img2col_tensor_producer, %transformed = transform.structured.convert_conv2d_to_img2col %0 : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
314+
transform.print %img2col_tensor_producer {name = "tensor_producer"}: !transform.any_op
315+
transform.print %transformed {name = "transformed"}: !transform.any_op
316316
}

‎mlir/test/Dialect/Linalg/hoisting.mlir

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -75,11 +75,11 @@ func.func @hoist_vector_transfer_pairs(
7575
}
7676

7777
transform.sequence failures(propagate) {
78-
^bb1(%arg1: !pdl.operation):
78+
^bb1(%arg1: !transform.any_op):
7979
%0 = transform.structured.match ops{["func.func"]} in %arg1
80-
: (!pdl.operation) -> !pdl.operation
80+
: (!transform.any_op) -> !transform.any_op
8181
transform.structured.hoist_redundant_vector_transfers %0
82-
: (!pdl.operation) -> !pdl.operation
82+
: (!transform.any_op) -> !transform.any_op
8383
}
8484

8585
// -----
@@ -164,11 +164,11 @@ func.func @hoist_vector_transfer_pairs_disjoint(
164164
}
165165

166166
transform.sequence failures(propagate) {
167-
^bb1(%arg1: !pdl.operation):
167+
^bb1(%arg1: !transform.any_op):
168168
%0 = transform.structured.match ops{["func.func"]} in %arg1
169-
: (!pdl.operation) -> !pdl.operation
169+
: (!transform.any_op) -> !transform.any_op
170170
transform.structured.hoist_redundant_vector_transfers %0
171-
: (!pdl.operation) -> !pdl.operation
171+
: (!transform.any_op) -> !transform.any_op
172172
}
173173

174174
// -----
@@ -209,11 +209,11 @@ func.func @hoist_vector_transfer_pairs_in_affine_loops(%memref0: memref<64x64xi3
209209
}
210210

211211
transform.sequence failures(propagate) {
212-
^bb1(%arg1: !pdl.operation):
212+
^bb1(%arg1: !transform.any_op):
213213
%0 = transform.structured.match ops{["func.func"]} in %arg1
214-
: (!pdl.operation) -> !pdl.operation
214+
: (!transform.any_op) -> !transform.any_op
215215
transform.structured.hoist_redundant_vector_transfers %0
216-
: (!pdl.operation) -> !pdl.operation
216+
: (!transform.any_op) -> !transform.any_op
217217
}
218218

219219
// -----
@@ -298,11 +298,11 @@ func.func @hoist_vector_transfer_pairs_tensor(
298298
}
299299

300300
transform.sequence failures(propagate) {
301-
^bb1(%arg1: !pdl.operation):
301+
^bb1(%arg1: !transform.any_op):
302302
%0 = transform.structured.match ops{["func.func"]} in %arg1
303-
: (!pdl.operation) -> !pdl.operation
303+
: (!transform.any_op) -> !transform.any_op
304304
transform.structured.hoist_redundant_tensor_subsets %0
305-
: (!pdl.operation) -> ()
305+
: (!transform.any_op) -> ()
306306
}
307307

308308
// -----
@@ -393,11 +393,11 @@ func.func @hoist_vector_transfer_pairs_disjoint_tensor(
393393
}
394394

395395
transform.sequence failures(propagate) {
396-
^bb1(%arg1: !pdl.operation):
396+
^bb1(%arg1: !transform.any_op):
397397
%0 = transform.structured.match ops{["func.func"]} in %arg1
398-
: (!pdl.operation) -> !pdl.operation
398+
: (!transform.any_op) -> !transform.any_op
399399
transform.structured.hoist_redundant_tensor_subsets %0
400-
: (!pdl.operation) -> ()
400+
: (!transform.any_op) -> ()
401401
}
402402

403403
// -----
@@ -510,11 +510,11 @@ func.func @hoist_vector_transfer_pairs_tensor_and_slices(
510510
}
511511

512512
transform.sequence failures(propagate) {
513-
^bb1(%arg1: !pdl.operation):
513+
^bb1(%arg1: !transform.any_op):
514514
%0 = transform.structured.match ops{["func.func"]} in %arg1
515-
: (!pdl.operation) -> !pdl.operation
515+
: (!transform.any_op) -> !transform.any_op
516516
transform.structured.hoist_redundant_tensor_subsets %0
517-
: (!pdl.operation) -> ()
517+
: (!transform.any_op) -> ()
518518
}
519519

520520
// -----
@@ -557,11 +557,11 @@ func.func @hoist_vector_transfer_write_pairs_disjoint_tensor(
557557
}
558558

559559
transform.sequence failures(propagate) {
560-
^bb1(%arg1: !pdl.operation):
560+
^bb1(%arg1: !transform.any_op):
561561
%0 = transform.structured.match ops{["func.func"]} in %arg1
562-
: (!pdl.operation) -> !pdl.operation
562+
: (!transform.any_op) -> !transform.any_op
563563
transform.structured.hoist_redundant_tensor_subsets %0
564-
: (!pdl.operation) -> ()
564+
: (!transform.any_op) -> ()
565565
}
566566

567567
// -----
@@ -670,11 +670,11 @@ func.func @hoist_vector_transfer_pairs_tensor_and_slices_static_large_tensor(
670670
}
671671

672672
transform.sequence failures(propagate) {
673-
^bb1(%arg1: !pdl.operation):
673+
^bb1(%arg1: !transform.any_op):
674674
%0 = transform.structured.match ops{["func.func"]} in %arg1
675-
: (!pdl.operation) -> !pdl.operation
675+
: (!transform.any_op) -> !transform.any_op
676676
transform.structured.hoist_redundant_tensor_subsets %0
677-
: (!pdl.operation) -> ()
677+
: (!transform.any_op) -> ()
678678
}
679679

680680
// -----
@@ -716,9 +716,9 @@ func.func @hoist_vector_transfer_read() {
716716
}
717717

718718
transform.sequence failures(propagate) {
719-
^bb1(%arg1: !pdl.operation):
719+
^bb1(%arg1: !transform.any_op):
720720
%0 = transform.structured.match ops{["func.func"]} in %arg1
721-
: (!pdl.operation) -> !pdl.operation
721+
: (!transform.any_op) -> !transform.any_op
722722
transform.structured.hoist_redundant_vector_transfers %0
723-
: (!pdl.operation) -> !pdl.operation
723+
: (!transform.any_op) -> !transform.any_op
724724
}

‎mlir/test/Dialect/Linalg/multisize-tiling-full.mlir

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,18 @@
33

44
// This implements a 2D multisize tiling with target sizes [3, 10].
55
transform.sequence failures(propagate) {
6-
^bb1(%arg1: !pdl.operation):
7-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
8-
%1:3 = transform.structured.multitile_sizes %0 { dimension = 0, target_size = 3} : (!pdl.operation) -> !pdl.operation
9-
%t:3 = transform.structured.multitile_sizes %0 { dimension = 1, target_size = 10} : (!pdl.operation) -> !pdl.operation
10-
%2:2 = transform.structured.split %0 after %1#2 { dimension = 0 } : !pdl.operation, !pdl.operation
11-
%3:2 = transform.structured.tile %2#0 [%1#0] : (!pdl.operation, !pdl.operation) -> (!pdl.operation, !pdl.operation)
12-
%4:2 = transform.structured.tile %2#1 [%1#1] : (!pdl.operation, !pdl.operation) -> (!pdl.operation, !pdl.operation)
13-
%5 = merge_handles %3#0, %4#0 : !pdl.operation
14-
%tt:3 = replicate num(%5) %t#0, %t#1, %t#2 : !pdl.operation, !pdl.operation, !pdl.operation, !pdl.operation
15-
%6:2 = transform.structured.split %5 after %tt#2 { dimension = 1 } : !pdl.operation, !pdl.operation
16-
transform.structured.tile %6#0 [0, %tt#0] : (!pdl.operation, !pdl.operation) -> (!pdl.operation, !pdl.operation)
17-
transform.structured.tile %6#1 [0, %tt#1] : (!pdl.operation, !pdl.operation) -> (!pdl.operation, !pdl.operation)
6+
^bb1(%arg1: !transform.any_op):
7+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
8+
%1:3 = transform.structured.multitile_sizes %0 { dimension = 0, target_size = 3} : (!transform.any_op) -> !transform.any_op
9+
%t:3 = transform.structured.multitile_sizes %0 { dimension = 1, target_size = 10} : (!transform.any_op) -> !transform.any_op
10+
%2:2 = transform.structured.split %0 after %1#2 { dimension = 0 } : !transform.any_op, !transform.any_op
11+
%3:2 = transform.structured.tile %2#0 [%1#0] : (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
12+
%4:2 = transform.structured.tile %2#1 [%1#1] : (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
13+
%5 = merge_handles %3#0, %4#0 : !transform.any_op
14+
%tt:3 = replicate num(%5) %t#0, %t#1, %t#2 : !transform.any_op, !transform.any_op, !transform.any_op, !transform.any_op
15+
%6:2 = transform.structured.split %5 after %tt#2 { dimension = 1 } : !transform.any_op, !transform.any_op
16+
transform.structured.tile %6#0 [0, %tt#0] : (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
17+
transform.structured.tile %6#1 [0, %tt#1] : (!transform.any_op, !transform.any_op) -> (!transform.any_op, !transform.any_op)
1818
}
1919

2020
func.func private @elem(%arg0: f32, %arg1: index, %arg2: index) -> f32
@@ -103,18 +103,18 @@ func.func @two_d(%arg0: tensor<10x34xf32>,
103103
// -----
104104

105105
transform.sequence failures(propagate) {
106-
^bb1(%arg1: !pdl.operation):
107-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
108-
%1:3 = transform.structured.multitile_sizes %0 { dimension = 0, target_size = 3} : (!pdl.operation) -> !transform.param<i64>
109-
%t:3 = transform.structured.multitile_sizes %0 { dimension = 1, target_size = 10} : (!pdl.operation) -> !transform.param<i64>
110-
%2:2 = transform.structured.split %0 after %1#2 { dimension = 0 } : !pdl.operation, !transform.param<i64>
111-
%3:2 = transform.structured.tile %2#0 [%1#0] : (!pdl.operation, !transform.param<i64>) -> (!pdl.operation, !pdl.operation)
112-
%4:2 = transform.structured.tile %2#1 [%1#1] : (!pdl.operation, !transform.param<i64>) -> (!pdl.operation, !pdl.operation)
113-
%5 = merge_handles %3#0, %4#0 : !pdl.operation
114-
%tt:3 = replicate num(%5) %t#0, %t#1, %t#2 : !pdl.operation, !transform.param<i64>, !transform.param<i64>, !transform.param<i64>
115-
%6:2 = transform.structured.split %5 after %tt#2 { dimension = 1 } : !pdl.operation, !transform.param<i64>
116-
transform.structured.tile %6#0 [0, %tt#0] : (!pdl.operation, !transform.param<i64>) -> (!pdl.operation, !pdl.operation)
117-
transform.structured.tile %6#1 [0, %tt#1] : (!pdl.operation, !transform.param<i64>) -> (!pdl.operation, !pdl.operation)
106+
^bb1(%arg1: !transform.any_op):
107+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
108+
%1:3 = transform.structured.multitile_sizes %0 { dimension = 0, target_size = 3} : (!transform.any_op) -> !transform.param<i64>
109+
%t:3 = transform.structured.multitile_sizes %0 { dimension = 1, target_size = 10} : (!transform.any_op) -> !transform.param<i64>
110+
%2:2 = transform.structured.split %0 after %1#2 { dimension = 0 } : !transform.any_op, !transform.param<i64>
111+
%3:2 = transform.structured.tile %2#0 [%1#0] : (!transform.any_op, !transform.param<i64>) -> (!transform.any_op, !transform.any_op)
112+
%4:2 = transform.structured.tile %2#1 [%1#1] : (!transform.any_op, !transform.param<i64>) -> (!transform.any_op, !transform.any_op)
113+
%5 = merge_handles %3#0, %4#0 : !transform.any_op
114+
%tt:3 = replicate num(%5) %t#0, %t#1, %t#2 : !transform.any_op, !transform.param<i64>, !transform.param<i64>, !transform.param<i64>
115+
%6:2 = transform.structured.split %5 after %tt#2 { dimension = 1 } : !transform.any_op, !transform.param<i64>
116+
transform.structured.tile %6#0 [0, %tt#0] : (!transform.any_op, !transform.param<i64>) -> (!transform.any_op, !transform.any_op)
117+
transform.structured.tile %6#1 [0, %tt#1] : (!transform.any_op, !transform.param<i64>) -> (!transform.any_op, !transform.any_op)
118118
}
119119

120120
func.func private @elem(%arg0: f32, %arg1: index, %arg2: index) -> f32

‎mlir/test/Dialect/Linalg/tile-conv.mlir

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@ func.func @conv(%arg0 : memref<?x?xf32>, %arg1 : memref<?x?xf32>, %arg2 : memref
1010
}
1111

1212
transform.sequence failures(propagate) {
13-
^bb0(%arg1: !pdl.operation):
14-
%0 = transform.structured.match ops{["linalg.conv_2d"]} in %arg1 : (!pdl.operation) -> !pdl.operation
15-
%1, %loop:2 = transform.structured.tile %0 [2, 3] : (!pdl.operation) -> (!pdl.operation, !pdl.operation, !pdl.operation)
13+
^bb0(%arg1: !transform.any_op):
14+
%0 = transform.structured.match ops{["linalg.conv_2d"]} in %arg1 : (!transform.any_op) -> !transform.any_op
15+
%1, %loop:2 = transform.structured.tile %0 [2, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
1616
}
1717

1818
// CHECK: func @conv

‎mlir/test/Dialect/Linalg/tile-indexed.mlir

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ func.func @indexed_vector(%arg0: memref<50xindex>) {
1212
}
1313

1414
transform.sequence failures(propagate) {
15-
^bb0(%arg1: !pdl.operation):
16-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
17-
%1, %loop = transform.structured.tile %0 [10] : (!pdl.operation) -> (!pdl.operation, !pdl.operation)
15+
^bb0(%arg1: !transform.any_op):
16+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
17+
%1, %loop = transform.structured.tile %0 [10] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
1818
}
1919

2020
// TILE-10n25-DAG: [[$MAP:#[a-zA-Z0-9_]*]] = affine_map<(d0, d1) -> (d0 + d1)>
@@ -42,9 +42,9 @@ func.func @indexed_matrix(%arg0: memref<50x50xindex>) {
4242
}
4343

4444
transform.sequence failures(propagate) {
45-
^bb0(%arg1: !pdl.operation):
46-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
47-
%1, %loop:2 = transform.structured.tile %0 [10, 25] : (!pdl.operation) -> (!pdl.operation, !pdl.operation, !pdl.operation)
45+
^bb0(%arg1: !transform.any_op):
46+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
47+
%1, %loop:2 = transform.structured.tile %0 [10, 25] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
4848
}
4949

5050
// TILE-10n25-DAG: [[$MAP:#[a-zA-Z0-9_]*]] = affine_map<(d0, d1) -> (d0 + d1)>

‎mlir/test/Dialect/Linalg/tile-tensors.mlir

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ func.func @matmul_tensors(
2828
}
2929

3030
transform.sequence failures(propagate) {
31-
^bb0(%arg1: !pdl.operation):
32-
%0 = transform.structured.match ops{["linalg.matmul"]} in %arg1 : (!pdl.operation) -> !pdl.operation
33-
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!pdl.operation) -> (!pdl.operation, !pdl.operation, !pdl.operation, !pdl.operation)
31+
^bb0(%arg1: !transform.any_op):
32+
%0 = transform.structured.match ops{["linalg.matmul"]} in %arg1 : (!transform.any_op) -> !transform.any_op
33+
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op, !transform.any_op)
3434
}
3535

3636
// -----
@@ -59,9 +59,9 @@ func.func @generic_op_tensors(
5959
}
6060

6161
transform.sequence failures(propagate) {
62-
^bb0(%arg1: !pdl.operation):
63-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
64-
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!pdl.operation) -> (!pdl.operation, !pdl.operation, !pdl.operation, !pdl.operation)
62+
^bb0(%arg1: !transform.any_op):
63+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
64+
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op, !transform.any_op)
6565
}
6666

6767
// CHECK-LABEL: func @generic_op_tensors
@@ -130,7 +130,7 @@ func.func @fold_extract_slice(
130130
}
131131

132132
transform.sequence failures(propagate) {
133-
^bb0(%arg1: !pdl.operation):
134-
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!pdl.operation) -> !pdl.operation
135-
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!pdl.operation) -> (!pdl.operation, !pdl.operation, !pdl.operation, !pdl.operation)
133+
^bb0(%arg1: !transform.any_op):
134+
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
135+
%1, %loops:3 = transform.structured.tile %0 [2, 3, 4] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op, !transform.any_op)
136136
}

0 commit comments

Comments
 (0)