[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