Page MenuHomePhabricator

[mlir][linalg] Allow controlling folding unit dim reshapes

Authored by antiagainst on Mar 22 2021, 2:47 PM.



This commit exposes an option to the pattern
FoldWithProducerReshapeOpByExpansion to allow
folding unit dim reshapes. This gives callers
more fine-grained controls.

Depends On D99016

Diff Detail

Event Timeline

antiagainst created this revision.Mar 22 2021, 2:47 PM
antiagainst requested review of this revision.Mar 22 2021, 2:47 PM
This revision is now accepted and ready to land.Mar 22 2021, 10:40 PM
This revision was landed with ongoing or failed builds.Mar 24 2021, 3:19 PM
This revision was automatically updated to reflect the committed changes.

same discussion as in another CL, subview/subtensor/subtensor_insert has nicer inplace semantics than reshape