[all-commits] [llvm/llvm-project] dd14e5: [mlir][vector] First step of vector distribution t...

[Thomas via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: dd14e5825209386129770296f9bc3a14ab0b4592
      https://github.com/llvm/llvm-project/commit/dd14e5825209386129770296f9bc3a14ab0b4592
  Author: Thomas Raoux <thomasraoux at google.com>
  Date:   2020-09-30 (Wed, 30 Sep 2020)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/VectorOps.td
    M mlir/include/mlir/Dialect/Vector/VectorTransforms.h
    M mlir/lib/Dialect/Vector/VectorOps.cpp
    M mlir/lib/Dialect/Vector/VectorTransforms.cpp
    M mlir/test/Dialect/Vector/invalid.mlir
    M mlir/test/Dialect/Vector/ops.mlir
    A mlir/test/Dialect/Vector/vector-distribution.mlir
    M mlir/test/lib/Transforms/TestVectorTransforms.cpp

  Log Message:
  -----------
  [mlir][vector] First step of vector distribution transformation

This is the first of several steps to support distributing large vectors. This
adds instructions extract_map and insert_map that allow us to do incremental
lowering. Right now the transformation only apply to simple pointwise operation
with a vector size matching the multiplicity of the IDs used to distribute the
vector.
This can be used to distribute large vectors to loops or SPMD.

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