[all-commits] [llvm/llvm-project] 5c1d35: [mlir][Linalg] Enable vectorization of explicit br...

[Diego Caballero via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 5c1d356c18c3ef9e63cf9607af660fb0b1c8a6ca
      https://github.com/llvm/llvm-project/commit/5c1d356c18c3ef9e63cf9607af660fb0b1c8a6ca
  Author: Diego Caballero <diegocaballero at google.com>
  Date:   2021-10-12 (Tue, 12 Oct 2021)

  Changed paths:
    M mlir/include/mlir/IR/AffineMap.h
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/lib/IR/AffineMap.cpp
    M mlir/test/Dialect/Linalg/vectorization.mlir

  Log Message:
  -----------
  [mlir][Linalg] Enable vectorization of explicit broadcasts

This patch teaches `isProjectedPermutation` and `inverseAndBroadcastProjectedPermutation`
utilities to deal with maps representing an explicit broadcast, e.g., (d0, d1) -> (d0, 0).
This extension is needed to enable vectorization of such explicit broadcast in Linalg.

Reviewed By: pifon2a, nicolasvasilache

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


  Commit: eeb09fd6465e3ee4d564fb0ab948231baf1daadd
      https://github.com/llvm/llvm-project/commit/eeb09fd6465e3ee4d564fb0ab948231baf1daadd
  Author: Diego Caballero <diegocaballero at google.com>
  Date:   2021-10-12 (Tue, 12 Oct 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/test/Dialect/Linalg/vectorization.mlir

  Log Message:
  -----------
  [mlir][Linalg] Enable vectorization of 'mul', 'and', 'or' and 'xor' reductions

This patch adds support for vectorizing 'mul', 'and', 'or' anx 'xor' reductions
to Linalg.

Reviewed By: pifon2a, ThomasRaoux, aartbik

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


Compare: https://github.com/llvm/llvm-project/compare/f09dce564e9c...eeb09fd6465e