[all-commits] [llvm/llvm-project] bcfd32: [mlir][linalg] Swap extract_slice(fill(x)) ops

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: bcfd32adc4b658dc45aa8c338d5dd03837e2a0e4
      https://github.com/llvm/llvm-project/commit/bcfd32adc4b658dc45aa8c338d5dd03837e2a0e4
  Author: Matthias Springer <springerm at google.com>
  Date:   2023-01-06 (Fri, 06 Jan 2023)

  Changed paths:
    M mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
    M mlir/test/Dialect/Linalg/canonicalize.mlir

  Log Message:
  -----------
  [mlir][linalg] Swap extract_slice(fill(x)) ops

This pattern is similar to `FoldFillWithTensorReshape`, which performs the same swapping with reshapes.

Fill the smaller extracted tensor slice instead of `x`. This allows for additional simplifications in case `x` is the result of another extract_slice.

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