[all-commits] [llvm/llvm-project] 9c4971: [mlir][Linalg] Refactor vectorization of conv1d mo...

[Nicolas Vasilache via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9c4971740b875530d21d3b73d8843fa88249085f
      https://github.com/llvm/llvm-project/commit/9c4971740b875530d21d3b73d8843fa88249085f
  Author: Nicolas Vasilache <nicolas.vasilache at gmail.com>
  Date:   2021-11-03 (Wed, 03 Nov 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/test/Dialect/Linalg/vectorize-convolution.mlir

  Log Message:
  -----------
  [mlir][Linalg] Refactor vectorization of conv1d more aggressively.

This better decouples transfer read/write from vector-only rewrite of conv.
This form is close to ready to plop into a new vector.conv op and the vector.transfer operations to be generalized as part of generic vectorization once the properties ConvolutionOpInterface are inferred from the indexing maps.

This also results in a nice perf boost in the dw == 1 cases.

Differential revision: https://reviews.llvm.org/D112822