Page MenuHomePhabricator

[mlir][Linalg] Generalize the logic to compute reassociation mapswhile folding tensor_reshape op.
ClosedPublic

Authored by mravishankar on Sep 29 2020, 4:16 PM.

Details

Summary

While folding reshapes that introduce unit extent dims, the logic to
compute the reassociation maps can be generalized to handle some
corner cases, for example, when the folded shape still has unit-extent
dims but corresponds to folded unit extent dims of the expanded shape.

Diff Detail

Event Timeline

mravishankar created this revision.Sep 29 2020, 4:16 PM
Herald added a project: Restricted Project. · View Herald TranscriptSep 29 2020, 4:16 PM
mravishankar requested review of this revision.Sep 29 2020, 4:16 PM
mravishankar retitled this revision from [mlir][Linalg] Generalize the logic to compute reassociation maps while folding tensor_reshape op. to [mlir][Linalg] Generalize the logic to compute reassociation mapswhile folding tensor_reshape op..Sep 29 2020, 4:17 PM
mravishankar added reviewers: hanchung, asaadaldien.
asaadaldien accepted this revision.Sep 30 2020, 7:54 AM
asaadaldien added a subscriber: maheshkhanwalkar.

LGTM! Thanks for fixing this @maheshkhanwalkar.

This revision is now accepted and ready to land.Sep 30 2020, 7:54 AM