[all-commits] [llvm/llvm-project] 0fd0fb: Reland: [mlir][Affine][Vector] Add initial support...

[Diego Caballero via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0fd0fb53299315d3942ce4ccbfaf4f4d4bc17315
      https://github.com/llvm/llvm-project/commit/0fd0fb53299315d3942ce4ccbfaf4f4d4bc17315
  Author: Diego Caballero <diego.caballero at intel.com>
  Date:   2021-03-12 (Fri, 12 Mar 2021)

  Changed paths:
    M mlir/lib/Dialect/Affine/Transforms/SuperVectorize.cpp
    M mlir/test/Dialect/Affine/SuperVectorize/vectorize_1d.mlir

  Log Message:
  -----------
  Reland: [mlir][Affine][Vector] Add initial support for 'iter_args' to Affine vectorizer.

This patch adds support for vectorizing loops with 'iter_args' when those loops
are not a vector dimension. This allows vectorizing outer loops with an inner
'iter_args' loop (e.g., reductions). Vectorizing scenarios where 'iter_args'
loops are vector dimensions would require more work (e.g., analysis,
generating horizontal reduction, etc.) not included in this patch.

Reviewed By: nicolasvasilache

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