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

[lorenzo chelini via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3718082e2b11cd1c66317bc067aac7cb094226cc
      https://github.com/llvm/llvm-project/commit/3718082e2b11cd1c66317bc067aac7cb094226cc
  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