[all-commits] [llvm/llvm-project] f38176: [MLIR][Linalg] introduce batch-reduce GEMM

[lorenzo chelini via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f381768a8da6bd6bde8bdff34f080bf12bf20064
      https://github.com/llvm/llvm-project/commit/f381768a8da6bd6bde8bdff34f080bf12bf20064
  Author: lorenzo chelini <l.chelini at icloud.com>
  Date:   2022-09-19 (Mon, 19 Sep 2022)

  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
    M mlir/test/Dialect/Linalg/named-ops.mlir

  Log Message:
  -----------
  [MLIR][Linalg] introduce batch-reduce GEMM

The batch-reduce GEMM kernel essentially multiplies a sequence of input tensor
blocks (which form a batch) and the partial multiplication results are reduced
into a single output tensor block.

See: https://ieeexplore.ieee.org/document/9139809 for more details.

Reviewed By: nicolasvasilache

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