Page MenuHomePhabricator

[mlir][Vector] Add a pattern to lower 2-D vector.transpose to shape_cast+shuffle.
ClosedPublic

Authored by nicolasvasilache on Nov 2 2021, 3:02 PM.

Details

Summary

The 2-D case can be rewritten to generate quite fewer instructions and a single vector.shuffle which seems to provide a nice performance boost.
Add this arrow to our quiver by exposing it with a new vector transform option.

Diff Detail

Event Timeline

nicolasvasilache requested review of this revision.Nov 2 2021, 3:02 PM
Herald added a project: Restricted Project. · View Herald TranscriptNov 2 2021, 3:02 PM
ThomasRaoux accepted this revision.Nov 2 2021, 3:05 PM
This revision is now accepted and ready to land.Nov 2 2021, 3:05 PM
This revision was landed with ongoing or failed builds.Nov 2 2021, 3:12 PM
This revision was automatically updated to reflect the committed changes.