[all-commits] [llvm/llvm-project] 1365ff: [mlir] allow repeated payload in structured.fuse_i...

[ftynse via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 1365ff74cb7d9f15feafdd4fbe2996d2f9e42a5e
      https://github.com/llvm/llvm-project/commit/1365ff74cb7d9f15feafdd4fbe2996d2f9e42a5e
  Author: Alex Zinenko <zinenko at google.com>
  Date:   2023-05-15 (Mon, 15 May 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
    M mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
    M mlir/test/Dialect/Linalg/transform-op-fuse-into-containing.mlir

  Log Message:
  -----------
  [mlir] allow repeated payload in structured.fuse_into_containing

Structured fusion proceeds by iteratively finding the next suitable
producer to be fused into the loop. Therefore, it shouldn't matter if
the same producer is listed multiple times (e.g., it is used as multiple
operands). Adjust the implementation of the transform op to support this
case.

Also fix the checking code in the interpreter to actually respect the
TransformOpInterface indication that repeated payload is allowed, it
seems to have been accidentally dropped in one of the refactorings.

Reviewed By: nicolasvasilache

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