Skip to content

Commit 54d6356

Browse files
committed
[Mlir] Skip default schedule for linalg.fill
Do not schedule the initialization operation linalg.fill, which is not controllable by the user. The default schedule applied can overconstraint lowering passes trying to fuse/optimize it.
1 parent 0853138 commit 54d6356

23 files changed

Lines changed: 1201 additions & 1892 deletions

src/xtc/backends/mlir/MlirCompilerPasses.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,9 @@ def _generate_scheduling(self) -> OpResult:
189189
assert self._named_sequence is not None
190190
handle = None
191191
for schedule in self._nodes_schedules:
192+
# Skip linalg.fill
193+
if schedule.node_name[-1] == "0": # identify with naming convention
194+
continue
192195
self._create_sdist_meshes(schedule)
193196
handle = structured_match(
194197
results_=transform.AnyOpType.get(),

tests/filecheck/backends/padding/test_gen_pad_dict_conv2d_mlir.py

Lines changed: 87 additions & 153 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_gen_pad_int_matmul_unpad_mlir.py

Lines changed: 80 additions & 131 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_gen_pad_tuple_conv2d_mlir.py

Lines changed: 86 additions & 152 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_gen_pad_tuple_matmul_unpad_mlir.py

Lines changed: 80 additions & 131 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_pad_constant_conv2d_mlir.py

Lines changed: 83 additions & 149 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_pad_conv2d_mlir.py

Lines changed: 83 additions & 149 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_pad_matmul_unpad_mlir.py

Lines changed: 82 additions & 133 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/padding/test_pad_tuple_matmul_unpad_mlir.py

Lines changed: 82 additions & 133 deletions
Large diffs are not rendered by default.

tests/filecheck/backends/test_conv2d_mini_mlir.py

Lines changed: 56 additions & 89 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)