Add linalg.batch_matvec named op
authorAhmed Taei <ataei@google.com>
Tue, 22 Jun 2021 19:50:10 +0000 (12:50 -0700)
committerAhmed Taei <ataei@google.com>
Wed, 30 Jun 2021 18:37:21 +0000 (11:37 -0700)
commit2c4f5690ab5e435691aafe554725dbbd521b3754
tree56d37b9f3670ad5c01c1585cf8e27cc45684ae11
parent03051f7ac8a3e2eda44b8280290c90863a064f0e
Add linalg.batch_matvec named op

    Similarly to batch_mat vec outer most dim is a batching dim
    and this op does |b| matrix-vector-products :
    C[b, i] = sum_k(A[b, i, k] * B[b, k])

Reviewed By: rsuderman

Differential Revision: https://reviews.llvm.org/D104739
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
mlir/test/Dialect/Linalg/generalize-named-ops.mlir