Skip to content

Commit 3d0f125

Browse files
authored
update transform tests (#91)
1 parent ba622c3 commit 3d0f125

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_transform.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def basic(target):
156156
module attributes {transform.with_named_sequence} {
157157
transform.named_sequence @__transform_main(%arg0: !transform.any_op) {
158158
%0 = transform.structured.match ops{["tensor.pad"]} in %arg0 : (!transform.any_op) -> !transform.any_op
159-
%tiled_linalg_op, %loops:2 = transform.structured.tile_using_for %0[2, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
159+
%tiled_linalg_op, %loops:2 = transform.structured.tile_using_for %0 tile_sizes [2, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
160160
transform.yield
161161
}
162162
}
@@ -267,7 +267,7 @@ def basic(target):
267267
module attributes {transform.with_named_sequence} {
268268
transform.named_sequence @__transform_main(%arg0: !transform.any_op) {
269269
%0 = transform.structured.match ops{["linalg.matmul"]} in %arg0 : (!transform.any_op) -> !transform.any_op
270-
%tiled_linalg_op, %loops:2 = transform.structured.tile_using_for %0[2, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
270+
%tiled_linalg_op, %loops:2 = transform.structured.tile_using_for %0 tile_sizes [2, 3] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
271271
transform.yield
272272
}
273273
}
@@ -859,7 +859,7 @@ def main(variant_op: any_op_t()):
859859
%allocated_buffer_19, %new_ops_20 = transform.structured.bufferize_to_allocation %pack_op_12 {bufferize_destination_only, memory_space = "local"} : !transform.any_op
860860
%11 = transform.get_producer_of_operand %packed_op_14[2] : (!transform.any_op) -> !transform.any_op
861861
%allocated_buffer_21, %new_ops_22 = transform.structured.bufferize_to_allocation %11 {bufferize_destination_only, memory_space = "local"} : !transform.any_op
862-
%tiled_linalg_op, %loops = transform.structured.tile_using_for %packed_op_14[0, 0, 1] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
862+
%tiled_linalg_op, %loops = transform.structured.tile_using_for %packed_op_14 tile_sizes [0, 0, 1] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
863863
transform.include @cleanup failures(propagate) (%arg0) : (!transform.any_op) -> ()
864864
transform.yield
865865
}

0 commit comments

Comments
 (0)