[all-commits] [llvm/llvm-project] 0516f4: Add linalg.mmt4d named op

Ahmed S. Taei via All-commits all-commits at lists.llvm.org
Thu Jul 1 12:41:22 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0516f49c081590305a9db972ebc7fceb942b8ce3
      https://github.com/llvm/llvm-project/commit/0516f49c081590305a9db972ebc7fceb942b8ce3
  Author: Ahmed Taei <ataei at google.com>
  Date:   2021-07-01 (Thu, 01 Jul 2021)

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

  Log Message:
  -----------
  Add linalg.mmt4d named op

This op performs matrix-matrix-transpose multiplication of 4-d inputs as the following:

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

Reviewed By: Benoit

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




More information about the All-commits mailing list