Google Git
Sign in
llvm / llvm-project / mlir / ea69e408960f5351f5b32c06870302c489ad9f2d
commitea69e408960f5351f5b32c06870302c489ad9f2d[log] [tgz]
authordonald chen <62002319+cxy-1993@users.noreply.github.com>Mon Apr 22 13:53:59 2024 +0800
committerCopybara-Service <copybara-worker@google.com>Sun Apr 21 23:00:10 2024 -0700
tree286530cf942e58d30e4bbe9f8243aaa7df5500ff
parent7579f1d85115fe7e505c643cd8b4ac5a88c672b3 [diff]
[MLIR][Linalg] Enable fuse consumer (#85528)

This patch adds support for consumer fusion to the tiling interface, and
implements fuse consumers on FuseIntoContainingOp.

- Add interface method 'getIterDomainTilePositionFromOperandPosition' to
tiling interface which get iteration domain position from operand
position.
- Add interface method 'getTiledImplementationFromOperandPosition' to
tiling interface which generate tiled implementation according to
operand position.
- Implemented the above two methods and supported consumer fusion for
FuseIntoContainingOp.

Signed-off-by: Donald Chen
GitOrigin-RevId: 2a47ee070145438424b065a35c4a680ea0cb0c1f
4 files changed
tree: 286530cf942e58d30e4bbe9f8243aaa7df5500ff
  1. benchmark/
  2. cmake/
  3. docs/
  4. examples/
  5. include/
  6. lib/
  7. python/
  8. test/
  9. tools/
  10. unittests/
  11. utils/
  12. .clang-format
  13. .clang-tidy
  14. CMakeLists.txt
  15. LICENSE.TXT
  16. README.md
README.md

Multi-Level Intermediate Representation

See https://mlir.llvm.org/ for more information.