tensorflow / mlir

"Multi-Level Intermediate Representation" Compiler Infrastructure
1.73k stars 257 forks source link

[Linalg] Add a Linalg iterator permutation transformation. #272

Closed nicolasvasilache closed 4 years ago

nicolasvasilache commented 4 years ago

Once #270 is landed, we can add a standalone iterator permutation transformation to Linalg.

This transformation should be very simple to write and should just compose a permutation map the with the maps in the "indexing_maps" attribute.

There is a C++ component and a Tablegen component (to make the transformation into a pattern) to this PR.

tetuante commented 4 years ago

Working on it!

nicolasvasilache commented 4 years ago

Done.