[all-commits] [llvm/llvm-project] 2c4f56: Add linalg.batch_matvec named op

[Ahmed S. Taei via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2c4f5690ab5e435691aafe554725dbbd521b3754
      https://github.com/llvm/llvm-project/commit/2c4f5690ab5e435691aafe554725dbbd521b3754
  Author: Ahmed Taei <ataei at google.com>
  Date:   2021-06-30 (Wed, 30 Jun 2021)

  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/generalize-named-ops.mlir

  Log Message:
  -----------
  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