[all-commits] [llvm/llvm-project] 7f4026: [mlir] Add linalg.batch_mmt4d named op

[Jerry Wu via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 7f4026e55ae9063794ebfcefb9730c9099fbc967
      https://github.com/llvm/llvm-project/commit/7f4026e55ae9063794ebfcefb9730c9099fbc967
  Author: Jerry Wu <cheyuw at google.com>
  Date:   2023-08-03 (Thu, 03 Aug 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
    M mlir/test/Dialect/Linalg/named-ops.mlir

  Log Message:
  -----------
  [mlir] Add linalg.batch_mmt4d named op

This op is the batched version of linalg.mmt4d. It performs matrix-matrix-transpose multiplication of batched 4-d (5d) inputs as the following:

```
C[b, m1, n1, m0, n0] = sum_{b, k1, k0}(A[b, m1, k1, m0, k0] * B[b, n1, k1, n0, k0])
```

The current use is to provide `linalg.batch_matmul` a lowering path similar to `linalg.matmul -> linalg.mmt4d`.

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