[all-commits] [llvm/llvm-project] 5d4603: [mlir][tensor] Support more cases in MergeConsecut...

[Lei Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 5d4603a02d0c3e0106b10d245322b1d2072c0c3d
      https://github.com/llvm/llvm-project/commit/5d4603a02d0c3e0106b10d245322b1d2072c0c3d
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2022-09-20 (Tue, 20 Sep 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Tensor/Transforms/TransformUtils.h
    M mlir/lib/Dialect/Tensor/Transforms/MergeConsecutiveInsertExtractSlicePatterns.cpp
    M mlir/test/Dialect/Tensor/fold-consecutive-insert-extract-slice.mlir

  Log Message:
  -----------
  [mlir][tensor] Support more cases in MergeConsecutiveExtractSlice

This commit adds utility functions to perform general merging of
OffsetSizeAndStrideOpInterface by supporting producer rank
reducing and non-unit strides.

With it we can extend MergeConsecutiveExtractSlice to support
more cases.

Co-authored-by: Mahesh Ravishankar <ravishankarm at google.com>

Reviewed By: ThomasRaoux

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