Lines Matching refs:transform
3 // RUN: mlir-opt --transform-interpreter -split-input-file --verify-diagnostics %s | FileCheck %s --check-prefix=BUILD-PACKING-LOOP-NEST
13 module attributes {transform.with_named_sequence} {
14 transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
15 %matmul = transform.structured.match ops{["linalg.matmul"]} in %arg1
16 : (!transform.any_op) -> !transform.any_op
18 %matmul_l1, %loops_l1 = transform.structured.tile_using_for %matmul tile_sizes [5] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
20 %matmul_padded, %0, %copy_back = transform.structured.pad %matmul_l1 {
23 } : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
27 %pad = transform.get_producer_of_operand %matmul_padded[1]
28 : (!transform.any_op) -> !transform.any_op
31 transform.structured.hoist_pad.build_packing_loop_nest %pad above %loops_l1
32 : (!transform.any_op, !transform.any_op) -> !transform.any_op
33 transform.yield
47 module attributes {transform.with_named_sequence} {
48 transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
49 %matmul = transform.structured.match ops{["linalg.matmul"]} in %arg1
50 : (!transform.any_op) -> !transform.any_op
52 %matmul_l1, %loops_l1 = transform.structured.tile_using_for %matmul tile_sizes [5] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
54 %matmul_padded, %0, %copy_back = transform.structured.pad %matmul_l1 {
57 } : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
59 %pad = transform.get_producer_of_operand %matmul_padded[2]
60 : (!transform.any_op) -> !transform.any_op
64 transform.structured.hoist_pad.build_packing_loop_nest %pad above %loops_l1
65 : (!transform.any_op, !transform.any_op) -> !transform.any_op
66 transform.yield
87 module attributes {transform.with_named_sequence} {
88 transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
89 %matmul = transform.structured.match ops{["linalg.matmul"]} in %arg1
90 : (!transform.any_op) -> !transform.any_op
92 %matmul_l1, %loops_l1 = transform.structured.tile_using_for %matmul tile_sizes [5] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
94 %matmul_padded, %0, %copy_back = transform.structured.pad %matmul_l1 {
97 } : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
99 %pad = transform.get_producer_of_operand %matmul_padded[0]
100 : (!transform.any_op) -> !transform.any_op
102 transform.structured.hoist_pad.build_packing_loop_nest %pad above %loops_l1
103 : (!transform.any_op, !transform.any_op) -> !transform.any_op
104 transform.yield
127 module attributes {transform.with_named_sequence} {
128 transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
129 %matmul = transform.structured.match ops{["linalg.matmul"]} in %arg1
130 : (!transform.any_op) -> !transform.any_op
132 %matmul_l1, %loops_l1 = transform.structured.tile_using_for %matmul tile_sizes [5] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
134 %matmul_padded, %0, %copy_back = transform.structured.pad %matmul_l1 {
137 } : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
139 %pad = transform.get_producer_of_operand %matmul_padded[0]
140 : (!transform.any_op) -> !transform.any_op
142 transform.structured.hoist_pad.build_packing_loop_nest %pad above %loops_l1, transpose by [1, 0]
143 : (!transform.any_op, !transform.any_op) -> !transform.any_op
144 transform.yield
165 module attributes {transform.with_named_sequence} {
166 transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
167 %matmul = transform.structured.match ops{["linalg.matmul"]} in %arg1
168 : (!transform.any_op) -> !transform.any_op
170 %matmul_l1, %loops_l1:2 = transform.structured.tile_using_for %matmul tile_sizes [5, 0, 7] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
172 %matmul_padded, %0, %copy_back = transform.structured.pad %matmul_l1 {
175 } : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
177 %pad = transform.get_producer_of_operand %matmul_padded[2]
178 : (!transform.any_op) -> !transform.any_op
180 transform.apply_licm to %loops_l1#1 : !transform.any_op
182 transform.structured.hoist_pad.build_packing_loop_nest %pad above %loops_l1#1
183 : (!transform.any_op, !transform.any_op) -> !transform.any_op
184 transform.yield