HomePhabricator

[mlir][linalg] Fusion on tensors.

Authored by gysit on Sep 20 2021, 7:16 AM.

Description

[mlir][linalg] Fusion on tensors.

Add a new version of fusion on tensors that supports the following scenarios:

  • support input and output operand fusion
  • fuse a producer result passed in via tile loop iteration arguments (update the tile loop iteration arguments)
  • supports only linalg operations on tensors
  • supports only scf::for
  • cannot add an output to the tile loop nest

The LinalgTileAndFuseOnTensors pass tiles the root operation and fuses its producers.

Reviewed By: nicolasvasilache, mravishankar

Differential Revision: https://reviews.llvm.org/D109766

Details

Committed
gysitSep 20 2021, 7:45 AM
Reviewer
nicolasvasilache
Differential Revision
D109766: [mlir][linalg] Fusion on tensors.
Parents
rG5dee50111c13: [analyzer] Move docs of SmartPtr to correct subcategory
Branches
Unknown
Tags
Unknown