[all-commits] [llvm/llvm-project] c503dc: [mlir][linalg] Add vectorization for element-wise ...

Thomas via All-commits all-commits at lists.llvm.org
Thu Dec 3 15:36:51 PST 2020


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c503dc1b8a52946e4daefa1a266e74a102382971
      https://github.com/llvm/llvm-project/commit/c503dc1b8a52946e4daefa1a266e74a102382971
  Author: Thomas Raoux <thomasraoux at google.com>
  Date:   2020-12-03 (Thu, 03 Dec 2020)

  Changed paths:
    M mlir/include/mlir/EDSC/Builders.h
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/lib/EDSC/Builders.cpp
    M mlir/test/Dialect/Linalg/transform-patterns-matmul-to-vector.mlir
    M mlir/test/Dialect/Linalg/transform-patterns.mlir
    A mlir/test/Dialect/Linalg/vectorization.mlir
    M mlir/test/lib/Transforms/TestLinalgTransforms.cpp

  Log Message:
  -----------
  [mlir][linalg] Add vectorization for element-wise linalg ops

Add support for vectorization for linalg.generic representing element-wise ops.
Those are converted to transfer_read + vector ops + transfer_write.
Also re-organize the vectorization tests to be together.

Implementation derived from the work of @burmako, @agrue and
@fedelebron.

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




More information about the All-commits mailing list