[all-commits] [llvm/llvm-project] 9f1221: Recommit "[mlir][vector] Allow unroll of contracti...

[Chris via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9f1221521f4b0e4ea4b76d84b0d6ef12da2a4a8b
      https://github.com/llvm/llvm-project/commit/9f1221521f4b0e4ea4b76d84b0d6ef12da2a4a8b
  Author: Christopher Bate <cbate at nvidia.com>
  Date:   2022-06-09 (Thu, 09 Jun 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h
    M mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp
    M mlir/test/Dialect/Vector/vector-transfer-unroll.mlir
    M mlir/test/Dialect/Vector/vector-unroll-options.mlir
    M mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp

  Log Message:
  -----------
  Recommit "[mlir][vector] Allow unroll of contraction in arbitrary order"

Fixed issue with vector.contract default unroll permutation.

Adds support for vector unroll transformations to unroll in different
orders. For example, the vector.contract can be unrolled into a
smaller set of contractions. There is a choice of how to unroll the
decomposition based on the traversal order of (dim0, dim1, dim2).
The choice of traversal order can now be specified by a callback which
given by the caller of the transform. For now, only the
vector.contract, vector.transfer_read/transfer_write operations
support the callback.

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