Page MenuHomePhabricator

[mlir][linalg] lower index operations during linalg to loop lowering.

Authored by gysit on Apr 12 2021, 11:46 PM.



The patch extends the linalg to loop lowering pass to replace all linalg index operations by the induction variables of the generated loop nests.

Diff Detail

Event Timeline

gysit created this revision.Apr 12 2021, 11:46 PM
gysit requested review of this revision.Apr 12 2021, 11:46 PM

The new test cases in loops.mlir duplicate the existing tests for the indexed_generic operation (below). The code after lowering is the same the only difference is the input code (linalg.generic + linalg.index instead of linalg.indexed_generic).

Nice, thanks!

This revision is now accepted and ready to land.Apr 12 2021, 11:57 PM