[all-commits] [llvm/llvm-project] 3a3ab2: [mlir][Vector] Add support for high-order masked c...

[Diego Caballero via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3a3ab2147d7a6673c7b22bd14d8bf33ab1276d85
      https://github.com/llvm/llvm-project/commit/3a3ab2147d7a6673c7b22bd14d8bf33ab1276d85
  Author: Diego Caballero <diegocaballero at google.com>
  Date:   2023-02-22 (Wed, 22 Feb 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h
    M mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp
    M mlir/test/Dialect/Vector/vector-contract-transforms.mlir

  Log Message:
  -----------
  [mlir][Vector] Add support for high-order masked contractions

This patch adds support for masked vector.contract ops that needs to be
decomposed using the ContractionOpLowering pattern. It just slices the
mask according to the rest of the lowering.

Reviewed By: ThomasRaoux

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