Page MenuHomePhabricator

[mlir][linalg] update fusion to support linalg index operations.
ClosedPublic

Authored by gysit on Apr 14 2021, 1:41 AM.

Details

Summary

The patch updates the linalg fusion pass to add the tile offsets to the indices.

Diff Detail

Event Timeline

gysit created this revision.Apr 14 2021, 1:41 AM
gysit requested review of this revision.Apr 14 2021, 1:41 AM
gysit added inline comments.
mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
235

Removing make_early_inc_range here since we are not erasing the index operation meaning the iterator should remain valid without early increment.

pifon2a accepted this revision.Apr 14 2021, 4:20 AM

Nice!

This revision is now accepted and ready to land.Apr 14 2021, 4:20 AM