[all-commits] [llvm/llvm-project] cd9bac: [mlir] generalize matchers to support batch matmul

[ftynse via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: cd9bacdf7fda6fcb1f06e96f39af7f537a2542ad
      https://github.com/llvm/llvm-project/commit/cd9bacdf7fda6fcb1f06e96f39af7f537a2542ad
  Author: Alex Zinenko <zinenko at google.com>
  Date:   2023-07-07 (Fri, 07 Jul 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgInterfaces.h
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgMatchOps.td
    M mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgMatchOps.cpp
    M mlir/test/Dialect/Linalg/match-ops-interpreter.mlir
    M mlir/test/Dialect/Linalg/match-ops-invalid.mlir
    A mlir/test/Integration/Dialect/Transform/match_batch_matmul.mlir
    M mlir/test/Integration/Dialect/Transform/match_matmul.mlir
    A mlir/test/Integration/Dialect/Transform/match_matmul_common.mlir

  Log Message:
  -----------
  [mlir] generalize matchers to support batch matmul

Mostly the same logic applies, with a different rank.

Additionally expose the logic to do identify contraction dimensions and
contraction-like bodies as independent transform ops. This allows us to
recognize "generic" operations and not only the named ones.

Rework the contraction body matching logic to no longer rely on
contraction operations beign uniquely named.

Reviewed By: springerm

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